Abstract:Code Language Models (codeLMs) and Graph Neural Networks (GNNs) are widely used in code vulnerability detection. However, a critical yet often overlooked issue is that GNNs primarily rely on aggregating information from adjacent nodes, limiting structural information transfer to single-layer updates. In code graphs, nodes and relationships typically require cross-layer information propagation to fully capture complex program logic and potential vulnerability patterns. Furthermore, while some studies utilize codeLMs to supplement GNNs with code semantic information, existing integration methods have not fully explored the potential of their collaborative effects. To address these challenges, we introduce Vul-LMGNNs that integrates pre-trained CodeLMs with GNNs, leveraging knowledge distillation to facilitate cross-layer propagation of both code semantic knowledge and structural information. Specifically, Vul-LMGNNs utilizes Code Property Graphs (CPGs) to incorporate code syntax, control flow, and data dependencies, while employing gated GNNs to extract structural information in the CPG. To achieve cross-layer information transmission, we implement an online knowledge distillation (KD) program that enables a single student GNN to acquire structural information extracted from a simultaneously trained counterpart through an alternating training procedure. Additionally, we leverage pre-trained CodeLMs to extract semantic features from code sequences. Finally, we propose an "implicit-explicit" joint training framework to better leverage the strengths of both CodeLMs and GNNs. In the implicit phase, we utilize CodeLMs to initialize the node embeddings of each student GNN. Through online knowledge distillation, we facilitate the propagation of both code semantics and structural information across layers. In the explicit phase, we perform linear interpolation between the CodeLM and the distilled GNN to learn a late fusion model. The proposed method, evaluated across four real-world vulnerability datasets, demonstrated superior performance compared to 17 state-of-the-art approaches. Our source code can be accessed via GitHub: https: //github.com/Vul-LMGNN/vul-LMGGNN.

Code classification with graph neural networks: Have you ever struggled to make it work?

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

Program Classification Using Gated Graph Attention Neural Network for Online Programming Service

Graph Neural Networks Inspired by Classical Iterative Algorithms

Application of Graph Neural Networks and graph descriptors for graph classification

Vul-LMGNNs: Fusing Language Models and Online-Distilled Graph Neural Networks for Code Vulnerability Detection

When Do Graph Neural Networks Help with Node Classification: Investigating the Homophily Principle on Node Distinguishability

Are Powerful Graph Neural Nets Necessary? A Dissection on Graph Classification

Semisupervised Graph Neural Networks for Graph Classification

Graph Neural Networks and Their Current Applications in Bioinformatics

When Do Graph Neural Networks Help with Node Classification? Investigating the Impact of Homophily Principle on Node Distinguishability

Graph Neural Networks: Foundation, Frontiers and Applications

Evaluating Deep Graph Neural Networks

A Survey on Graph Classification and Link Prediction based on GNN

Bag of Tricks for Node Classification with Graph Neural Networks

Graph structure reforming framework enhanced by commute time distance for graph classification

AutoGraph: Automated Graph Neural Network

Text Level Graph Neural Network for Text Classification

Text classification optimization algorithm based on graph neural network

Imbalanced Graph Classification via Graph-of-Graph Neural Networks

A Novel Two-Channel Classification Approach Using Graph Attention Network with K-Nearest Neighbor