摘要
本发明公开了一种通过像素行过采样增强的代码图像漏洞检测方法,包括以下步骤:步骤1、提取函数样本并进行预处理;步骤2、生成代码属性图以表示函数样本,并基于节点关系生成代码映射图;步骤3、构建过采样代码列表及对应的中心性指标列表;步骤4、生成包含三通道的代码图像;步骤5、进行数据集划分;步骤6、将生成的三通道代码图像输入卷积神经网络进行训练,得到漏洞检测模型。本发明通过采用基于像素行过采样的图像增强技术,有效解决了代码图像在卷积池化过程中关键信息丢失的问题,显著提升了基于代码图像的漏洞检测精度;本发明方法能够增强代码行间的上下文信息,提高漏洞特征的提取效果,具备较强的漏洞检测能力与较低的计算开销。
技术关键词
漏洞检测方法
生成代码
像素
列表
三通道
指标
卷积神经网络模型
图像增强技术
样本
抽象语法树
漏洞特征
源程序
元素
节点数
数据
关系