Title
An Attention Mechanism for Combination of CNN and VAE for Image-Based Malware Classification
Abstract
Currently, malware is increasing in both number and complexity dramatically. Several techniques and methodologies have been proposed to detect and neutralize malicious software. However, traditional methods based on the signatures or behaviors of malware often require considerable computational time and resources for feature engineering. Recent studies have applied machine learning to the problems of identifying and classifying malware families. Combining many state-of-the-art techniques has become popular but choosing the appropriate combination with high efficiency is still a problem. The classification performance has been significantly improved using complex neural network architectures. However, the more complex the network, the more resources it requires. This paper proposes a novel lightweight architecture by combining small Convolutional Neural Networks and advanced Variational Autoencoder, enhanced by channel and spatial attention mechanisms. We achieve overperformance and sufficient time through various experiments compared to other cutting-edge techniques using unbalanced and balanced Malimg datasets.
Year
DOI
Venue
2022
10.1109/ACCESS.2022.3198072
IEEE ACCESS
Keywords
DocType
Volume
Malware, Feature extraction, Computer architecture, Convolutional neural networks, Computational modeling, Codes, Behavioral sciences, Encoding, Information security, Channel capacity, Malware classification, variational autoencoder, channel attention, spatial attention, latent representation, information security
Journal
10
ISSN
Citations 
PageRank 
2169-3536
0
0.34
References 
Authors
0
3
Name
Order
Citations
PageRank
Tuan Van Dao100.34
Hiroshi Sato200.34
Masao Kubo300.34