Malicious Software Detection Based on Improved Convolution Neural Network

Tianyue Liu,Hongqi Zhang,Haixia Long
DOI: https://doi.org/10.1109/icfeict57213.2022.00065
2022-01-01
Abstract:Malicious software detection is the initial and primary step for any network-based security system. Machine learning algorithm has become one of the main methods. Due to the high computational complexity and low computational accuracy of traditional machine learning algorithms, an improved deep learning method is proposed in this paper, Convolution Neural Network (CNN) with Batch Normalization and Inception-Residual network (BIR-CNN). Firstly, a new method is designed to feature extraction. Each vector value of each sample is aggregated, using the functions mean, median, standard deviation, skew, and kurtosis, resulting in a 5-dimension feature for each vector. Secondly, the supervised neural BIR-CNN method is adopted to detect malicious software. Convolution layers and an Inception-Residual network are used in the model to improve its capacity for learning. Batch normalization helps hasten model training while preventing over-fitting. Finally, research is done using network traffic that is accessible to the general public data set CICAndMal2017 and evaluated against three conventional algorithms for CNN and algorithms, and the BIR-CNN model obtains an accuracy of 0.97 and the AUC (Area Under Curve) of 0.99. Therefore, utilizing deep learning for the detection of dangerous software, BIR-CNN can be employed.
What problem does this paper attempt to address?