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.

Enhancing the Breast Histopathology Image Analysis for Cancer Detection Using Variational Autoencoder

“…Classification may be quite chal-lenging, especially when dealing with several classes, because of the variety within a category and the similarity across classes [1]. The limits of feature extraction techniques for histopathology pictures of breast cancer present the second difficulty [39]. Scale-Invariant Feature Transform (SIFT) and Gray-Level Co-Occurrence Matrix (GLCM), two common approaches for extracting features, all depend on supervised data.…”

FEdroid: Lightweight and Interpretable Detection of Android Malware Using Local Key Information and Feature Selection

“…Currently, many studies on Android malware detection do not perform feature selection [14][15][16][17]. However, feature selection offers several advantages in malware detection [18].…”