Abstract:In recent years, with the application of Internet of Things (IoT) and cloud technology in smart industrialization, Industrial Internet of Things (IIoT) has become an emerging hot topic. The increasing amount of data and device numbers in IIoT poses significant challenges to its security issues, making anomaly detection particularly important. Existing methods for anomaly detection in the IIoT often fall short when dealing with data imbalance, and the huge amount of IIoT data makes feature selection challenging and computationally intensive. In this paper, we propose an optimal deep learning model for anomaly detection in IIoT. Firstly, by setting different thresholds of eXtreme Gradient Boosting (XGBoost) for feature selection, features with importance above the given threshold are retained, while those below are ignored. Different thresholds yield different numbers of features. This approach not only secures effective features but also reduces the feature dimensionality, thereby decreasing the consumption of computational resources. Secondly, an optimized loss function is designed to study its impact on model performance in terms of handling imbalanced data, highly similar categories, and model training. We select the optimal threshold and loss function, which are part of our optimal model, by comparing metrics such as accuracy, precision, recall, False Alarm Rate (FAR), Area Under the Receiver Operating Characteristic Curve (AUC-ROC), and Area Under the Precision–Recall Curve (AUC-PR) values. Finally, combining the optimal threshold and loss function, we propose a model named MIX_LSTM for anomaly detection in IIoT. Experiments are conducted using the UNSW-NB15 and NSL-KDD datasets. The proposed MIX_LSTM model can achieve 0.084 FAR, 0.984 AUC-ROC, and 0.988 AUC-PR values in the binary anomaly detection experiment on the UNSW-NB15 dataset. In the NSL-KDD dataset, it can achieve 0.028 FAR, 0.967 AUC-ROC, and 0.962 AUC-PR values. By comparing the evaluation indicators, the model shows good performance in detecting abnormal attacks in the Industrial Internet of Things compared with traditional deep learning models, machine learning models and existing technologies.

Variational LSTM Enhanced Anomaly Detection for Industrial Big Data

Long-short Memory with Feature Representation by Latent-variable Model for Industrial Soft Sensor

Adaptive Anomaly Detection in Industrial Systems: an EVT-DTS Approach with LSTM Autoencoders

Lightweight Long Short-Term Memory Variational Auto-Encoder for Multivariate Time Series Anomaly Detection in Industrial Control Systems

Federated Variational Learning for Anomaly Detection in Multivariate Time Series

Anomaly Detection for Multivariate Time Series in IoT Using Discrete Wavelet Decomposition and Dual Graph Attention Networks

Improved AutoEncoder with LSTM module and KL divergence

A Hybrid Algorithm Incorporating Vector Quantization and One-Class Support Vector Machine for Industrial Anomaly Detection

Learning Sparse Representation With Variational Auto-Encoder for Anomaly Detection.

Unsupervised Industrial Internet Time Series Anomaly Detection Based on Reinforcement Learning

An effective method for anomaly detection in industrial Internet of Things using XGBoost and LSTM

Toward interpretable anomaly detection for autonomous vehicles with denoising variational transformer

Anomaly Detection Model for Log Based on LSTM Network and Variational Autoencoder

DL-RLSTM: an Anomaly Detection Framework for High Dimensional Time Series Data

Variance error of multi-classification based anomaly detection for time series data

Unsupervised Deep Anomaly Detection for Industrial Multivariate Time Series Data

LUAD: A lightweight unsupervised anomaly detection scheme for multivariate time series data

VAE-based Deep SVDD for anomaly detection

Exploring Large Vision-Language Models for Robust and Efficient Industrial Anomaly Detection

Deep Anomaly Detection for Time-Series Data in Industrial IoT: A Communication-Efficient On-Device Federated Learning Approach

Minimal-Configuration Anomaly Detection for IIoT Sensors