We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: ReGVD: Revisiting Graph Neural Networks for Vulnerability Detection

Abstract: Identifying vulnerabilities in the source code is essential to protect the software systems from cyber security attacks. It, however, is also a challenging step that requires specialized expertise in security and code representation. To this end, we aim to develop a general, practical, and programming language-independent model capable of running on various source codes and libraries without difficulty. Therefore, we consider vulnerability detection as an inductive text classification problem and propose ReGVD, a simple yet effective graph neural network-based model for the problem. In particular, ReGVD views each raw source code as a flat sequence of tokens to build a graph, wherein node features are initialized by only the token embedding layer of a pre-trained programming language (PL) model. ReGVD then leverages residual connection among GNN layers and examines a mixture of graph-level sum and max poolings to return a graph embedding for the source code. ReGVD outperforms the existing state-of-the-art models and obtains the highest accuracy on the real-world benchmark dataset from CodeXGLUE for vulnerability detection. Our code is available at: \url{this https URL}.
Comments: Accepted to ICSE 2022 (Demonstrations). The first two authors contributed equally to this work
Subjects: Machine Learning (cs.LG); Cryptography and Security (cs.CR)
Cite as: arXiv:2110.07317 [cs.LG]
  (or arXiv:2110.07317v3 [cs.LG] for this version)

Submission history

From: Dai Quoc Nguyen [view email]
[v1] Thu, 14 Oct 2021 12:44:38 GMT (238kb,D)
[v2] Tue, 1 Feb 2022 00:28:46 GMT (187kb,D)
[v3] Sat, 5 Feb 2022 02:24:42 GMT (146kb,D)

Link back to: arXiv, form interface, contact.