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 条
  • [1] Auto-encoder generative adversarial networks
    Zhai, Zhonghua
    JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2018, 35 (03) : 3043 - 3049
  • [2] Network Intrusion Detection Based on Supervised Adversarial Variational Auto-Encoder With Regularization
    Yang, Yanqing
    Zheng, Kangfeng
    Wu, Bin
    Yang, Yixian
    Wang, Xiujuan
    IEEE ACCESS, 2020, 8 : 42169 - 42184
  • [3] Unregularized Auto-Encoder with Generative Adversarial Networks for Image Generation
    Wang, Jiayu
    Zhou, Wengang
    Tang, Jinhui
    Fu, Zhongqian
    Tian, Qi
    Li, Houqiang
    PROCEEDINGS OF THE 2018 ACM MULTIMEDIA CONFERENCE (MM'18), 2018, : 709 - 717
  • [4] Self-Attention-Based Conditional Variational Auto-Encoder Generative Adversarial Networks for Hyperspectral Classification
    Chen, Zhitao
    Tong, Lei
    Qian, Bin
    Yu, Jing
    Xiao, Chuangbai
    REMOTE SENSING, 2021, 13 (16)
  • [5] Twin Variational Auto-Encoder for Representation Learning in IoT Intrusion Detection
    Phai Vu Dinh
    Nguyen Quang Uy
    Nguyen, Diep N.
    Dinh Thai Hoang
    Son Pham Bao
    Dutkiewicz, Eryk
    2022 IEEE WIRELESS COMMUNICATIONS AND NETWORKING CONFERENCE (WCNC), 2022, : 848 - 853
  • [6] Constrained Twin Variational Auto-Encoder for Intrusion Detection in IoT Systems
    Dinh, Phai Vu
    Nguyen, Quang Uy
    Hoang, Dinh Thai
    Nguyen, Diep N.
    Bao, Son Pham
    Dutkiewicz, Eryk
    IEEE INTERNET OF THINGS JOURNAL, 2024, 11 (08): : 14789 - 14803
  • [7] A Lightweight Unsupervised Intrusion Detection Model Based on Variational Auto-Encoder
    Ren, Yi
    Feng, Kanghui
    Hu, Fei
    Chen, Liangyin
    Chen, Yanru
    SENSORS, 2023, 23 (20)
  • [8] Coupled generative adversarial stacked Auto-encoder: CoGASA
    Kiasari, Mohammad Ahangar
    Moirangthem, Dennis Singh
    Lee, Minho
    NEURAL NETWORKS, 2018, 100 : 1 - 9
  • [9] Effective fuzzing testcase generation based on variational auto-encoder generative adversarial network
    Qin, Zhongyuan
    Fan, Jiarong
    Liu, Xujian
    Li, Zeru
    Sun, Xin
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2025, 144
  • [10] A Novel Fault Detection Method Based on Adversarial Auto-Encoder
    Wang Jian
    Han Zhiyan
    PROCEEDINGS OF THE 39TH CHINESE CONTROL CONFERENCE, 2020, : 4166 - 4170