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.

M2CVD: Multi-Model Collaboration for Code Vulnerability Detection

M2CVD: Enhancing Vulnerability Semantic through Multi-Model Collaboration for Code Vulnerability Detection

Function-Level Vulnerability Detection Through Fusing Multi-Modal Knowledge

Investigating Large Language Models for Code Vulnerability Detection: An Experimental Study

Cross-Modality Mutual Learning for Enhancing Smart Contract Vulnerability Detection on Bytecode

Generalization-Enhanced Code Vulnerability Detection via Multi-Task Instruction Fine-Tuning

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

Towards Effectively Detecting and Explaining Vulnerabilities Using Large Language Models

Multi-role Consensus through LLMs Discussions for Vulnerability Detection

How Far Have We Gone in Vulnerability Detection Using Large Language Models

MVD: A Multi-Lingual Software Vulnerability Detection Framework

Outside the Comfort Zone: Analysing LLM Capabilities in Software Vulnerability Detection

An Empirical Study of Automated Vulnerability Localization with Large Language Models

VulDetectBench: Evaluating the Deep Capability of Vulnerability Detection with Large Language Models

Multi-view Pre-trained Model for Code Vulnerability Identification

Enhancing Large Language Models for Secure Code Generation: A Dataset-driven Study on Vulnerability Mitigation

Finetuning Large Language Models for Vulnerability Detection

Security Vulnerability Detection with Multitask Self-Instructed Fine-Tuning of Large Language Models

Vulnerability Detection with Code Language Models: How Far Are We?

A Machine Learning Approach for Vulnerability Curation.

VTT-LLM: Advancing Vulnerability-to-Tactic-and-Technique Mapping through Fine-Tuning of Large Language Model