Message Passing Graph Neural Networks for Software Security Vulnerability Detection
Author
Abstract

Vulnerability Detection 2022 - With the booming development of deep learning and machine learning, the use of neural networks for software source code security vulnerability detection has become a hot pot in the field of software security. As a data structure, graphs can adequately represent the complex syntactic information, semantic information, and dependencies in software source code. In this paper, we propose the MPGVD model based on the idea of text classification in natural language processing. The model uses BERT for source code pre-training, transforms graphs into corresponding feature vectors, uses MPNN (Message Passing Neural Networks) based on graph neural networks in the feature extraction phase, and finally outputs the detection results. Our proposed MPGVD, compared with other existing vulnerability detection models on the same dataset CodeXGLUE, obtain the highest detection accuracy of 64.34\%.

Year of Publication
2022
Date Published
sep
Publisher
IEEE
Conference Location
Xi an, China
ISBN Number
978-1-66549-109-9
URL
https://ieeexplore.ieee.org/document/9933216/
DOI
10.1109/ICCNEA57056.2022.00041
Google Scholar | BibTeX | DOI