arrow
返回

Source Code Vulnerability Detection Method Based on Hybrid Retrieval-Augmented Generation Technology

delete2026-01-01
delete0
PRE
AI
X
Xie, Honggang
L
Li, Xiangdong *
S
Shen, Zenghui
DOI:10.1007/978-981-95-5294-8_35delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
随着软件系统规模和复杂性的持续增长,漏洞检测已成为确保软件安全的关键任务。近年来,结合大型语言模型(LLMs)的检索增强生成(RAG)技术在源代码漏洞检测中展现出显著潜力。然而,现有的基于RAG的方法通常依赖单一的检索策略,即稀疏检索或密集检索。虽然稀疏检索擅长精确的关键词匹配,但在捕捉深层语义关系方面存在困难,而密集检索虽然能够建模语义相似性,却可能无法检索低频或完全匹配的关键词。这种权衡限制了漏洞信息检索的覆盖面和准确性。为解决这些挑战,本文提出HRvul-RAG,一种基于混合检索的漏洞检测框架,整合了BM25和bge-m3双模态检索。通过利用稀疏和密集检索的互补优势,HRvul-RAG提升了召回率和检索精度。此外,为优化检索排序,我们引入加权倒数排名融合(WRRF)算法,该算法基于查询相关性动态调整排序权重,确保检索融合的鲁棒性和有效性。优化后的检索结果随后输入LLM进行漏洞分析,从而提升源代码漏洞检测的准确性和智能化水平。
Keyword:
Vulnerability search
RAG
LLM
bge-m3
BM25

期刊

C
COMPUTER APPLICATIONS, CCF NCCA 2025
IF:
0
论文数:
35
被引数:
0

机构

Z
zhongyuan university of technology
学者数:
1.0K
论文数: 306
被引数: 0
引用论文

引用论文

err分享
err收藏
Techniques for Inverted Index Compression
err2020-12-06
err33
errOAAI
errPibiri, Giulio Ermanno; Venturini, Rossano
err分享
err收藏
Ensemble clustering based on dense representation
err2019-09-01
err49
PREAI
errZhou, Jie; Zheng, Hongchan; Pan, Lulu
err分享
err收藏
A Survey of Sparse Representation: Algorithms and Applications稀疏表示综述: 算法与应用
err2015-01-01
err957
errOAAI
errZhang, Zheng; Xu, Yong; Yang, Jian; Li, Xuelong; Zhang, David
err分享
err收藏
没有更多内容