Security practitioners maintain vulnerability reports (e.g., GitHub Advisory)
to help developers mitigate security risks. An important task for these
databases is automatically extracting structured information mentioned in the
report, e.g., the affected software packages, to accelerate the defense of the
vulnerability ecosystem.
However, it is challenging for existing work on affected package
identification to achieve a high accuracy. One reason is that all existing work
focuses on relatively smaller models, thus they cannot harness the knowledge
and semantic capabilities of large language models.
To address this limitation, we propose VulLibGen, the first method to use LLM
for affected package identification. In contrast to existing work, VulLibGen
proposes the novel idea to directly generate the affected package. To improve
the accuracy, VulLibGen employs supervised fine-tuning (SFT), retrieval
augmented generation (RAG) and a local search algorithm. The local search
algorithm is a novel postprocessing algorithm we introduce for reducing the
hallucination of the generated packages. Our evaluation results show that
VulLibGen has an average accuracy of 0.806 for identifying vulnerable packages
in the four most popular ecosystems in GitHub Advisory (Java, JS, Python, Go)
while the best average accuracy in previous work is 0.721. Additionally,
VulLibGen has high value to security practice: we submitted 60 <vulnerability,
affected package> pairs to GitHub Advisory (covers four ecosystems). 34 of them
have been accepted and merged and 20 are pending approval. Our code and dataset
can be found in the attachments.
外部データセット
VulLib
参考文献
IEEE Transactions on Dependable and Secure Computing
Cleaning the nvd: Comprehensive quality assessment, improvements, and analyses
Afsah Anwar, Ahmed Abusnaina, Songqing Chen, Frank Li, David Mohaisen
Published: 2021
ACM SIGSAC Conference on Computer and Communications Security
Reliable third-party library detection in Android and its security applications
Michael Backes, Sven Bugiel, Erik Derr
Published: 2016
International ACM SIGIR Conference on Research and Development in Information Retrieval
Recent advances in retrieval-augmented text generation
Deng Cai, Yan Wang, Lemao Liu, Shuming Shi
Published: 2022
The International Conference on Learning Representations
CodeT: Code Generation with Generated Tests
Bei Chen, Fengji Zhang, Anh Nguyen, Daoguang Zan, Zeqi Lin, Jian-Guang Lou, Weizhu Chen
C. Olsson, N. Elhage, N. Nanda, N. Joseph, N. DasSarma, T. Henighan, B. Mann, A. Askell, Y. Bai, A. Chen, T. Conerly, D. Drain, D. Ganguli, Z. Hatfield-Dodds, D. Hernandez, S. Johnston, A. Jones, J. Kernion, L. Lovitt, K. Ndousse, D. Amodei, T. Brown, J. Clark, J. Kaplan, S. McCandlish, C. Olah
Published: 2022
Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies
Fast lexically constrained decoding with dynamic beam allocation for neural machine translation
Matt Post, David Vilar
Published: 2018
Sensors
Transfer learning for sentiment analysis using BERT based supervised fine-tuning
Nusrat Jahan Prottasha, Abdullah As Sami, Md Kowsher, Saydul Akbar Murad, Anupam Kumar Bairagi, Mehedi Masud, Mohammed Baz
Published: 2022
International Conference on Learning Representations
Leveraging automated unit tests for unsupervised code translation
Baptiste Roziere, Jie Zhang, Francois Charton, Mark Harman, Gabriel Synnaeve, Guillaume Lample
Published: 2022
Chemometrics and intelligent laboratory systems
Analysis of variance (ANOVA)
Lars St, Svante Wold
Published: 1989
Proceedings of the 19th International Conference on Mining Software Repositories
LibDB: An Effective and Efficient Framework for Detecting Third-Party Libraries in Binaries