Self-attention based convolutional-LSTM for android malware detection using network traffics grayscale image

Limin Shen,Jiayin Feng,Zhen Chen,Zhongkui Sun,Dongkui Liang,Hui Li,Yuying Wang
DOI: https://doi.org/10.1007/s10489-022-03523-2
IF: 5.3
2022-04-20
Applied Intelligence
Abstract:To accurately find malware in a large number of mobile APPs, and determine which family it belongs to is one of the most important challenges in Android malware detection. Existed research focuses on using the extracted features to distinguish Android malicious APPs, and less attention is paid to the category and family classification of Android malware. Meanwhile, feature selection has always been a choose-difficult issue in malware detection with machine learning methods. In this paper, SelAttConvLstm was designed to classify android malware by category and family without manually selecting features. To identify Android malware, we first convert all the network traffic flows into grayscale images according to chronological order through data preprocessing. Second, we design SelAttConvLstm, a deep learning model to detect malicious Android APPs with network flows images. This model can consider both the spatial and temporal features of network flow at the same time. In addition, to improve the performance of the model, self-attention weights are added to focus on different features of the input. Finally, comprehensive experiments are conducted to verify the effectiveness of the detection model. Experimental results showed that our method can not only effectively detect malware, but also classify malware in detail and accurately by category and family.
computer science, artificial intelligence
What problem does this paper attempt to address?