DeepGray: Malware Classification Using Grayscale Images with Deep Learning


  • Haodi Jiang Sam Houston State University
  • Harshitha Polsani Sam Houston State University
  • Yuexin Liu Texas A&M University



In the ever-evolving landscape of cybersecurity, the threat posed by malware continues to loom large, necessitating innovative and robust approaches for its effective detection and classification. In this paper, we introduce a novel method, DeepGray, for multi-class malware classification utilizing malware images and the power of deep learning. Our dataset combines the malware sample from the BODMAS dataset and the benign sample from the DikeDataset. The methodology involves transforming executable files into a deep learning-friendly format by converting them into grayscale images while preserving essential data characteristics. Subsequently, Principal Component Analysis (PCA) is applied to distill the most significant features. The study harnesses the power of deep learning and transfer learning, utilizing established neural network architectures such as VGG16, InceptionV3, Efficientnetv2b0, and Vision Transformers (ViT) for malware classification. Experimental results demonstrate the effectiveness of the proposed method in accurately classifying malware.




How to Cite

Jiang, H., Polsani, H., & Liu, Y. (2024). DeepGray: Malware Classification Using Grayscale Images with Deep Learning. The International FLAIRS Conference Proceedings, 37(1).