CNN-LSTM and Transfer Learning Models for Malware Classification based on Opcodes and API Calls

Ahmed Bensaoud, Jugal Kalita
5/4/2024
cs.CRcs.AIcs.LG

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

Abstract

In this paper, we propose a novel model for a malware classification system based on Application Programming Interface (API) calls and opcodes, to improve classification accuracy. This system uses a novel design of combined Convolutional Neural Network and Long Short-Term Memory. We extract opcode sequences and API Calls from Windows malware samples for classification. We transform these features into N-grams (N = 2, 3, and 10)-gram sequences. Our experiments on a dataset of 9,749,57 samples produce high accuracy of 99.91% using the 8-gram sequences. Our method significantly improves the malware classification performance when using a wide range of recent deep learning architectures, leading to state-of-the-art performance. In particular, we experiment with ConvNeXt-T, ConvNeXt-S, RegNetY-4GF, RegNetY-8GF, RegNetY-12GF, EfficientNetV2, Sequencer2D-L, Swin-T, ViT-G/14, ViT-Ti, ViT-S, VIT-B, VIT-L, and MaxViT-B. Among these architectures, Swin-T and Sequencer2D-L architectures achieved high accuracies of 99.82% and 99.70%, respectively, comparable to our CNN-LSTM architecture although not surpassing it.

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. 4 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

Cite this paper

@article{bensaoud2024cnnlstm,
  title  = {CNN-LSTM and Transfer Learning Models for Malware Classification based on Opcodes and API Calls},
  author = {Ahmed Bensaoud and Jugal Kalita},
  year   = {2024},
  eprint = {2405.02548},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.02548v1}
}

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.02548.svg)](https://codeofpaper.com/paper/2405.02548)

Discussion