View : 397 Download: 0
Stacked Autoencoder-Based Probabilistic Feature Extraction for On-Device Network Intrusion Detection
- Title
- Stacked Autoencoder-Based Probabilistic Feature Extraction for On-Device Network Intrusion Detection
- Authors
- Dao, Thi-Nga; Lee, HyungJune
- Ewha Authors
- 이형준
- SCOPUS Author ID
- 이형준
- Issue Date
- 2022
- Journal Title
- IEEE INTERNET OF THINGS JOURNAL
- ISSN
- 2327-4662
- Citation
- IEEE INTERNET OF THINGS JOURNAL vol. 9, no. 16, pp. 14438 - 14451
- Keywords
- Feature extraction; Neurons; Image edge detection; Network intrusion detection; Internet of Things; Computational modeling; Probabilistic logic; Anomaly classification; feature extraction; network intrusion detection system (NIDS); on-device AI
- Publisher
- IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
- Indexed
- SCIE; SCOPUS
- Document Type
- Article
- Abstract
- Due to the outbreak of recent network attacks, it is necessary to develop a robust network intrusion detection system (NIDS) that can quickly and effectively identify the network attack. Although the state-of-the-art detection algorithms have shown quite promising detection performance, they suffer from computationally intensive operations and large memory footprint, making themselves infeasible to applications at the resourceconstrained edge devices. We propose a lightweight yet effective NIDS scheme that incorporates a stacked autoencoder with a network pruning technique. By removing a set of ineffective neurons across layers in the autoencoder network with a certain probability based on their importance, a considerably large portion of relatively nominal training parameters are reduced. Then, the pruned and pretrained encoder network is used as-is and is connected with a separate classifier network for attack type inference, avoiding a full retraining from scratch. Experimental results indicate that our stacked autoencoder-based classification network with probabilistic feature extraction has outperformed the state-of-the-art NIDSs in terms of attack detection rate. Further, we have shown that our lightweight NIDS scheme has significantly reduced the computational complexity throughout the architecture, making it feasible to the edge, while maintaining a similar attack type detection quality compared with its original fully connected neural network.
- DOI
- 10.1109/JIOT.2021.3078292
- Appears in Collections:
- 인공지능대학 > 컴퓨터공학과 > Journal papers
- Files in This Item:
There are no files associated with this item.
- Export
- RIS (EndNote)
- XLS (Excel)
- XML