arrow
返回

Code Aggregate Graph: Effective Representation for Graph Neural Networks to Detect Vulnerable Code

delete2022-01-01
delete4
delete
OA
AI
H
Hoang Viet Nguyen *
J
Junjun Zheng
A
Atsuo Inomata
T
Tetsutaro Uehara
DOI:10.1109/ACCESS.2022.3216395delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Deep learning, especially graph neural networks (GNNs), provides efficient, fast, and automated methods to detect vulnerable code. However, the accuracy could be improved as previous studies were limited by existing code representations. Additionally, the diversity of embedding techniques and GNN models can make selecting the appropriate method challenging. Herein we propose Code Aggregate Graph (CAG) to improve vulnerability detection efficiency. CAG combines the principles of different code analyses such as abstract syntax tree, control flow graph, and program dependence graph with dominator and post-dominator trees. This extensive representation empowers deep graph networks for enhanced classification. We also implement different data encoding methods and neural networks to provide a multidimensional view of the system performance. Specifically, three word embedding approaches and three deep GNNs are utilized to build classifiers. Then CAG is evaluated using two datasets: a real-world open-source dataset and the software assurance reference dataset. CAG is also compared with seven state-of-the-art methods and six classic representations. CAG shows the best performance. Compared to previous studies, CAG has an increased accuracy (5.4%) and F1-score (5.1%). Additionally, experiments confirm that encoding has a positive impact on accuracy (4-6%) but the network type does not. The study should contribute to a meaningful benchmark for future research on code representations, data encoding, and GNNs.
Keyword:
Codes
Source coding
Computer bugs
Syntactics
Deep learning
Feature extraction
Graph neural networks
Vulnerability detection
code representation
graph neural networks
deep learning

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

T
the university of osaka
学者数:
2.8W
论文数: 1.8W
被引数: 6
R
ritsumeikan university
学者数:
4.0K
论文数: 3.6K
被引数: 0
引用论文

引用论文

Titanium ketimide complexes as α-olefin homo- and copolymerisation catalysts. X-ray diffraction structures of [TiCp′(NCtBu2)Cl2] (Cp′=Ind, Cp*)
err2004-01-01
err0
PREAI
errAlberto R. Dias; M. Teresa Duarte; Anabela C. Fernandes; Susete Fernandes; Maria M. Marques; Ana M. Martins; João F. da Silva; Sandra S. Rodrigues
err分享
err收藏
Nature-Based Citizen Science as a Mechanism to Improve Human Health in Urban Areas
err2021-12-22
err0
errOAAI
errCraig R. Williams; Sophie M. Burnell; Michelle Rogers; Emily J. Flies; Katherine L. Baldock
err分享
err收藏
err分享
err收藏
err分享
err收藏
err分享
err收藏
Improving Vulnerability Inspection Efficiency Using Active Learning
err2021-11-01
err31
errOAAI
errYu, Zhe; Theisen, Christopher; Williams, Laurie; Menzies, Tim
err分享
err收藏
学者 查看更多内容