Multi-loss Siamese neural network with batch normalization layer for malware detection

Jinting Zhu*, Julian Jang-Jaccard, Paul A. Watters

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

4 Citations (Scopus)
6 Downloads (Pure)

Abstract

Malware detection is an essential task in cyber security. As the trend of malicious attacks grows, unknown malware detection with high accuracy becomes more and more challenging. The current deep learning-based approaches for malware detection are typically trained with large amounts of samples using labeled and existing malware families in the training set, thus, their capability to detect new unseen malware (such as a zero-day attack) is limited. To address this issue, we propose a new one-shot model called ‘‘Multi-Loss Siamese Neural Network with Batch Normalization Layer’’ that can work with fewer samples while providing high detection accuracy. Our model utilizes the Siamese Neural Network to detect new variants of malware that is trained with only a few samples. Our model is equipped with batch normalization and multiple loss functions to address the overfitting issue, due to the use of small samples, that can create the vanishing gradient problem as a result of binary cross-entropy loss, and feature embedding space to improve the detection accuracy. In addition, we illustrate a way to convert raw binary files into malware gray scale images, to work with the popular Siamese Neural Network by generating the positive and negative pairs for training. Our experimental results show that our model outperforms existing similar methods.

Original languageEnglish
Pages (from-to)171542-171550
Number of pages9
JournalIEEE Access
Volume8
DOIs
Publication statusPublished - 2020

Bibliographical note

Version archived for private and non-commercial use with the permission of the author/s and according to publisher conditions. For further rights please contact the publisher.

Keywords

  • Feature embedding space
  • Malware detection
  • Siamese neural network (SNN)
  • Vanishing gradient problem
  • Zero-day attack

Fingerprint

Dive into the research topics of 'Multi-loss Siamese neural network with batch normalization layer for malware detection'. Together they form a unique fingerprint.

Cite this