跳到主要导航 跳到搜索 跳到主要内容

Multi-view Pre-trained Model for Code Vulnerability Identification

  • East China Normal University
  • Guangdong University of Finance & Economics

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

Vulnerability identification is crucial for cyber security in the software-related industry. Early identification methods require significant manual efforts in crafting features or annotating vulnerable code. Although the recent pre-trained models alleviate this issue, they overlook the multiple rich structural information contained in the code itself. In this paper, we propose a novel Multi-View Pre-Trained Model (MV-PTM) that encodes both sequential and multi-type structural information of the source code and uses contrastive learning to enhance code representations. The experiments conducted on two public datasets demonstrate the superiority of MV-PTM. In particular, MV-PTM improves GraphCodeBERT by 3.36% on average in terms of F1 score.

源语言英语
主期刊名Wireless Algorithms, Systems, and Applications - 17th International Conference, WASA 2022, Proceedings
编辑Lei Wang, Michael Segal, Jenhui Chen, Tie Qiu
出版商Springer Science and Business Media Deutschland GmbH
127-135
页数9
ISBN(印刷版)9783031192104
DOI
出版状态已出版 - 2022
活动17th International Conference on Wireless Algorithms, Systems, and Applications, WASA 2022 - Dalian, 中国
期限: 24 11月 202226 11月 2022

出版系列

姓名Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
13473 LNCS
ISSN(印刷版)0302-9743
ISSN(电子版)1611-3349

会议

会议17th International Conference on Wireless Algorithms, Systems, and Applications, WASA 2022
国家/地区中国
Dalian
时期24/11/2226/11/22

指纹

探究 'Multi-view Pre-trained Model for Code Vulnerability Identification' 的科研主题。它们共同构成独一无二的指纹。

引用此