Explainable Deep Learning-Based Feature Selection and Intrusion Detection Method on the Internet of Things

• Published in: Sensors (Basel, Switzerland) Vol. 24; no. 16; p. 5223
• Main Authors: Chen, Xuejiao, Liu, Minyao, Wang, Zixuan, Wang, Yun
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 12.08.2024; MDPI
• Subjects: Algorithms; Approximation; Artificial intelligence; convolutional neural network; Datasets; Decision making; Deep learning; Feature selection; information gain; Internet of Things; Intrusion detection systems; Machine learning; Methods; model interpretability; Neural networks; Optimization; random forest; Support vector machines; Visualization
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s24165223

With the rapid advancement of the Internet of Things, network security has garnered increasing attention from researchers. Applying deep learning (DL) has significantly enhanced the performance of Network Intrusion Detection Systems (NIDSs). However, due to its complexity and “black box” problem, deploying DL-based NIDS models in practical scenarios poses several challenges, including model interpretability and being lightweight. Feature selection (FS) in DL models plays a crucial role in minimizing model parameters and decreasing computational overheads while enhancing NIDS performance. Hence, selecting effective features remains a pivotal concern for NIDSs. In light of this, this paper proposes an interpretable feature selection method for encrypted traffic intrusion detection based on SHAP and causality principles. This approach utilizes the results of model interpretation for feature selection to reduce feature count while ensuring model reliability. We evaluate and validate our proposed method on two public network traffic datasets, CICIDS2017 and NSL-KDD, employing both a CNN and a random forest (RF). Experimental results demonstrate superior performance achieved by our proposed method.