Enhanced DNNs for malware classification with GAN-based adversarial training

Yunchun Zhang,Haorui Li,Yang Zheng,Shaowen Yao,Jiaqi Jiang
DOI: https://doi.org/10.1007/s11416-021-00378-y
2021-01-01
Journal of Computer Virology and Hacking Techniques
Abstract:Deep learning based malware classification gains momentum recently. However, deep learning models are vulnerable to adversarial perturbation attacks especially when applied in network security application. Deep neural network (DNN)-based malware classifiers by eating the whole bit sequences are also vulnerable despite their satisfactory performance and less feature-engineering job. Therefore, this paper proposes a DNN-based malware classifier on the raw bit sequences of programs in Windows. We then propose two adversarial attacks targeting our trained DNNs to generate adversarial malware. A defensive mechanism is proposed by treating perturbations as noise added on bit sequences. In our defensive mechanism, a generative adversary network (GAN)-based model is designed to filter out the perturbation noise and those that with the highest probability to fool the target DNNs are chosen for adversarial training. The experiments show that GAN with filter-based model produced the highest quality adversarial samples with medium cost. The evasion ratio under GAN with filter-based model is as high as 50.64
What problem does this paper attempt to address?