Twin Variational Auto-Encoder for Representation Learning in IoT Intrusion Detection

• Published in: 2022 IEEE Wireless Communications and Networking Conference (WCNC) pp. 848 - 853
• Main Authors: Dinh, Phai Vu, Quang Uy, Nguyen, Nguyen, Diep N., Thai Hoang, Dinh, Bao, Son Pham, Dutkiewicz, Eryk
• Format: Conference Proceeding
• Language: English
• Published: IEEE 10.04.2022
• Subjects: Botnet; Deep learning; Intrusion detection; IoT; Neural networks; Representation learning; Stochastic processes; Support vector machines; TVAE
• ISSN: 1558-2612
• DOI: 10.1109/WCNC51071.2022.9771793

Intrusion detection systems (IDSs) play a pivotal role in defending IoT systems. However, developing a robust and efficient IDS is challenging due to the rapid and continuing evolving of various forms of cyber-attacks as well as a massive number of low-end IoT devices. In this paper, we introduce a novel deep learning architecture based on auto-encoders that allows to develop a robust intrusion detection system. Specifically, we propose a novel neural network architecture called Twin Variational Auto-Encoder (TVAE) for representation learning. TVAE includes a variational Auto-Encoder (VAE) and an Auto-Encoder (AE) that share a common stage where the decoder of the VAE is used as the encoder of the AE. The TVAE is trained in an unsupervised manner to effectively transform the original representation of data at the input of the VAE into a new representation at the output of the AE. In the new representation space, the difference between normal and attack data is more distinguishable. A variant of TVAE, namely Twin Sparse Variational Auto-Encoder (TSVAE) is also introduced by imposing a sparsity constraint on the representation units. The effectiveness of TVAE and TSVAE is evaluated using popular IDS and IoT botnet datasets. The simulation results show that the accuracy of TVAE and TSVAE can achieve the best results on six datasets, which is higher than those of state-of-the-art AE and VAE variants. We also investigate various characteristics of TVAE in the latent space as well as in the data extraction process. Besides applications on the IoT IDS, TVAE can also be applicable to all conventional network IDSs.