On the Performance of Generative Adversarial Network by Limiting Mode Collapse for Malware Detection Systems

Sensors (Basel). 2021 Dec 30;22(1):264. doi: 10.3390/s22010264.

Abstract

Generative adversarial network (GAN) has been regarded as a promising solution to many machine learning problems, and it comprises of a generator and discriminator, determining patterns and anomalies in the input data. However, GANs have several common failure modes. Typically, a mode collapse occurs when a GAN fails to fit the set optimizations and leads to several instabilities in the generative model, diminishing the capability to generate new content regardless of the dataset. In this paper, we study conditional limiter solutions for mode collapse for the Intrusion Detection System (IDS) Control Flow GAN (ICF-GAN) model. Specifically, the ICF-GAN's mode collapse instances are limited by a mini-batch method that significantly improves the model accuracy. Performance evaluation is conducted using numerical results obtained from experiments.

Keywords: algorithms; discriminator; generative adversarial networks; generator model; intrusion detection system; long short-term memory; machine learning; malware; mode collapse.

MeSH terms

  • Image Processing, Computer-Assisted*
  • Machine Learning
  • Neural Networks, Computer*
  • Research Design