一种基于多关系图网络的漏洞模型的构建方法及检测方法

摘要

本发明公开了一种基于多关系图网络的漏洞模型的构建方法及检测方法，收集源码程序文件，并对源码程序文件进行预处理得到函数块，根据函数块对应的函数名中是否包含good和bad，对应标签为无漏洞的函数块和有漏洞的函数块；对所述的无漏洞的函数块和有漏洞的函数块处理得到各自对应的第二抽象语法树；对第二抽象语法树进行第一遍历得到7种边关系；本发明在现有技术中提取抽象语法树节点信息的基础上，增加了7种边关系，7种边关系能在漏洞检测方面显著的提升准确率，并且降低漏报率；在搭建GGNN漏洞模型时，融入GRU和highway gate来提高向量在不同关系图之间的传播速度，进而缩短模型训练的周期，解决了现有技术中漏洞检测模型检测效果不佳的技术问题。