公开(公告)号：CN120068084A

公开(公告)日：2025-05-30

申请号：CN202510070419.2

申请日：2025-01-16

Applicant: 大连海事大学

Inventor： 李辉 ,  张健 ,  李美琳 ,  雷雅彬 ,  郭世凯 ,  陈荣

IPC: G06F21/57 ,  G06N3/045 ,  G06F18/214 ,  G06F18/213 ,  G06F18/22 ,  G06F18/25

Abstract: 本发明提供了一种利用多重注意力机制进行代码漏洞修复检测的方法，包括如下步骤：S1、构建漏洞修复数据集；用CodeBERT对漏洞修复数据集中的代码更改提取嵌入，通过特征提取器获得代码更改的特征；S2、通过计算代码更改的特征之间的相似度，划分出补丁内容相关特征和补丁内容无关特征；S3、得到自注意力组件处理后的特征；S4、将自注意力组件处理后的补丁无关和补丁相关特征映射到同一纬度，输入到匹配注意力组件中进行联合计算，获得注意力权重，通过注意力权重进行加权获得预测结果。本发明引入了多重注意力机制，通过自注意力机制捕获代码片段的全局依赖关系，通过匹配注意力机制分析代码修改前后的差异，进而提高了对漏洞修复的识别精度。