An intrusion detection method combining variational auto-encoder and generative adversarial networks

被引:1
|
作者
Li, Zhengfa [1 ]
Huang, Chuanhe [1 ]
Qiu, Wanyu [1 ]
机构
[1] Wuhan Univ, Sch Comp Sci, Wuhan, Peoples R China
关键词
Network intrusion detection; Network security; GAN; VAE; NSL-KDD; DEEP LEARNING APPROACH; SMOTE; IOT;
D O I
10.1016/j.comnet.2024.110724
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Deep learning is a crucial research area in network security, particularly when it comes to detecting network attacks. While some deep learning algorithms have shown promising results in distinguishing between normal and abnormal traffic, identifying different types of imbalanced anomalous traffic data is still a challenging task at present. To enhance the detection performance of unbalanced anomalous flows, we propose a new intrusion detection architecture based on a variational auto-encoder (VAE) and generative adversarial networks (GAN) in this research. Firstly, we present the VAE-WGAN model, which combines the advantages of VAE and GAN and enables us to generate data with predefined labels to balance the original training dataset. In the intrusion detection phase, we use a hybrid neural network model based on stacked Long Short-Term Memory (LSTM) and Multi-Scale Convolutional Neural Network (MSCNN). Stacked LSTM and MSCNN networks can extract network characteristics at different depths and scales, and subsequent feature fusion is used to increase network attack detection rates. Finally, the results from the NSL-KDD and AWID datasets indicate that the proposed network intrusion detection model improves the accuracy of network attack detection. The model outperforms other existing intrusion detection approaches in terms of accuracy, precision, recall, and f1-score, obtaining 83.45% accuracy and 83.69% f1-score on the NSL-KDD dataset. Moreover, it attains an accuracy and f1-score exceeding 98.9% on the AWID dataset.
引用
收藏
页数:16
相关论文
共 50 条
  • [21] Exploiting the Auto-Encoder Residual Error for Intrusion Detection
    Andresini, Giuseppina
    Appice, Annalisa
    Di Mauro, Nicola
    Loglisci, Corrado
    Malerba, Donato
    2019 4TH IEEE EUROPEAN SYMPOSIUM ON SECURITY AND PRIVACY WORKSHOPS (EUROS&PW), 2019, : 281 - 290
  • [22] Hamiltonian Variational Auto-Encoder
    Caterini, Anthony L.
    Doucet, Arnaud
    Sejdinovic, Dino
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 31 (NIPS 2018), 2018, 31
  • [23] COMPOUND VARIATIONAL AUTO-ENCODER
    Su, Shang-Yu
    Lin, Shan-Wei
    Chen, Yun-Nung
    2019 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2019, : 3577 - 3581
  • [24] Network Intrusion Detection System Based on an Adversarial Auto-Encoder with Few Labeled Training Samples
    Kohei Shiomoto
    Journal of Network and Systems Management, 2023, 31
  • [25] Network Intrusion Detection System Based on an Adversarial Auto-Encoder with Few Labeled Training Samples
    Shiomoto, Kohei
    JOURNAL OF NETWORK AND SYSTEMS MANAGEMENT, 2023, 31 (01)
  • [26] Brain Lesion Synthesis via Progressive Adversarial Variational Auto-Encoder
    Huo, Jiayu
    Vakharia, Vejay
    Wu, Chengyuan
    Sharan, Ashwini
    Ko, Andrew
    Ourselin, Sebastien
    Sparks, Rachel
    SIMULATION AND SYNTHESIS IN MEDICAL IMAGING, SASHIMI 2022, 2022, 13570 : 101 - 111
  • [27] A METHOD FOR FACE FUSION BASED ON VARIATIONAL AUTO-ENCODER
    Li, Xiang
    Wen, Jin-Mei
    Chen, An-Long
    Chen, Bo
    2018 15TH INTERNATIONAL COMPUTER CONFERENCE ON WAVELET ACTIVE MEDIA TECHNOLOGY AND INFORMATION PROCESSING (ICCWAMTIP), 2018, : 77 - 80
  • [28] Imbalanced sample fault diagnosis of rotating machinery using conditional variational auto-encoder generative adversarial network
    Wang, You-ren
    Sun, Guo-dong
    Jin, Qi
    APPLIED SOFT COMPUTING, 2020, 92
  • [29] Research on the application of deep auto-encoder network in intrusion detection
    Ding H.
    Wan L.
    Long T.
    Harbin Gongye Daxue Xuebao/Journal of Harbin Institute of Technology, 2019, 51 (05): : 185 - 194
  • [30] Anomaly-based Intrusion Detection Using Auto-encoder
    Nguimbous, Yves Nsoga
    Ksantini, Riadh
    Bouhoula, Adel
    2019 27TH INTERNATIONAL CONFERENCE ON SOFTWARE, TELECOMMUNICATIONS AND COMPUTER NETWORKS (SOFTCOM), 2019, : 505 - 509