Efficient and Generalized Image-Based CNN Algorithm for Multi-Class Malware Detection

doi:10.1109/ACCESS.2024.3435362

2024-01-01

DOI：10.1109/ACCESS.2024.3435362

原文链接

原文求助

摘要

En 中文

With the popularity of electronic devices, the number of malware has increased dramatically, posing a serious threat to the digital world. Accurately identifying malware has become a research focus. However, there are many difficulties in the research, such as insufficient algorithm generalization ability, unbalanced datasets, and long processing and identification times. To address these problems, this study proposes a malware detection framework (VBDN) based on a convolutional neural network (CNN). The framework incorporates data visualization, balanced adoption, data augmentation, and convolutional neural network techniques to achieve over 90% accuracy in classifying malware on all four open-source datasets. The experimental work has two other contributions: first, it not only focuses on the overall recognition effect of the algorithm during the research process, but also on the recognition effect of each category with the help of a confusion matrix, which provides useful information for cybersecurity personnel, researchers, and others to carry out subsequent targeted research. Secondly, the balanced approach adopted in this paper has the following advantages: no need to construct a new dataset, consumes fewer hardware resources, automatically evaluates the sampling weights, etc. Additionally, to enhance the generalization ability of the algorithm and alleviate the overfitting problem, this paper employs data augmentation techniques to improve the adopted method. By comparing with several state-of-the-art algorithms, it can be observed that the VBDN framework proposed in this paper achieves the desired results in time with acceptable accuracy.

Keyword:

Malware

Feature extraction

Biological neural networks

Convolutional neural networks

Classification algorithms

Deep learning

Mathematical models

Data visualization

Malware detection

convolutional neural network (CNN)

data visualization