Source Code Vulnerability Detection Based on Joint Graph and Multimodal Feature Fusion

被引:0
作者
Jin, Dun [1 ]
He, Chengwan [1 ]
Zou, Quan [1 ]
Qin, Yan [1 ]
Wang, Boshu [1 ]
机构
[1] Wuhan Inst Technol, Sch Comp Sci & Engn, Hubei Key Lab Intelligent Robot, Wuhan 430205, Peoples R China
来源
ELECTRONICS | 2025年 / 14卷 / 05期
关键词
deep learning; vulnerability detection; joint graph; multimodal; feature fusion; FRAMEWORK;
D O I
10.3390/electronics14050975
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Code vulnerability detection is a crucial approach to ensuring software security, aiming to automatically identify potential security vulnerabilities in source code. However, existing static vulnerability detection methods often face challenges such as feature loss and an insufficient expressive power when extracting program features. To address these issues, in this paper, we propose a source code vulnerability detection method based on joint graph and multimodal feature fusion. Innovatively, we construct a joint graph that integrates multiple program dependencies and semantic edges to achieve more comprehensive feature extraction. Additionally, by combining Graph Attention Networks (GATs) with the Transformer architecture, both structural and sequential features of code snippets are captured, further enhancing the model's expressive capabilities. Finally, we introduce pre-fusion and post-fusion strategies to fully integrate multimodal features, thereby improving the accuracy and performance of vulnerability detection. Experimental results on the SARD dataset demonstrate the method's excellent performance in detecting various types of vulnerabilities, achieving an F1 score of 85.20% and an accuracy of 86.50%. On the Real-Vul real-world dataset, the method achieves an F1 score of 73.9% and an accuracy of 86.50%. The detection results exhibit remarkable stability, ensuring a reliable and consistent performance. Overall, the proposed method surpasses the performance of existing mainstream detection approaches.
引用
收藏
页数:20
相关论文
共 50 条
  • [31] CSGVD: A deep learning approach combining sequence and graph embedding for source code vulnerability detection
    Tang, Wei
    Tang, Mingwei
    Ban, Minchao
    Zhao, Ziguo
    Feng, Mingjun
    [J]. JOURNAL OF SYSTEMS AND SOFTWARE, 2023, 199
  • [32] SlicedLocator: Code vulnerability locator based on sliced dependence graph
    Wu, Bolun
    Zou, Futai
    Yi, Ping
    Wu, Yue
    Zhang, Liang
    [J]. COMPUTERS & SECURITY, 2023, 134
  • [33] An adaptive multi-graph neural network with multimodal feature fusion learning for MDD detection
    Xing, Tao
    Dou, Yutao
    Chen, Xianliang
    Zhou, Jiansong
    Xie, Xiaolan
    Peng, Shaoliang
    [J]. SCIENTIFIC REPORTS, 2024, 14 (01):
  • [34] Improving Vulnerability Detection with Hybrid Code Graph Representation
    Meng, Xiangxin
    Lu, Shaoxiao
    Wang, Xu
    Liu, Xudong
    Hu, Chunming
    [J]. PROCEEDINGS OF THE 2023 30TH ASIA-PACIFIC SOFTWARE ENGINEERING CONFERENCE, APSEC 2023, 2023, : 259 - 268
  • [35] Research on Extraction of Useful Tourism Online Reviews Based on Multimodal Feature Fusion
    Li, Meng
    [J]. ACM TRANSACTIONS ON ASIAN AND LOW-RESOURCE LANGUAGE INFORMATION PROCESSING, 2021, 20 (05)
  • [36] A vulnerability detection framework with enhanced graph feature learning
    Cheng, Jianxin
    Chen, Yizhou
    Cao, Yongzhi
    Wang, Hanpin
    [J]. JOURNAL OF SYSTEMS AND SOFTWARE, 2024, 216
  • [37] HeVulD: A Static Vulnerability Detection Method Using Heterogeneous Graph Code Representation
    Huang, Yuanming
    He, Mingshu
    Wang, Xiaojuan
    Zhang, Jie
    [J]. IEEE TRANSACTIONS ON INFORMATION FORENSICS AND SECURITY, 2024, 19 : 9129 - 9144
  • [38] VulBERTa: Simplified Source Code Pre-Training for Vulnerability Detection
    Hanif, Hazim
    Maffeis, Sergio
    [J]. 2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,
  • [39] FSFM: A Feature Square Tower Fusion Module for Multimodal Object Detection
    Liu, Xiaomin
    Zhu, Chen
    Yang, Chunyu
    Zhou, Linna
    [J]. IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2023, 72
  • [40] Graph Fusion Network-Based Multimodal Learning for Freezing of Gait Detection
    Hu, Kun
    Wang, Zhiyong
    Martens, Kaylena A. Ehgoetz
    Hagenbuchner, Markus
    Bennamoun, Mohammed
    Tsoi, Ah Chung
    Lewis, Simon J. G.
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (03) : 1588 - 1600