From Data and Model Levels: Improve the Performance of Few-Shot Malware Classification

Chai, Y, Qiu, J, Yin, L, Zhang, L, Gupta, B B and Tian, Z (2022) From Data and Model Levels: Improve the Performance of Few-Shot Malware Classification. IEEE Transactions on Network and Service Management. ISSN 2373-7379

Full text not available from this repository. (Request a copy)


Existing malware classification methods cannot handle the open-ended growth of new or unknown malware well because it only focuses on pre-defined malware classes with sufficient training data. Due to the superiority of the visualization method, some researchers use it for solving few-shot malware classification. However, the malware images generated by existing visualization methods contain insufficient semantic information. At the same time, existing few-shot models tend to converge to sharp minima resulting in poor generalization performance. By synthesizing the observations, we think that accurate and effective few-shot malware classification methods are affected by generated malware images and classification models, which can be called data and model levels, respectively. To solve the above problems, we propose a novel method from the Data and Model levels, which is used to classify new or unknown malware well, called DMMal. More specifically, we propose a multi-channel malware image generation method based on multi-view so that malware images can contain more prosperous information at the data level. In addition, we investigated adaptive sharpness-aware minimization in a few-shot scenario from the perspective of model optimization at the model level to minimize the loss value and sharpness simultaneously. This enhances the generalization ability of the model and improves the ability of the model to classify new or unknown classes. Experiments on two few-shot malware classification datasets show that the method proposed can improve the performance of few-shot malware classification from the data and model levels.

Affiliation: Skyline University College
SUC Author(s): Gupta, B B
All Author(s): Chai, Y, Qiu, J, Yin, L, Zhang, L, Gupta, B B and Tian, Z
Item Type: Article
Uncontrolled Keywords: Cyber-security, Few-shot Malware Classification, Malware Visualization, Flat Minima,
Subjects: B Information Technology > BB Information Technology
Divisions: Skyline University College > School of IT
Depositing User: Mr Veeramani Rasu
Date Deposited: 08 Sep 2022 05:54
Last Modified: 08 Sep 2022 05:54
Publisher URL:
Publisher OA policy:
Related URLs:

Actions (login required)

View Item
View Item
Statistics for SkyRep ePrint 572 Statistics for this ePrint Item