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 条
  • [41] Generative Adversarial Network and Auto Encoder based Anomaly Detection in Distributed IoT Networks
    Tian Zixu
    Liyanage, Kushan Sudheera Kalupahana
    Gurusamy, Mohan
    2020 IEEE GLOBAL COMMUNICATIONS CONFERENCE (GLOBECOM), 2020,
  • [42] MOTION DETECTION VIA A COUPLE OF AUTO-ENCODER NETWORKS
    Xu, Pei
    Ye, Mao
    Liu, Qihe
    Li, Xudong
    Pei, Lishen
    Ding, Jian
    2014 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO (ICME), 2014,
  • [43] Similarity Constrained Conditional Generative Auto-encoder with Generalized Dilated Networks
    Reimann, Jan Niclas
    Shukla, Bhargav Bharat
    Schwung, Andreas
    Ding, Steven X.
    PATTERN RECOGNITION APPLICATIONS AND METHODS, ICPRAM 2021, ICPRAM 2022, 2023, 13822 : 23 - 51
  • [44] Intelligent Bearing Anomaly Detection for Industrial Internet of Things Based on Auto-Encoder Wasserstein Generative Adversarial Network
    Liu, Ruonan
    Xiao, Dong
    Lin, Di
    Zhang, Weidong
    IEEE INTERNET OF THINGS JOURNAL, 2024, 11 (13): : 22869 - 22879
  • [45] Unsupervised Intrusion Detection Based on Asymmetric Auto-Encoder Feature Extraction
    Liu, Chunbo
    Wang, Liyin
    Zhang, Zhikai
    Xiang, Chunmiao
    Gu, Zhaojun
    Wang, Zhi
    Wang, Shuang
    IEICE TRANSACTIONS ON INFORMATION AND SYSTEMS, 2024, E107D (09) : 1161 - 1173
  • [46] Intrusion detection of industrial control system based on stacked auto-encoder
    Zhang, Rui
    Chen, Hongwei
    2019 CHINESE AUTOMATION CONGRESS (CAC2019), 2019, : 5638 - 5643
  • [47] A two-stage intrusion detection system with auto-encoder and LSTMs
    Mushtaq, Earum
    Zameer, Aneela
    Umer, Muhammad
    Abbasi, Asima Akber
    APPLIED SOFT COMPUTING, 2022, 121
  • [48] AVAE: Adversarial Variational Auto Encoder
    Plumerault, Antoine
    Le Borgne, Herve
    Hudelot, Celine
    2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, : 8687 - 8694
  • [49] Exploring Variational Auto-encoder Architectures, Configurations, and Datasets for Generative Music Explainable AI
    Bryan-Kinns, Nick
    Zhang, Bingyuan
    Zhao, Songyan
    Banar, Berker
    MACHINE INTELLIGENCE RESEARCH, 2024, 21 (01) : 29 - 45
  • [50] Cloud Intrusion Detection Method Based on Stacked Contractive Auto-Encoder and Support Vector Machine
    Wang, Wenjuan
    Du, Xuehui
    Shan, Dibin
    Qin, Ruoxi
    Wang, Na
    IEEE TRANSACTIONS ON CLOUD COMPUTING, 2022, 10 (03) : 1634 - 1646