arrow
返回

CosFormer: A Code Semantic-Aware Transformer for Vulnerability Detection

delete2026-01-30
delete0
PRE
AI
Q
Qianyue Wei
Z
Zheng Zhang
Q
Qiuping Yi
Z
Zongcheng Ji
H
Hongliang Liang
DOI:10.1109/tse.2026.3659630delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
基于深度学习的漏洞检测已取得显著进展,超越了传统的静态和动态分析方法。然而,现有方法包括图神经网络(GNNs)和基于Transformer的模型,仍难以完全捕捉复杂的代码语义。本文提出了一种新颖的代码语义感知Transformer,命名为CosFormer,专门用于漏洞检测。CosFormer引入两个关键组件:代码语义感知嵌入,增强词元和行级别的语义表示;以及空间依赖感知编码,整合控制流图(CFGs)和程序依赖图(PDGs)的结构依赖,引导注意力集中于与漏洞相关的代码。我们在四个基准数据集(包括一个真实数据集)上评估了CosFormer,并展示了其优越的性能。CosFormer在所有任务中均获得最高F1分数,优于最先进的GNN和Transformer模型以及大型语言模型(LLMs)。值得注意的是,CosFormer在泛化评估中实现Cross F1 69.37和Mixed F1 58.42,超越所有基线。这些结果突显了CosFormer在检测不同且未见代码库中的漏洞时的有效性和鲁棒性。
Keyword:
Vulnerability Detection
transformer-based models
deep learning for code
code representation learning

期刊

IEEE Transactions on Software Engineering 封面图
IEEE Transactions on Software Engineering
IF:
5.6
论文数:
2.9K
被引数:
1.1W

机构

P
pingan technology
学者数:
2
论文数: 1
被引数: 0
B
beijing university of posts and telecommunications
学者数:
2.3K
论文数: 861
被引数: 0
引用论文

引用论文

err分享
err收藏
学者 查看更多内容