Deep Multi-Task Learning for Malware Image Classification

Ahmed Bensaoud, Jugal Kalita
5/9/2024
cs.CRcs.CVcs.LG

No confident code implementation found for this paper. 9 possible matches can be shown below.

Abstract

Malicious software is a pernicious global problem. A novel multi-task learning framework is proposed in this paper for malware image classification for accurate and fast malware detection. We generate bitmap (BMP) and (PNG) images from malware features, which we feed to a deep learning classifier. Our state-of-the-art multi-task learning approach has been tested on a new dataset, for which we have collected approximately 100,000 benign and malicious PE, APK, Mach-o, and ELF examples. Experiments with seven tasks tested with 4 activation functions, ReLU, LeakyReLU, PReLU, and ELU separately demonstrate that PReLU gives the highest accuracy of more than 99.87% on all tasks. Our model can effectively detect a variety of obfuscation methods like packing, encryption, and instruction overlapping, strengthing the beneficial claims of our model, in addition to achieving the state-of-art methods in terms of accuracy.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 9 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

A Survey of Malware Detection Using Deep Learning

88% match2024

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

87% match1 repo72023

Accelerating Malware Classification: A Vision Transformer Solution

84% match1 repo12024

Machine Learning for Windows Malware Detection and Classification: Methods, Challenges and Ongoing Research

83% match2024

OpCode-Based Malware Classification Using Machine Learning and Deep Learning Techniques

83% match2025

A Comparison of Selected Image Transformation Techniques for Malware Classification

83% match2025

Cite this paper

@article{bensaoud2024deep,
  title  = {Deep Multi-Task Learning for Malware Image Classification},
  author = {Ahmed Bensaoud and Jugal Kalita},
  year   = {2024},
  eprint = {2405.05906},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.05906v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.05906.svg)](https://codeofpaper.com/paper/2405.05906)

Discussion