arXiv· Mahdieh Abazar·· 13 天前推荐指数40
CAT-RAG:用类别感知的智能体检索改进安全代码生成
CAT-RAG: Improving Secure Code Generation with Category-Aware Agentic Retrieval
AI 导读
CAT-RAG 是一个智能体式安全类别感知检索增强代码生成框架,用语言特定的安全类别替代 CWE 级检索,可作为独立安全层或 RACG 管线的即插即用附加组件,无需修改底层模型。在 CyberSecEval 上,其安全代码生成改进幅度接近当前最优方法的两倍(23.57% vs. 12.11%),同时减少 41.71% 的 token 开销。评估还覆盖 CodeGuard+ 与 HumanEval。
来源:arXiv · arxiv.org