• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2025, Vol. 47 ›› Issue (2): 256-264.

• Computer Network and Znformation Security • Previous Articles     Next Articles

Research on intrusion detection method based on SAE and WGAN

LIU Yongmin1,2,XU Cheng1,2,HUANG Hao1,2,ZHANG Qianlei1,2,ZHAO Junjie1,2   

  1. (1.School of Electronic Information and Physics,Central South University of Forestry & Technology,Changsha 410004;
    2.Research Center of Smart Forest Cloud,Central South University of Forestry & Technology,Changsha 410004,China)
  • Received:2023-05-18 Revised:2023-12-09 Online:2025-02-25 Published:2025-02-24

Abstract: In recent years, the rapid development of technologies in the field of machine learning (ML) and deep learning (DL) has led to increasing research on their application in intrusion detection systems (IDS). However, current datasets in the field of intrusion detection face issues such as feature redundancy and an imbalance in the number of samples across different attack categories. To solve these problems, a network anomaly detecting method based on stacked autoencoder (SAE) and Wasserstein generative adversarial network (WGAN) is proposed. Firstly, to address the problem of feature redundancy, this paper employs the encoding-hidden layer-decoding concept of SAEs for data dimensionality reduction. This approach refines various features and extracts lower-dimensional features that are more suitable for classification. Secondly, to tackle the issue of sample imbalance (limited data volume and diversity), the processed data is used as input for the generator in the WGAN model. The generative capabilities of the generative adversarial network are utilized for sample augmentation, thereby compensating for the lack of certain types of samples during the training of the classification model. Finally, the random forest (RF) classification model is used for detection. Experimental results on NSL-KDD dataset show that SAE-WGAN-RF model which based on  the proposed method achieves an F1-Score of 95.58%, Recall of 96.54%, and Precision of 96.03%, representing significant improvements compared to common classical algorithms.

Key words: deep learning, generative adversarial networks, anomaly detection, stack autoencoder