arrow
Return

PAFE: A lightweight visualization-based fast malware classification method

delete2024-08-01
delete0
delete
OA
AI
S
Sicong Li
J
Jian Wang
S
Shuo Wang
Y
Yafei Song *
DOI:10.1016/j.heliyon.2024.e35965delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
With the development of automated malware toolkits, cybersecurity faces evolving threats. Although visualization-based malware analysis has proven to be an effective method, existing approaches struggle with challenging malware samples due to alterations in the texture features of binary images during the visualization preprocessing stage, resulting in poor performance. Furthermore, to enhance classification accuracy, existing methods sacrifice prediction time by designing deeper neural network architectures. This paper proposes PAFE, a lightweight and visualization-based rapid malware classification method. It addresses the issue of texture feature variations in preprocessing through pixel-filling techniques and applies data augmentation to overcome the challenges of class imbalance in small sample datasets. PAFE combines multi-scale feature fusion and a channel attention mechanism, enhancing feature expression through modular design. Extensive experimental results demonstrate that PAFE outperforms the current state-ofthe-art methods in both efficiency and effectiveness for malware variant classification, achieving an accuracy rate of 99.25 % with a prediction time of 10.04 ms.
Keywords:
Malware visualization
Pixel padding
Data augmentation
Multi-scale feature fusion
Channel attention
Deep neural networks
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Heliyon cover
Heliyon
IF:
3.6
Papers:
3.8W
Citations:
10.5W

Organization

A
Air Force Engineering University
Scholars:
4.7K
Papers: 3.0K
Citations: 1.9K