摘要
本发明属于软件测试领域,具体涉及一种结合增强的图表示学习与transformer的漏洞检测方法、系统和装置。该方法构建包含预处理模块、特征提取模块和分类模块的漏洞检测模型。获取大量包含漏洞标签信息的源代码构成训练集和测试集,并利用包含聚焦损失的三元损失集对漏洞检测模块进行训练和测试;最后,利用经过测试的漏洞检测模型对源代码进行漏洞检测。其中,预处理模块根据源代码生成代码切片和切片子图。特征提取模块采用CodeBERT和EA‑GGNN模型进行特征提取并获得融合特征向量。分类模块采用MLP并基于融合特征向量生成源代码的漏洞检测结果。本发明解决了现有方法无法充分挖掘代码间依赖关系和全局信息,导致漏洞检测效率、精度和泛化性不足的问题。
技术关键词
漏洞检测方法
代码切片
特征提取模块
注意力机制
漏洞检测系统
生成代码
分支
序列
对源代码
节点依赖关系
漏洞检测装置
样本
前馈神经网络
表达式
语法特征
系统为您推荐了相关专利信息
坐标系
车辆
kalman滤波
道路安防技术
矩阵
智能修正方法
司机
梯度提升树模型
多头注意力机制
订单
入侵检测方法
注意力机制
入侵检测数据
BiLSTM模型
特征提取模块
内容提取方法
表格
构建知识图谱
BERT模型
更新模型参数
移动机器人
生成三维边界框
彩色图像
运动
深度图