一种基于注意力增强图匹配神经网络的同源漏洞检测方法
申请号:CN202410898761
申请日期:2024-07-05
公开号:CN118779885A
公开日期:2024-10-15
类型:发明专利
摘要
本发明公开一种基于注意力增强图的二进制同源漏洞检测方法及装置。方法包括:由规范的汇编代码提取操作码类型和操作数类型数据,将汇编代码中所有操作码和操作数类型的标签按照地址顺序组成指令类型序列;基于预训练后的Bert模型提取指令类型序列的指令语义特征;将目标函数的ACFG和漏洞函数的ACFG映射到嵌入空间中,分别生成目标图嵌入和漏洞图嵌入,通过余弦相似度来计算目标图嵌入和漏洞图嵌入之间的相似度。本发明结合自然语言处理模型和图神经网络,能够实现高精度的同源漏洞检测。
技术关键词
漏洞检测方法
语义特征
指令
序列
漏洞检测装置
多层感知机
标签
节点特征
注意力机制
自然语言
模块
数据
编码