一种基于改进图神经网络的源代码漏洞检测方法

本发明提供的一种基于改进图神经网络的源代码漏洞检测方法，包括结合自监督注意力模型SGAT和局部增强模型LA，构建基于改进图神经网络LA_SGAT的漏洞检测方法；收集源代码数据集并进行预处理，提取源代码的节点特征；通过局部增强模型LA，对节点特征进行局部增强，并与节点特征结合，获得第一输出特征；基于节点的第一输出特征和邻接特征，通过自监督注意力模型SGAT进行训练，获得第二输出特征；基于第二输出特征，通过金字塔池化、全连接层和Softmax分类层，得到源代码漏洞检测结果。因此，采用上述方法，能够全面、高效地捕获节点特征，提升漏洞检测的有效性。