Towards Early and Accurate IoT Device-Type Identification with Global Attention Mechanism

Xiaoyan Hu,Yuxin Shi,Guang Cheng,Ruidong Li,Hua Wu,Gang Wang
DOI: https://doi.org/10.1109/globecom54140.2023.10437132
2023-01-01
Abstract:With the rapid development of Internet of Things (IoT) technology, there is explosive growth in the number of IoT devices. Meanwhile, the low security and network heterogeneity of IoT networks have brought new challenges to implementing network management and security strategies in smart homes and small offices. Early and accurate IoT device-type identification is the first step towards the security management of IoT networks. The existing machine learning-based and deep learning-based models for IoT traffic classification have achieved decent results. However, most of these methods rely on a long-term window to collect IoT device traffic for identification, resulting in limited real-time performance. This work proposes IoT-GFCN, an early and accurate IoT device-type identification model with global attention mechanism. IoT-GFCN first constructs a multi-feature sequence for each device from a small packet window. Then IoT-GFCN resorts to the global attention mechanism to efficiently mine temporal information and feature relationships and obtain an updated embedding of each multi-feature sequence. Finally, a fully convolutional neural network is trained based on the updated embeddings of traffic features to identify IoT device types. Our experimental study suggests that IoT-GFCN can efficiently capture distinguishable representations for packet-level features of IoT traffic and outperforms state-of-the-art IoT identification methods. It achieves an average accuracy of 98.88% with a window size of 75 packets (the traffic of about three minutes) on the UNSW dataset.
What problem does this paper attempt to address?