详细信息
基于多通道图神经网络和CNN-BiLSTM的漏洞检测方法
Vulnerability Detection Method Based on Multi-Channel Graph Neural Network and CNN-BiLSTM
文献类型:期刊文献
中文题名:基于多通道图神经网络和CNN-BiLSTM的漏洞检测方法
英文题名:Vulnerability Detection Method Based on Multi-Channel Graph Neural Network and CNN-BiLSTM
作者:李鹏威[1];郑红[1];单蓉胜[2]
机构:[1]华东理工大学信息科学与工程学院,上海200237;[2]上海交通大学网络空间安全学院,上海200240
年份:2025
卷号:51
期号:6
起止页码:835
中文期刊名:华东理工大学学报(自然科学版)
外文期刊名:Journal of East China University of Science and Technology
收录:;北大核心:【北大核心2023】;
基金:上海市2024年度“科技创新行动计划”资助(24BC3200500,24BC3200300)。
语种:中文
中文关键词:漏洞检测;图神经网络;特征融合;中心性分析;深度学习
外文关键词:vulnerability detection;graph neural network;feature fusion;centrality analysis;deep learning
摘要:针对当前基于深度学习的漏洞检测方法大多仅考虑代码序列语义或代码结构语义的问题,本文提出了基于多通道图神经网络(MCGNN)和卷积神经网络-双向长短期记忆网络(CNNBiLSTM)的漏洞检测模型MCGCBVul。该方法通过中心性分析将代码属性图的节点特征矩阵扩展为多通道的类图像矩阵,并使用图注意力网络(GATv2)和二维卷积神经网络(2D-CNN)提取图结构特征;同时,采用双尺度一维卷积神经网络(1D-CNN)以及双向长短期记忆网络(BiLSTM)提取序列特征;最终将图特征和序列特征进行特征融合,以达到更好的漏洞检测性能。结果表明,MCGCBVul在F1得分、准确率等多项指标上优于其他6种对比模型,FEMPeg+Qemu和Reveal这两个数据集的准确率分别达到了63.952%和92.007%。此外,本文通过消融实验进一步证明了模型各模块改进的有效性。
In view of the problem that most of the current vulnerability detection methods based on deep learning only consider the semantics of code sequence or code structure,this paper proposes a vulnerability detection method MCGCBVul based on MCGNN(Multi-Channel Graph Neural Network)and CNN-BiLSTM(Convolution Neural Network-Bidirectional Long Short-Term Memory).This method expands the node feature matrix of the CPG(Code Property Graph)into a multi-channel image-like matrix through the centrality analysis,and uses the graph attention network(GATv2)and the two-dimensional convolutional neural network(2D-CNN)to extract graph features.At the same time,the dual-scale one-dimensional convolutional neural network(1D-CNN)and the bidirectional long shortterm memory network(BiLSTM)are used to extract sequence features.Finally,the graph features and sequence features are fused to achieve better vulnerability detection performance.After experiments on the FFMPeg+Qemu and Reveal datasets,the experimental results show that MCGCBVul outperforms the six comparison models in this paper in multiple indicators such as F1 score and accuracy,with accuracy rates of 63.952%and 92.007%respectively.In addition,this paper further proves the effectiveness of the improvement of each module of the model through ablation experiments.
参考文献:
正在载入数据...
