Malware Classification using a Hybrid Hidden Markov Model-Convolutional Neural Network

Ritik Mehta, Olha Jureckova, Mark Stamp
12/25/2024
cs.LG

No confident code implementation found for this paper. 1 possible match can be shown below.

Abstract

The proliferation of malware variants poses a significant challenges to traditional malware detection approaches, such as signature-based methods, necessitating the development of advanced machine learning techniques. In this research, we present a novel approach based on a hybrid architecture combining features extracted using a Hidden Markov Model (HMM), with a Convolutional Neural Network (CNN) then used for malware classification. Inspired by the strong results in previous work using an HMM-Random Forest model, we propose integrating HMMs, which serve to capture sequential patterns in opcode sequences, with CNNs, which are adept at extracting hierarchical features. We demonstrate the effectiveness of our approach on the popular Malicia dataset, and we obtain superior performance, as compared to other machine learning methods -- our results surpass the aforementioned HMM-Random Forest model. Our findings underscore the potential of hybrid HMM-CNN architectures in bolstering malware classification capabilities, offering several promising avenues for further research in the field of cybersecurity.

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. 1 weaker match is 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{mehta2024malware,
  title  = {Malware Classification using a Hybrid Hidden Markov Model-Convolutional Neural Network},
  author = {Ritik Mehta and Olha Jureckova and Mark Stamp},
  year   = {2024},
  eprint = {2412.18932},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.18932v1}
}

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

Discussion