High-resolution Image-based Malware Classification using Multiple Instance Learning

2023-11-21 · Tim Peters, Hikmat Farhat

This paper proposes a novel method of classifying malware into families using high-resolution greyscale images and multiple instance learning to overcome adversarial binary enlargement. Current methods of visualisation-based malware classification largely rely on lossy transformations of inputs such as resizing to handle the large, variable-sized images. Through empirical analysis and experimentation, it is shown that these approaches cause crucial information loss that can be exploited. The proposed solution divides the images into patches and uses embedding-based multiple instance learning with a convolutional neural network and an attention aggregation function for classification. The implementation is evaluated on the Microsoft Malware Classification dataset and achieves accuracies of up to $96.6\%$ on adversarially enlarged samples compared to the baseline of $22.8\%$. The Python code is available online at https://github.com/timppeters/MIL-Malware-Images .

📄 PDF Abstract BibTeX arXiv:2311.12760

Code (1)

timppeters/mil-malware-images 공식 구현 pytorch

Tasks

Classification Malware Classification Multiple Instance Learning

Malware Classification Using Transfer Learning

2021-07-29 · Hikmat Farhat, Veronica Rammouz

With the rapid growth of the number of devices on the Internet, malware poses a threat not only to the affected devices but also their ability to use said devices to launch attacks on the Internet ecosystem. Rapid malwar…

Classification Deep Learning Malware Classification Transfer Learning

BinImg2Vec: Augmenting Malware Binary Image Classification with Data2Vec

2022-09-02 · Joon Sern Lee, Kai Keng Tay, Zong Fu Chua

Rapid digitalisation spurred by the Covid-19 pandemic has resulted in more cyber crime. Malware-as-a-service is now a booming business for cyber criminals. With the surge in malware activities, it is vital for cyber defe…

Classification image-classification Image Classification Self-Supervised Learning

Orthrus: A Bimodal Learning Architecture for Malware Classification

2020-09-28 · Daniel Gibert, Carles Mateu, Jordi Planes

Malware detection and classification is a challenging problem and an active area of research. Traditional machine learning methods depend almost entirely on the ability to extract a set of discriminative features into wh…

Classification Deep Learning Feature Engineering General Classification+2

Exploring Optimal Deep Learning Models for Image-based Malware Variant Classification

2020-04-10 · Rikima Mitsuhashi, Takahiro Shinagawa

Analyzing a huge amount of malware is a major burden for security analysts. Since emerging malware is often a variant of existing malware, automatically classifying malware into known families greatly reduces a part of t…

Classification Computer Security Deep Learning General Classification+3

Malware Images:Visualization and Automatic Classification

2011-07-20 · Proceedings of the 8th International Symposium on Visualization for Cyber Security 2011 7 · Nataraj L., Karthikeyan S., Jacob G., Manjunath B. S.

We propose a simple yet effective method for visualizing and classifying malware using image processing techniques. Malware binaries are visualized as gray-scale images, with the observation that for many malware familie…

Classification