Abstract:Automated detection of software vulnerabilities is a fundamental problem in software security. Existing program analysis techniques either suffer from high false positives or false negatives. Recent progress in Deep Learning (DL) has resulted in a surge of interest in applying DL for automated vulnerability detection. Several recent studies have demonstrated promising results achieving an accuracy of up to 95% at detecting vulnerabilities. In this paper, we ask, "how well do the state-of-the-art DL-based techniques perform in a real-world vulnerability prediction scenario?". To our surprise, we find that their performance drops by more than 50%. A systematic investigation of what causes such precipitous performance drop reveals that existing DL-based vulnerability prediction approaches suffer from challenges with the training data (e.g., data duplication, unrealistic distribution of vulnerable classes, etc.) and with the model choices (e.g., simple token-based models). As a result, these approaches often do not learn features related to the actual cause of the vulnerabilities. Instead, they learn unrelated artifacts from the dataset (e.g., specific variable/function names, etc.). Leveraging these empirical findings, we demonstrate how a more principled approach to data collection and model design, based on realistic settings of vulnerability prediction, can lead to better solutions. The resulting tools perform significantly better than the studied baseline: up to 33.57% boost in precision and 128.38% boost in recall compared to the best performing model in the literature. Overall, this paper elucidates existing DL-based vulnerability prediction systems' potential issues and draws a roadmap for future DL-based vulnerability prediction research. In that spirit, we make available all the artifacts supporting our results: <a class="link-external link-https" href="https://git.io/Jf6IA" rel="external noopener nofollow">this https URL</a>.

Systematic Analysis of Deep Learning Model for Vulnerable Code Detection

Ignnvd: A Novel Software Vulnerability Detection Model Based on Integrated Graph Neural Networks

Survey of Source Code Vulnerability Analysis Based on Deep Learning

Vulnerability Detection in C/C++ Code with Deep Learning

Automated software vulnerability detection with machine learning

Meta-heuristic-based hybrid deep learning model for vulnerability detection and prevention in software system

Software Vulnerability Mining and Analysis Based on Deep Learning

A Comparative Study of Deep Learning-Based Vulnerability Detection System

On the Code Vulnerability Detection Based on Deep Learning: A Comparative Study

Deep Learning based Vulnerability Detection: Are We There Yet?

A Hierarchical Deep Neural Network for Detecting Lines of Codes with Vulnerabilities

Software Vulnerability Detection via Deep Learning over Disaggregated Code Graph Representation

Vulnerability Detection Using Two-Stage Deep Learning Models

Security Vulnerability Detection Using Deep Learning Natural Language Processing

A Survey on Automated Software Vulnerability Detection Using Machine Learning and Deep Learning

Transformer-based Vulnerability Detection in Code at EditTime: Zero-shot, Few-shot, or Fine-tuning?

DiverseVul: A New Vulnerable Source Code Dataset for Deep Learning Based Vulnerability Detection

Detecting code vulnerabilities by learning from large-scale open source repositories

Automated Vulnerability Detection Using Deep Learning Technique

On Security Weaknesses and Vulnerabilities in Deep Learning Systems

Malware Analysis Using Machine Learning and Deep Learning Techniques