公开(公告)号：CN116910760A

公开(公告)日：2023-10-20

申请号：CN202310615838.0

申请日：2023-05-29

Applicant: 南通大学

Inventor： 陆桂龙 ,  鞠小林 ,  陈亮 ,  陈翔 ,  杨少宇 ,  裴文龙 ,  蔡之龙 ,  程伟

IPC: G06F21/57 ,  G06F16/33 ,  G06F16/35

Abstract: 本发明提供了一种使用CodeT5模型和提示微调的源代码漏洞检测方法，属于计算机技术领域，解决了传统的漏洞检测模型中检测准确率不高的技术问题。包括以下步骤：S1：从中收集开源项目的C和C++源代码，构成数据集；S2：对数据集进行预处理；S3：对构建的数据集随机划分成训练集、验证集和测试集；S4：使用hard soft方式创建为漏洞检测任务定制的提示模板；S5：创建为漏洞检测任务定制的Verbalizer；S6：训练集训练分类器；S7：对测试集进行最终结果预测。本发明的有益效果为：使用基于提示微调的预训练模型CodeT5进行漏洞检测，从而提高漏洞检测的性能。