<i>HGExplainer</i>: Heterogeneous Graph Explainer for IoT Device Identification
IoT device identification is vital for network asset and security management. However, existing methods use statistical features that can not identify IoT devices accurately in complex network environments. <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">GraphIoT</i> proposes using non-statistical features and building a heterogeneous graph neural network to identify IoT devices accurately. However, heterogeneous graph neural networks lack interpretability, which reduces trust in the model. Besides, it is difficult to deploy on resource-constrained devices, limiting the broad application of IoT device identification. To make IoT device identification interpretable, easy to deploy, and with high accuracy, we get the interpretation results of <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">GraphIoT</i> through interpretability and further build the rule set based on the interpretation results. Considering there is no suitable interpreter for <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">GraphIoT</i> with many nodes and edges, we propose <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">HGExplainer</i>, which reduces the time complexity by splitting the interpretation target into important relation solving and edge solving and uses a novel solution method, ExpandTree. Then, we also designed a rule extractor, which can build rule sets based on the interpretation results. Experimental results on Yourthings and UNSW datasets show that <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">HGExplainer</i> can build high fidelity, concise sample-level explanations in less than 3 seconds, and the established rule set can precisely identify IoT devices.
Authors
- Xuan Shen (ORCID: https://orcid.org/0000-0002-7985-8661)
- Chaocan Xiang (ORCID: https://orcid.org/0000-0002-1473-6006)
- Duohe Ma (ORCID: https://orcid.org/0000-0002-2160-8344)
- Guanglei Song (ORCID: https://orcid.org/0000-0003-4736-6264)
- Linna Fan (ORCID: https://orcid.org/0000-0003-2269-1175)
- Yongfeng Huang (ORCID: https://orcid.org/0000-0003-3825-2230)
- Bo Wu (ORCID: https://orcid.org/0000-0003-0695-019X)
- Jun He
- Gang Yang (ORCID: https://orcid.org/0000-0001-8798-9760)
Institutions
- Chongqing University (CN)
- National University of Defense Technology (CN)
- Chinese Academy of Sciences (CN)
- Institute of Information Engineering (CN)
- Tsinghua University (CN)
Publication Details
- Journal
- IEEE Transactions on Mobile Computing
- Published
- 2024-10-28
- DOI
- https://doi.org/10.1109/tmc.2024.3486717
- Citations
- 2
- Primary Topic
- Advanced Malware Detection Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.70