arXiv:2501.15266cs.LG2025-01被引 24

用自编码器提升工业物联网入侵检测,轻量模型实测精度达99.94%。

Enhanced Intrusion Detection in IIoT Networks: A Lightweight Approach with Autoencoder-Based Feature Learning

  • 用自编码器降维增强特征学习,解决数据不平衡问题。
  • 在Edge-IIoTset数据集上达99.94%准确率和F1分数。
  • 模型轻量可部署于边缘设备,推理速度仅0.187毫秒。

工业互联网(IIoT)的快速发展推动了数字技术与工业系统的融合,但也加剧了网络攻击风险,亟需可靠的入侵检测系统(IDS)。尽管基于机器学习的IDS有潜力,但现有模型常因类别不平衡和多类数据集导致检测精度下降。本研究提出六项创新方法,包括利用自编码器进行降维以优化特征学习。所提出的决策树模型在Edge-IIoTset数据集上达到99.94%的准确率和F1分数。同时注重轻量化设计,确保在资源受限的边缘设备上部署可行。首次将模型部署于Jetson Nano,二分类推理时间仅0.185毫秒,多分类为0.187毫秒。结果表明该方法在处理不平衡与多类数据时具有显著优势,为工业物联网安全提供了高效实用的解决方案。

原文摘要 · Abstract (English)

The rapid expansion of the Industrial Internet of Things (IIoT) has significantly advanced digital technologies and interconnected industrial systems, creating substantial opportunities for growth. However, this growth has also heightened the risk of cyberattacks, necessitating robust security measures to protect IIoT networks. Intrusion Detection Systems (IDS) are essential for identifying and preventing abnormal network behaviors and malicious activities. Despite the potential of Machine Learning (ML)--based IDS solutions, existing models often face challenges with class imbalance and multiclass IIoT datasets, resulting in reduced detection accuracy. This research directly addresses these challenges by implementing six innovative approaches to enhance IDS performance, including leveraging an autoencoder for dimensional reduction, which improves feature learning and overall detection accuracy. Our proposed Decision Tree model achieved an exceptional F1 score and accuracy of 99.94% on the Edge-IIoTset dataset. Furthermore, we prioritized lightweight model design, ensuring deployability on resource-constrained edge devices. Notably, we are the first to deploy our model on a Jetson Nano, achieving inference times of 0.185 ms for binary classification and 0.187 ms for multiclass classification. These results highlight the novelty and robustness of our approach, offering a practical and efficient solution to the challenges posed by imbalanced and multiclass IIoT datasets, thereby enhancing the detection and prevention of network intrusions.

入侵检测轻量模型自编码器边缘计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。