中文

LProtector:基于大语言模型的漏洞检测系统

密码学与安全 2024-11-15 v2 人工智能

摘要

本文提出 LProtector,一个面向 C/C++ 代码库的自动化漏洞检测系统,由大语言模型(LLM)GPT-4o 和检索增强生成(RAG)驱动。随着软件复杂度的增长,传统方法在有效检测漏洞方面面临挑战。LProtector 利用 GPT-4o 强大的代码理解和生成能力执行二分类,识别目标代码库中的漏洞。我们在 Big-Vul 数据集上进行了实验,结果表明 LProtector 在 F1 分数上优于两个最先进基线,展示了将大语言模型与漏洞检测相结合的潜力。

关键词

引用

@article{arxiv.2411.06493,
  title  = {LProtector: An LLM-driven Vulnerability Detection System},
  author = {Ze Sheng and Fenghua Wu and Xiangwu Zuo and Chao Li and Yuxin Qiao and Lei Hang},
  journal= {arXiv preprint arXiv:2411.06493},
  year   = {2024}
}

备注

5 pages, 4 figures. This is a preprint version of the article. The final version will be published in the proceedings of the IEEE conference