Automatic Data Labeling for Software Vulnerability Prediction Models: How Far Are We?
Not on the map. Why?
Abstract (arXiv)
Background: Software Vulnerability (SV) prediction needs large-sized and high-quality data to perform well. Current SV datasets mostly require expensive labeling efforts by experts (human-labeled) and thus are limited in size. Meanwhile, there are growing efforts in automatic SV labeling at scale. However, the fitness of auto-labeled data for SV prediction is still largely unknown. Aims: We quantitatively and qualitatively study the quality and use of the state-of-the-art auto-labeled SV data, D2A, for SV prediction. Method: Using multiple sources and manual validation, we curate clean SV data from human-labeled SV-fixing commits in two well-known projects for investigating the auto-labeled counterparts. Results: We discover that 50+% of the auto-labeled SVs are noisy (incorrectly labeled), and they hardly overlap with the publicly reported ones. Yet, SV prediction models utilizing the noisy auto-labeled SVs can perform up to 22% and 90% better in Matthews Correlation Coefficient and Recall, respectively, than the original models. We also reveal the promises and difficulties of applying noise-reduction methods for automatically addressing the noise in auto-labeled SV data to maximize the data utilization for SV prediction. Conclusions: Our study informs the benefits and challenges of using auto-labeled SVs, paving the way for large-scale SV prediction.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 30 of its 83 references that are in the map. First posted on arXiv 25 Jul 2024.
Builds on
30 of 83 references.
References · 30
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2013Distributed Representations of Words and Phrases and their Compositionalitycited by 3,474
- 2016Enriching Word Vectors with Subword Informationcited by 1,583
- 2020CodeBERT: A Pre-Trained Model for Programming and Natural Languagescited by 1,310
- 2014Distributed Representations of Sentences and Documentscited by 678
- 2019Confident Learning: Estimating Uncertainty in Dataset Labelscited by 413
- 2020Challenges in Deploying Machine Learning: a Survey of Case Studiescited by 152
- 2021CVEfixes: Automated Collection of Vulnerabilities and Their Fixes from Open-Source Softwarecited by 91
- 2021Vulnerability Detection with Fine-grained Interpretationscited by 56
- 2020Deep Learning for Source Code Modeling and Generation: Models, Applications and Challengescited by 55
- 2022VulBERTa: Simplified Source Code Pre-Training for Vulnerability Detectioncited by 51
18 more
- 2021D2A: A Dataset Built for AI-Based Vulnerability Detection Methods Using Differential Analysiscited by 49
- 2022An Empirical Study of Deep Learning Models for Vulnerability Detectioncited by 46
- 2022LineVD: Statement-level Vulnerability Detection using Graph Neural Networkscited by 35
- 2021A Survey on Data-driven Software Vulnerability Assessment and Prioritizationcited by 18
- 2021Automated Software Vulnerability Assessment with Concept Driftcited by 15
- 2018A Practical Approach to the Automatic Classification of Security-Relevant Commitscited by 13
- 2020A Large-scale Study of Security Vulnerability Support on Developer Q&A Websitescited by 12
- 2020PUMiner: Mining Security Posts from Developer Question and Answer Websites with PU Learningcited by 10
- 2024Are Latent Vulnerabilities Hidden Gems for Software Vulnerability Prediction? An Empirical Studycited by 9
- 2024Software Vulnerability Prediction in Low-Resource Languages: An Empirical Study of CodeBERT and ChatGPTcited by 9
- 2021An Empirical Study of Rule-Based and Learning-Based Approaches for Static Application Security Testingcited by 7
- 2022Noisy Label Learning for Security Defectscited by 7
- 2021Automated Security Assessment for the Internet of Thingscited by 6
- 2022Less is More: Supporting Developers in Vulnerability Detection during Code Reviewcited by 6
- 2024Systematic Literature Review on Application of Learning-based Approaches in Continuous Integrationcited by 6
- 2024Automated Code-centric Software Vulnerability Assessment: How Far Are We? An Empirical Study in C/C++cited by 6
- 2020Learning from What We Know: How to Perform Vulnerability Prediction using Noisy Historical Datacited by 5
- 2019"Won't We Fix this Issue?" Qualitative Characterization and Automated Identification of Wontfix Issues on GitHubcited by 4
Built on by
Citing papers per year: 2024: 1; 2025: 5.
6 papers.
- 2024VulnLLMEval: A Framework for Evaluating Large Language Models in Software Vulnerability Detection and Patchingcited by 9
- 2025LLMSecConfig: An LLM-Based Approach for Fixing Software Container Misconfigurationscited by 9
- 2025Data and Context Matter: Towards Generalizing AI-based Software Vulnerability Detectioncited by 3
- 2025Securing the AI Supply Chain: What Can We Learn From Developer-Reported Security Issues and Solutions of AI Projects?cited by 1
- 2025AVIATOR: Towards AI-Agentic Vulnerability Injection Workflow for High-Fidelity, Large-Scale Code Security Datasetcited by 0
- 2025Software Vulnerability Management in the Era of Artificial Intelligence: An Industry Perspectivecited by 0
- 2025Securing the AI Supply Chain: What Can We Learn From Developer-Reported Security Issues and Solutions of AI Projects?29 Dec 2025
- 2025Software Vulnerability Management in the Era of Artificial Intelligence: An Industry Perspective20 Dec 2025
- 2025AVIATOR: Towards AI-Agentic Vulnerability Injection Workflow for High-Fidelity, Large-Scale Code Security Dataset28 Aug 2025
- 2025Data and Context Matter: Towards Generalizing AI-based Software Vulnerability Detection14 Aug 2025
- 2025LLMSecConfig: An LLM-Based Approach for Fixing Software Container Misconfigurations4 Feb 2025
- 2024VulnLLMEval: A Framework for Evaluating Large Language Models in Software Vulnerability Detection and Patching16 Sep 2024