一种基于多层次特征深度提取的漏洞代码检测方法

技术特征：

1.一种基于多层次特征深度提取的漏洞代码检测方法，其特征在于：包括以下步骤：

2.根据权利要求1所述的基于多层次特征深度提取的漏洞代码检测方法，其特征在于：所述步骤s1中获取漏洞代码样本数据集的具体方法为：

3.根据权利要求1所述的基于多层次特征深度提取的漏洞代码检测方法，其特征在于：所述步骤s2中对漏洞代码训练样本进行切片处理和数据清洗的方法为：

4.根据权利要求1所述的基于多层次特征深度提取的漏洞代码检测方法，其特征在于：所述步骤s2中通过三个嵌入层转换为嵌入向量的具体步骤为：

5.根据权利要求1所述的基于多层次特征深度提取的漏洞代码检测方法，其特征在于：所述步骤s5的详细过程为；

6.根据权利要求1所述的基于多层次特征深度提取的漏洞代码检测方法，其特征在于：所述步骤s6结合函数级代码特征以及词级代码特征进行漏洞代码分类的具体方法为：

技术总结
本发明公开一种基于多层次特征深度提取的漏洞代码检测方法，采用代码切片的技术，利用Joern平台针对某些与漏洞相关的库函数或API函数通过数据流和控制流分析提取出节点间的具有数据依赖和控制依赖关系的切片，然后对切片进行数据清洗；对代码切片通过标记嵌入、片段嵌入、位置嵌入，将输入转化为可使用BERT模型进行特征提取的嵌入向量；通过对BERT模型的输出结果进行分类讨论，分析出基于函数级别、词级别不同层次的漏洞代码特征；对词级别特征通过LSTM与Multi‑Head Attention的进一步处理，最后与函数级别特征进行结合，进而从多层次对漏洞代码进行分析检测。本发明极大提高了漏洞代码检测的准确率与速度。

技术研发人员：朱二周,李智超
受保护的技术使用者：安徽大学
技术研发日：
技术公布日：2024/1/12