An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT Datasets

The rapid expansion of Edge and Industrial Internet of Things (IIoT) systems has intensified the risk and complexity of cyberattacks. Detecting advanced intrusions in these heterogeneous and high-dimensional environments remains challenging. As the IIoT becomes integral to critical infrastructure, e...

Full description

Saved in:
Bibliographic Details
Main Authors: Ahmad Salehiyan, Pardis Sadatian Moghaddam, Masoud Kaveh
Format: Article
Language:English
Published: MDPI AG 2025-06-01
Series:Future Internet
Subjects:
Online Access:https://www.mdpi.com/1999-5903/17/7/279
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1850077421129170944
author Ahmad Salehiyan
Pardis Sadatian Moghaddam
Masoud Kaveh
author_facet Ahmad Salehiyan
Pardis Sadatian Moghaddam
Masoud Kaveh
author_sort Ahmad Salehiyan
collection DOAJ
description The rapid expansion of Edge and Industrial Internet of Things (IIoT) systems has intensified the risk and complexity of cyberattacks. Detecting advanced intrusions in these heterogeneous and high-dimensional environments remains challenging. As the IIoT becomes integral to critical infrastructure, ensuring security is crucial to prevent disruptions and data breaches. Traditional IDS approaches often fall short against evolving threats, highlighting the need for intelligent and adaptive solutions. While deep learning (DL) offers strong capabilities for pattern recognition, single-model architectures often lack robustness. Thus, hybrid and optimized DL models are increasingly necessary to improve detection performance and address data imbalance and noise. In this study, we propose an optimized hybrid DL framework that combines a transformer, generative adversarial network (GAN), and autoencoder (AE) components, referred to as Transformer–GAN–AE, for robust intrusion detection in Edge and IIoT environments. To enhance the training and convergence of the GAN component, we integrate an improved chimp optimization algorithm (IChOA) for hyperparameter tuning and feature refinement. The proposed method is evaluated using three recent and comprehensive benchmark datasets, WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT, widely recognized as standard testbeds for IIoT intrusion detection research. Extensive experiments are conducted to assess the model’s performance compared to several state-of-the-art techniques, including standard GAN, convolutional neural network (CNN), deep belief network (DBN), time-series transformer (TST), bidirectional encoder representations from transformers (BERT), and extreme gradient boosting (XGBoost). Evaluation metrics include accuracy, recall, AUC, and run time. Results demonstrate that the proposed Transformer–GAN–AE framework outperforms all baseline methods, achieving a best accuracy of 98.92%, along with superior recall and AUC values. The integration of IChOA enhances GAN stability and accelerates training by optimizing hyperparameters. Together with the transformer for temporal feature extraction and the AE for denoising, the hybrid architecture effectively addresses complex, imbalanced intrusion data. The proposed optimized Transformer–GAN–AE model demonstrates high accuracy and robustness, offering a scalable solution for real-world Edge and IIoT intrusion detection.
format Article
id doaj-art-25befaafe55040feb7861d9705d4d6a6
institution DOAJ
issn 1999-5903
language English
publishDate 2025-06-01
publisher MDPI AG
record_format Article
series Future Internet
spelling doaj-art-25befaafe55040feb7861d9705d4d6a62025-08-20T02:45:49ZengMDPI AGFuture Internet1999-59032025-06-0117727910.3390/fi17070279An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT DatasetsAhmad Salehiyan0Pardis Sadatian Moghaddam1Masoud Kaveh2School of Industrial Engineering and Management, Oklahoma State University, Stillwater, OK 74078, USADepartment of Computer Science, Georgia State University, Atlanta, GA 30302, USADepartment of Information and Communication Engineering, Aalto University, 02150 Espoo, FinlandThe rapid expansion of Edge and Industrial Internet of Things (IIoT) systems has intensified the risk and complexity of cyberattacks. Detecting advanced intrusions in these heterogeneous and high-dimensional environments remains challenging. As the IIoT becomes integral to critical infrastructure, ensuring security is crucial to prevent disruptions and data breaches. Traditional IDS approaches often fall short against evolving threats, highlighting the need for intelligent and adaptive solutions. While deep learning (DL) offers strong capabilities for pattern recognition, single-model architectures often lack robustness. Thus, hybrid and optimized DL models are increasingly necessary to improve detection performance and address data imbalance and noise. In this study, we propose an optimized hybrid DL framework that combines a transformer, generative adversarial network (GAN), and autoencoder (AE) components, referred to as Transformer–GAN–AE, for robust intrusion detection in Edge and IIoT environments. To enhance the training and convergence of the GAN component, we integrate an improved chimp optimization algorithm (IChOA) for hyperparameter tuning and feature refinement. The proposed method is evaluated using three recent and comprehensive benchmark datasets, WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT, widely recognized as standard testbeds for IIoT intrusion detection research. Extensive experiments are conducted to assess the model’s performance compared to several state-of-the-art techniques, including standard GAN, convolutional neural network (CNN), deep belief network (DBN), time-series transformer (TST), bidirectional encoder representations from transformers (BERT), and extreme gradient boosting (XGBoost). Evaluation metrics include accuracy, recall, AUC, and run time. Results demonstrate that the proposed Transformer–GAN–AE framework outperforms all baseline methods, achieving a best accuracy of 98.92%, along with superior recall and AUC values. The integration of IChOA enhances GAN stability and accelerates training by optimizing hyperparameters. Together with the transformer for temporal feature extraction and the AE for denoising, the hybrid architecture effectively addresses complex, imbalanced intrusion data. The proposed optimized Transformer–GAN–AE model demonstrates high accuracy and robustness, offering a scalable solution for real-world Edge and IIoT intrusion detection.https://www.mdpi.com/1999-5903/17/7/279Industrial Internet of Thingsintrusion detectiontransformergenerative adversarial networkautoencoderchimp optimization algorithm
spellingShingle Ahmad Salehiyan
Pardis Sadatian Moghaddam
Masoud Kaveh
An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT Datasets
Future Internet
Industrial Internet of Things
intrusion detection
transformer
generative adversarial network
autoencoder
chimp optimization algorithm
title An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT Datasets
title_full An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT Datasets
title_fullStr An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT Datasets
title_full_unstemmed An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT Datasets
title_short An Optimized Transformer–GAN–AE for Intrusion Detection in Edge and IIoT Systems: Experimental Insights from WUSTL-IIoT-2021, EdgeIIoTset, and TON_IoT Datasets
title_sort optimized transformer gan ae for intrusion detection in edge and iiot systems experimental insights from wustl iiot 2021 edgeiiotset and ton iot datasets
topic Industrial Internet of Things
intrusion detection
transformer
generative adversarial network
autoencoder
chimp optimization algorithm
url https://www.mdpi.com/1999-5903/17/7/279
work_keys_str_mv AT ahmadsalehiyan anoptimizedtransformerganaeforintrusiondetectioninedgeandiiotsystemsexperimentalinsightsfromwustliiot2021edgeiiotsetandtoniotdatasets
AT pardissadatianmoghaddam anoptimizedtransformerganaeforintrusiondetectioninedgeandiiotsystemsexperimentalinsightsfromwustliiot2021edgeiiotsetandtoniotdatasets
AT masoudkaveh anoptimizedtransformerganaeforintrusiondetectioninedgeandiiotsystemsexperimentalinsightsfromwustliiot2021edgeiiotsetandtoniotdatasets
AT ahmadsalehiyan optimizedtransformerganaeforintrusiondetectioninedgeandiiotsystemsexperimentalinsightsfromwustliiot2021edgeiiotsetandtoniotdatasets
AT pardissadatianmoghaddam optimizedtransformerganaeforintrusiondetectioninedgeandiiotsystemsexperimentalinsightsfromwustliiot2021edgeiiotsetandtoniotdatasets
AT masoudkaveh optimizedtransformerganaeforintrusiondetectioninedgeandiiotsystemsexperimentalinsightsfromwustliiot2021edgeiiotsetandtoniotdatasets