arrow
Return

SeqTrans: Automatic Vulnerability Fix Via Sequence to Sequence Learning

delete2023-02-01
delete21
delete
OA
AI
J
Jianlei Chi *
Y
Yu Qu
刘烃 cover
刘烃 (Ting Liu)
Q
Qinghua Zheng
H
Heng Yin
DOI:10.1109/TSE.2022.3156637delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Software vulnerabilities are now reported unprecedentedly due to the recent development of automated vulnerability hunting tools. However, fixing vulnerabilities still mainly depends on programmers' manual efforts. Developers need to deeply understand the vulnerability and affect the system's functions as little as possible. In this paper, with the advancement of Neural Machine Translation (NMT) techniques, we provide a novel approach called SeqTrans to exploit historical vulnerability fixes to provide suggestions and automatically fix the source code. To capture the contextual information around the vulnerable code, we propose to leverage data-flow dependencies to construct code sequences and feed them into the state-of-the-art transformer model. The fine-tuning strategy has been introduced to overcome the small sample size problem. We evaluate SeqTrans on a dataset containing 1,282 commits that fix 624 CVEs in 205 Java projects. Results show that the accuracy of SeqTrans outperforms the latest techniques and achieves 23.3% in statement-level fix and 25.3% in CVE-level fix. In the meantime, we look deep inside the result and observe that the NMT model performs very well in certain kinds of vulnerabilities like CWE-287 (Improper Authentication) and CWE-863 (Incorrect Authorization).
Keywords:
Maintenance engineering
Codes
Computer bugs
Predictive models
Transformers
Decoding
Training
Machine learning
neural machine translation
software engineering
vulnerability fix

Journal

IEEE Transactions on Software Engineering cover
IEEE Transactions on Software Engineering
IF:
5.6
Papers:
2.8K
Citations:
1.1W

Organization

X
xi'an jiaotong university
Scholars:
9.2W
Papers: 6.6W
Citations: 75
University of California System cover
University of California System
Scholars:
37.5W
Papers: 33.7W
Citations: 6.6K