Affiliation:
1. Department of Computer Science, National Defense Academy of Japan, Yokosuka 239-8686, Japan
Abstract
Malware is becoming an effective support tool not only for professional hackers but also for amateur ones. Due to the support of free malware generators, anyone can easily create various types of malicious code. The increasing amount of novel malware is a daily global problem. Current machine learning-based methods, especially image-based malware classification approaches, are attracting significant attention because of their accuracy and computational cost. Convolutional Neural Networks are widely applied in malware classification; however, CNN needs a deep architecture and GPUs for parallel processing to achieve high performance. By contrast, a simple model merely contained a Multilayer Perceptron called MLP-mixer with fewer hyperparameters that can run in various environments without GPUs and is not too far behind CNN in terms of performance. In this study, we try applying an Autoencoder (AE) to improve the performance of the MLP-mixer. AE is widely used in several applications as dimensionality reduction to filter out the noise and identify crucial elements of the input data. Taking this advantage from AE, we propose a lightweight ensemble architecture by combining a customizer MLP-mixer and Autoencoder to refine features extracted from the MLP-mixer with the encoder-decoder architecture of the autoencoder. We achieve overperformance through various experiments compared to other cutting-edge techniques using Malimg and Malheur datasets which contain 9939 (25 malware families) and 3133 variant samples (24 malware families).
Reference44 articles.
1. (2022, December 14). Malware Attacks Targeting Ukraine Government-Microsoft on the Issues. Available online: https://blogs.microsoft.com/on-the-issues/2022/01/15/mstic-malware-cyberattacks-ukraine-government/.
2. (2022, December 14). Malware Statistics & Trends Report|AV-TEST. Available online: https://www.av-test.org/en/statistics/malware.
3. Raghuraman, C., Suresh, S., Shivshankar, S., and Chapaneri, R. (2019, January 29–30). Static and dynamic malware analysis using machine learning. Proceedings of the First International Conference on Sustainable Technologies for Computational Intelligence, Jaipur, India.
4. A survey on malware detection using data mining techniques;Ye;ACM Comput. Surv. (CSUR),2017
5. A chi-square-based decision for real-time malware detection using PE-file features;Belaoued;J. Inf. Process. Syst.,2016
Cited by
1 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
1. ZSL-SLCNN: Zero-Shot Learning with Semantic Label CNN for Malware Classification;2023 12th International Conference on Control, Automation and Information Sciences (ICCAIS);2023-11-27