Identify Vulnerability Fix Commits Automatically Using Hierarchical Attention Network

Mingxin Sun,Wenjie Wang,Hantao Feng,Hongu Sun,Yuqing Zhang
DOI: https://doi.org/10.4108/eai.13-7-2018.164552
2020-01-01
ICST Transactions on Security and Safety
Abstract:The application of machine learning and deep learning in the field of vulnerability detection is a hot topic in security research, but currently it faces the problem of lack of dataset. Considering vulnerable code can be obtained from vulnerability fix commits, we propose an automatic vulnerability commit identification tool based on hierarchical attention network (HAN) to expand existing vulnerability dataset. HAN can model the input data at the word and sentence levels respectively and pay attention to the changes in the characteristics of di ff erent words in di ff erent categories, which improves the classification performance. Experimental results show that the accuracy and F1 of our model both achieve 92%. Through the vulnerability fix commit, researchers can quickly locate the vulnerable code. And extracting vulnerable code from open-source software can e ff ectively expand the current dataset due to the enormous number of open-source software.
What problem does this paper attempt to address?