Enhancing Graph-Based Vulnerability Detection through Standardized Deep Learning Pipelines

Identifying vulnerable code is crucial to software system security. With the rise of deep learning, graph neural networks (GNNs) have received much attention for detecting vulnerabilities. While many studies aim to enhance GNN performance by refining graph structures, improving datasets, or optimizing model architectures, three key questions remain unanswered: 1) How do simple graph structures impact model performance? 2) How similar are artificially generated datasets to vulnerable code in the real world? 3) Does a more complex model provide better benefits in vulnerability detection? To find answers to these questions, we did experiments to show how basic graph structures affect GNN models and give guidelines on how to choose datasets. In addition, we introduce VulGCANet, a model with a relatively simple architecture that utilizes code property graphs and combines Graph Convolutional Networks (GCN) and Graph Attention Network (GAT) layers for vulnerability detection. The experimental results demonstrate that VulGCANet exhibits 30% improvements in recall while performing similarly to other graph neural network-based models, highlighting the importance of reducing the complexity of GNN model architectures in graph-based vulnerability detection. These findings provide valuable insights for advancing GNN-based vulnerability detection efforts.

Authors

Institutions

Publication Details

Published
2024-12-17
DOI
https://doi.org/10.1109/trustcom63139.2024.00174
Primary Topic
Information and Cyber Security
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Enhancing Graph-Based Vulnerability Detection through Standardized Deep Learning Pipelines

Youngwoo Kwon, J. Hao
Information and Cyber Security
article

Enhancing Graph-Based Vulnerability Detection through Standardized Deep Learning Pipelines

Youngwoo Kwon, J. Hao
article en

Abstract

Identifying vulnerable code is crucial to software system security. With the rise of deep learning, graph neural networks (GNNs) have received much attention for detecting vulnerabilities. While many studies aim to enhance GNN performance by refining graph structures, improving datasets, or optimizing model architectures, three key questions remain unanswered: 1) How do simple graph structures impact model performance? 2) How similar are artificially generated datasets to vulnerable code in the real world? 3) Does a more complex model provide better benefits in vulnerability detection? To find answers to these questions, we did experiments to show how basic graph structures affect GNN models and give guidelines on how to choose datasets. In addition, we introduce VulGCANet, a model with a relatively simple architecture that utilizes code property graphs and combines Graph Convolutional Networks (GCN) and Graph Attention Network (GAT) layers for vulnerability detection. The experimental results demonstrate that VulGCANet exhibits 30% improvements in recall while performing similarly to other graph neural network-based models, highlighting the importance of reducing the complexity of GNN model architectures in graph-based vulnerability detection. These findings provide valuable insights for advancing GNN-based vulnerability detection efforts.

Kyungpook National University (KR)
Ministry of Education, National Research Foundation
Openalex Percentile: Top 54%
Information and Cyber Security
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Enhancing Graph-Based Vulnerability Detection through Standardized Deep Learning Pipelines — Youngwoo Kwon, J. Hao · (2024) | TGRS Research Map | TGRS