Research on the LLM-Driven Vulnerability Detection System Using LProtector
The security issues of large-scale software systems and frameworks have become increasingly severe with the development of technology. As complexity of software grows, vulnerabilities are becoming more challenging to detect. Although traditional machine learning methods have been applied in cybersecurity for a long time, there has been no significant breakthrough until now. With the recent rise of large language models (LLMs), a turning point seems to have arrived. The powerful code comprehension and generation capabilities of LLMs make fully automated vulnerability detection systems a possibility. This paper presents LProtector, an automated vulnerability detection system for C/C++ codebases based on GPT-4o and Retrieval-Augmented Generation (RAG). LProtector performs binary classification to identify vulnerabilities in target codebases. To evaluate its effectiveness, we conducted experiments on the Big- Vul dataset. Results show that LProtector outperforms two state-of-the-art baselines in terms of F1 score, demonstrating the potential of integrating LLMs with vulnerability detection.
Authors
- Fengying Wu (ORCID: https://orcid.org/0000-0001-5129-0528)
- Chao Li (ORCID: https://orcid.org/0000-0001-6332-5898)
- Hang Lei (ORCID: https://orcid.org/0000-0001-7736-2251)
- Ze Sheng
- Xiangwu Zuo
- Yuxin Qiao
Institutions
- University of Electronic Science and Technology of China (CN)
- Georgetown University (US)
- Northern Arizona University (US)
- University of Pennsylvania (US)
- Texas A&M University (US)
- Philadelphia University (US)
Publication Details
- Published
- 2024-11-22
- DOI
- https://doi.org/10.1109/icdsca63855.2024.10859408
- Citations
- 9
- Primary Topic
- Network Security and Intrusion Detection
- Type
- article
- Field-Weighted Citation Impact
- 3.14