A malicious code family classification method based on self-attention mechanism

Ru Zhang,Xinjian Zhao,Jiaqi Li,Song Zhang,Zhijie Shang
DOI: https://doi.org/10.1088/1742-6596/2010/1/012066
2021-09-01
Journal of Physics: Conference Series
Abstract:Abstract Malicious code families have become a major threat to network security. Many current methods convert malicious code into images and use deep learning to classify malicious code families. However, the family classification method based on deep learning incorporates the overall characteristics of the malicious code into the classification model, which may cause interference with redundant information in the malicious code. This paper proposes a malicious code family classification method based on a self-attention mechanism. When analysing a noisy data structure such as malicious code images, the attention mechanism is introduced to filter the interference information in the malicious code. The experimental results show that the classification accuracy rate of this method for malicious code families is 99.56%, and the recall rate is 98.06%. Rigorous theoretical analyses and numerous experiments prove our method is efficient and reliable.
What problem does this paper attempt to address?