Multi-scale Convolutional Feature Fusion Network Based on Attention Mechanism for IoT Traffic Classification

被引:0
|
作者
Niandong Liao
Jiayu Guan
机构
[1] Changsha University of Science and Technology,
关键词
IoT traffic classification; Attention mechanism; Multi-scale convolution; Feature reuse;
D O I
暂无
中图分类号
学科分类号
摘要
The Internet of Things (IoT) has been extensively utilized in domains such as smart homes, healthcare, and other industries. With the exponential growth of Internet of Things (IoT) devices, they have become prime targets for malicious cyber-attacks. Effective classification of IoT traffic is, therefore, imperative to enable robust intrusion detection systems. However, IoT traffic data contain intricate spatial relationships and topological information, which traditional methods for traffic identification lack the capability to fully extract features and capture crucial characteristics. We propose a multi-scale convolutional feature fusion network augmented with a Convolutional Block Attention Module (MCF-CBAM) for accurate IoT traffic classification. The network incorporates three critical innovations: (1) Parallel convolution extracts multi-scale spatial features from traffic data. The 1 × 1 convolution operation reduces the amount of parameters and calculations of the network, thereby improving work efficiency. (2) The attention module suppresses less informative features while highlighting the most discriminative ones, enabling focused learning on decisive features. (3) Cross-scale connections with channel jumps reuse features from prior layers to enhance generalization. We evaluate the method extensively on three widely adopted public datasets. Quantitative results demonstrate MCF-CBAM establishes new state-of-the-art performance benchmarks for IoT traffic classification, surpassing existing methods by a significant margin. Qualitative visualizations of the learned attention weights provide intuitive insights into how the network automatically discovers the most decisive spatial features for identification. With its strong empirical performance and interpretable attention mechanisms, this work presents a promising deep learning solution to augment real-world IoT intrusion detection systems against growing cybersecurity threats.
引用
收藏
相关论文
共 50 条
  • [1] Multi-scale Convolutional Feature Fusion Network Based on Attention Mechanism for IoT Traffic Classification
    Liao, Niandong
    Guan, Jiayu
    INTERNATIONAL JOURNAL OF COMPUTATIONAL INTELLIGENCE SYSTEMS, 2024, 17 (01)
  • [2] Facial Expression Recognition Based on Multi-scale Feature Fusion Convolutional Neural Network and Attention Mechanism
    Wu, Yana
    Jia, Kebin
    Sun, Zhonghua
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2021, PT II, 2021, 13020 : 324 - 335
  • [3] A Convolutional Neural Network Based on Soft Attention Mechanism and Multi-Scale Fusion for Skin Cancer Classification
    Bao, Qiwei
    Han, Hua
    Huang, Li
    Muzahid, A. A. M.
    INTERNATIONAL JOURNAL OF PATTERN RECOGNITION AND ARTIFICIAL INTELLIGENCE, 2023, 37 (14)
  • [4] MFANet: Multi-scale feature fusion network with attention mechanism
    Wang, Gaihua
    Gan, Xin
    Cao, Qingcheng
    Zhai, Qianyu
    VISUAL COMPUTER, 2023, 39 (07): : 2969 - 2980
  • [5] MFANet: Multi-scale feature fusion network with attention mechanism
    Gaihua Wang
    Xin Gan
    Qingcheng Cao
    Qianyu Zhai
    The Visual Computer, 2023, 39 : 2969 - 2980
  • [6] Dynamic graph convolutional network for assembly behavior recognition based on attention mechanism and multi-scale feature fusion
    Chengjun Chen
    Xicong Zhao
    Jinlei Wang
    Dongnian Li
    Yuanlin Guan
    Jun Hong
    Scientific Reports, 12
  • [7] Dynamic graph convolutional network for assembly behavior recognition based on attention mechanism and multi-scale feature fusion
    Chen, Chengjun
    Zhao, Xicong
    Wang, Jinlei
    Li, Dongnian
    Guan, Yuanlin
    Hong, Jun
    SCIENTIFIC REPORTS, 2022, 12 (01)
  • [8] A lightweight network for traffic sign recognition based on multi-scale feature and attention mechanism
    Wei, Wei
    Zhang, Lili
    Yang, Kang
    Li, Jing
    Cui, Ning
    Han, Yucheng
    Zhang, Ning
    Yang, Xudong
    Tan, Hongxin
    Wang, Kai
    HELIYON, 2024, 10 (04)
  • [9] BACNN: Multi-scale feature fusion-based bilinear attention convolutional neural network for wood NIR classification
    Wan, Zihao
    Yang, Hong
    Xu, Jipan
    Mu, Hongbo
    Qi, Dawei
    JOURNAL OF FORESTRY RESEARCH, 2024, 35 (01)
  • [10] HAMNet: hyperspectral image classification based on hybrid neural network with attention mechanism and multi-scale feature fusion
    Shen, Jinyue
    Zheng, Zhouzhou
    Sun, Yingwei
    Zhao, Mengmeng
    Chang, Yankang
    Shao, Yuyi
    Zhang, Yan
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2022, 43 (11) : 4233 - 4258