Aiming at the traditional malware classification using image with low accuracy rate, weak anti-aliasing ability and long time for model to reach convergence, this research makes improvements in the malware image representation method, converting malware, bytes Bigram, and Lst file to three kinds of grayscale images and combining grayscale images into three-channel color images for classification, and then uses the EfficientNet model with excellent image classification effects for malware image classification with the fine-tuning technology in the field of migration learning. The classification weights from the ImageNet dataset are applied to EfficientNet to improve the convergence speed and classification effect of the model, and to reduce the training cost of the model. Experiments show that the model converges faster than pre-training under the fine-tuning technology, and the best fine-tuned model can classify 20 kinds of malwares with an accuracy of 97.22%. The fine-tuned model has better classification accuracy and fewer FLOPs(floating point operations) and parameters than ResNet, VGG16 and other models.
National Internet Emergency Center. 2020 China Internet Cyber Security Report [R]. Beijing: National Internet Emergency Center, 2021(Ch). DOI: 10.29268/stbd.2016.3.1.3 .
[3]
ASLANÖ A, SAMETR. A comprehensive review on malware detection approaches[J]. IEEE Access, 2020, 8:6249-6271. DOI:10.1109/ACCESS.2019.2963724 .
[4]
NARM, KAKISIMA G, ÇARKACIN, et al. Analysis and comparison of opcode-based malware detection approaches [C]//2018 3rd International Conference on Computer Science and Engineering (UBMK). Piscataway:IEEE, 2018: 498-503. DOI:10.1109/UBMK.2018.8566292 .
[5]
FUJ W, XUEJ F, WANGY, et al. Malware visualization for fine-grained classification [J]. IEEE Access, 2018, 6:14510-14523. DOI:10.1109/ACCESS.2018.2805301 .
WANGB, CAIH H, SUY. Classification of malicious code variants based on VGGNet [J]. Journal of Computer Applications, 2020, 40(1):162-167. DOI: 10.11772/j.issn.1001-9081.2019050953(Ch ).
[12]
NATARAJL, KARTHIKEYANS, JACOBG, et al. Malware images: Visualization and automatic classification[C]//Proceedings of the 8th International Symposium on Visualization for Cyber Security. New York: ACM Press, 2011:1-7. DOI: 10.1145/2016904.2016908 .
[13]
GHAZIM M, YANIKOGLUB, APTOULAE. Plant identification using deep neural networks via optimization of transfer learning parameters [J]. Neurocomputing, 2017, 235:228-235. DOI:10.1016/j.neucom.2017.01.018 .
[14]
VASAND, ALAZABM, WASSANS, et al. Image-based malware classification using ensemble of CNN architectures (IMCEC) [J]. Computers & Security, 2020, 92:101748. DOI:10.1016/j.cose.2020.101748 .
[15]
XIEQ Z, LUONGM T, HOVYE, et al. Self-training with noisy student improves ImageNet classification [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway:IEEE, 2020:10687-10698. DOI: 10.1109/cvpr42600.2020.01070 .
[16]
TANM X, LEQ. EfficientNet: Rethinking model scaling for convolutional neural networks[C]//International Conference on Machine Learning. Cambridge:JMLR, 2019:6105-6114.
[17]
ALHICHRIH, ALSWAYEDA S, BAZIY, et al. Classification of remote sensing images using EfficientNet-B3 CNN model with attention[J]. IEEE Access, 2021, 9:14078-14094. DOI:10.1109/ACCESS.2021.3051085 .
[18]
ATILAÜ, UÇARM, AKYOLK, et al. Plant leaf disease classification using EfficientNet deep learning model[J]. Ecological Informatics, 2021, 61:101182. DOI:10.1016/j.ecoinf.2020.101182 .
[19]
WANGJ, LIUQ Y, XIEH T, et al. Boosted EfficientNet: Detection of lymph node metastases in breast cancer using convolutional neural networks [J]. Cancers, 2021, 13(4):661. DOI:10.3390/cancers13040661 .
[20]
TANM X, CHENB, PANGR M, et al. MnasNet: Platform-aware neural architecture search for mobile [C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway:IEEE,2019:2820-2828. DOI: 10.1109/cvpr.2019.00293 .
[21]
HEK, ZHANGX, RENS, et al. Deep residual learning for image recognition [C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Piscataway:IEEE,2016: 770-778. DOI: 10.1109/cvpr.2016.90 .
[22]
CHOLLETF. Xception: Deep learning with depthwise separable convolutions[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Piscataway:IEEE,2017: 1251-1258. DOI: 10.1109/cvpr.2017.195 .
[23]
SZEGEDYC, VANHOUCKEV, IOFFES, et al. Rethinking the inception architecture for computer vision [C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Piscataway:IEEE, 2016: 2818-2826. DOI: 10.1109/cvpr.2016.308 .
[24]
HOWARDA G, ZHUM L, CHENB, et al. Mobilenets: Efficient convolutional neural networks for mobile vision applications [DB/OL]. [2021-05-06]. DOI: 10.1109/cvpr.2018.00286 .
[25]
SIMONYANK, ZISSERMANA. Very deep convolutional networks for large-scale image recognition [DB/OL]. [2021-05-06]. DOI: 10.5244/c.28.6 .