An XGBoost-Based Intrusion Detection Framework with Interpretability Analysis for IoT Networks
Source abstract
Abstract
With the rapid development of the Internet of Things (IoT) and Industrial IoT (IIoT), Network Intrusion Detection Systems (NIDSs) play a critical role in securing modern networked environments. Despite advances in multi-class intrusion detection, existing approaches face challenges from high-dimensional heterogeneous traffic data, severe class imbalance, and limited interpretability of high-performance “black-box” models. To address these issues, this study presents an XGBoost-based NIDSs integrating optimized strategies for feature dimensionality reduction and class balancing, alongside SHAP-based interpretability analysis. Feature reduction is investigated by comparing selection methods that preserve original features with generation methods that create transformed features, aiming to balance detection performance and computational efficiency. Class balancing techniques are evaluated to improve minority-class detection, particularly reducing false negatives for rare attack types. SHAP analysis reveals the model’s decision process and key feature contributions. The experimental results demonstrate that the method enhances multi-class detection performance while providing interpretability and computational efficiency, highlighting its potential for practical deployment in IoT security scenarios.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Network Security and Intrusion Detection · Anomaly Detection Techniques and Applications · Explainable Artificial Intelligence (XAI)
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Computer Science and Engineering, University of Electronic Science and Technology of China, Chengdu 611731, China
References (57)
Show source-provided references
1. Intrusion detection for IoT based on improved genetic algorithm and deep belief network10.1109/ACCESS.2019.2903723 ↗
2. An advanced intrusion detection system for IIoT based on GA and tree based algorithms10.1109/ACCESS.2021.3104113 ↗
3. Intrusion detection techniques and approaches10.1016/S0140-3664(02)00037-3 ↗
4. The UCI KDD archive of large data sets for data mining research and experimentation10.1145/380995.381030 ↗
5. (2025, October 10). NSL-KDD | Datasets | Research | Canadian Institute for Cybersecurity |UNB. Available online: https://www.unb.ca/cic/datasets/nsl.html.
6. A new distributed architecture for evaluating AI-based security systems at the edge: Network TON_IoT datasets10.1016/j.scs.2021.102994 ↗
7. A hierarchical intrusion detection model based on the PCA neural networks10.1016/j.neucom.2006.10.146 ↗
8. Studies on optimization algorithms for some artificial neural networks based on genetic algorithm (GA)10.4304/jcp.6.5.939-946 ↗
9. An effective convolutional neural network based on SMOTE and Gaussian mixture model for intrusion detection in imbalanced dataset10.1016/j.comnet.2020.107315 ↗
10. Anderson, J.P. (1980). Computer Security Threat Monitoring and Surveillance, James P. Anderson Company. Technical Report.
11. Axelsson, S. (2000). Intrusion Detection Systems: A Survey and Taxonomy. [Ph.D. Thesis, Chalmers University of Technology].
12. TSE-IDS: A two-stage classifier ensemble for intelligent anomaly-based intrusion detection system10.1109/ACCESS.2019.2928048 ↗
13. Hall, M.A. (1999). Correlation-Based Feature Selection for Machine Learning. [Ph.D. Thesis, The University of Waikato].
14. Building an intrusion detection system using a filter-based feature selection algorithm10.1109/TC.2016.2519914 ↗
15. An ensemble intrusion detection technique based on proposed statistical flow features for protecting network traffic of internet of things10.1109/JIOT.2018.2871719 ↗
16. Enhancing security in industrial IoT networks: Machine learning solutions for feature selection and reduction10.1109/ACCESS.2024.3481459 ↗
17. Performance analysis of machine learning models for intrusion detection system using Gini Impurity-based Weighted Random Forest (GIWRF) feature selection technique10.1186/s42400-021-00103-8 ↗
18. A hybrid method consisting of GA and SVM for intrusion detection system10.1007/s00521-015-1964-2 ↗
19. A GA-LR wrapper approach for feature selection in network intrusion detection10.1016/j.cose.2017.06.005 ↗
20. Research on intrusion detection based on particle swarm optimization in IoT10.1109/ACCESS.2021.3063671 ↗
21. Sarwar, A., Alnajim, A.M., Marwat, S.N.K., Ahmed, S., Alyahya, S., and Khan, W.U. (2022). Enhanced anomaly detection system for iot based on improved dynamic SBPSO. Sensors, 22.10.3390/s22134926 ↗
22. Effective feature extraction via stacked sparse autoencoder to improve intrusion detection system10.1109/ACCESS.2018.2858277 ↗
23. Xu, X., and Wang, X. (2025, January 22–24). An adaptive Network Intrusion Detection Method Based on PCA and Support Vector Machines. Proceedings of the International Conference on Advanced Data Mining and Applications, Berlin/Heidelberg, Germany.
24. Abdulhammed, R., Faezipour, M., Musafer, H., and Abuzneid, A. (2019, January 18–20). Efficient Network Intrusion Detection Using PCA-Based Dimensionality Reduction of Features. Proceedings of the 2019 International Symposium on Networks, Computers and Communications (ISNCC), Istanbul, Turkey.10.1109/ISNCC.2019.8909140 ↗
25. Moustafa, N., and Slay, J. (2015, January 10–12). UNSW-NB15: A Comprehensive Data Set for Network Intrusion Detection Systems. Proceedings of the 2015 Military Communications and Information Systems Conference (MilCIS), Canberra, Australia.10.1109/MilCIS.2015.7348942 ↗
26. A novel two-stage deep learning model for efficient network intrusion detection10.1109/ACCESS.2019.2899721 ↗
27. Variational LSTM enhanced anomaly detection for industrial big data10.1109/TII.2020.3022432 ↗
28. Hybrid deep learning for botnet attack detection in the internet-of-things networks10.1109/JIOT.2020.3034156 ↗
29. Optimizing IoT intrusion detection system: Feature selection versus feature extraction in machine learning10.1186/s40537-024-00892-y ↗
30. Stacked autoencoder-based probabilistic feature extraction for on-device network intrusion detection10.1109/JIOT.2021.3078292 ↗
31. Network traffic classification using deep convolutional recurrent autoencoder neural networks for spatial–temporal features extraction10.1016/j.jnca.2020.102890 ↗
32. SMOTE: Synthetic minority over-sampling technique10.1613/jair.953 ↗
33. Belarbi, O., Khan, A., Carnelli, P., and Spyridopoulos, T. (2022, January 10–12). An Intrusion Detection System Based on Deep Belief Networks. Proceedings of the International Conference on Science of Cyber Security, Shimane, Japan.10.1007/978-3-031-17551-0_25 ↗
34. Chalichalamala, S., Govindan, N., and Kasarapu, R. (2023). Logistic regression ensemble classifier for intrusion detection system in internet of things. Sensors, 23.10.3390/s23239583 ↗
35. Intrusion detection system after data augmentation schemes based on the VAE and CVAE10.1109/TR.2022.3164877 ↗
36. Detection of real-time malicious intrusions and attacks in IoT empowered cybersecurity infrastructures10.1109/ACCESS.2023.3238664 ↗
37. Induction of decision trees10.1023/A:1022643204877 ↗
38. Hosmer, D.W., Lemeshow, S., and Sturdivant, R.X. (2013). Applied Logistic Regression, John Wiley & Sons.10.1002/9781118548387 ↗
39. Anomaly-based intrusion detection system for IoT networks through deep learning model10.1016/j.compeleceng.2022.107810 ↗
40. Alalmaie, A., Nanda, P., and He, X. (February, January 31). Zero Trust Network Intrusion Detection System (NIDS) Using Auto Encoder for Attention-Based CNN-BiLSTM. Proceedings of the 2023 Australasian Computer Science Week, Melbourne, Australia.10.1145/3579375.3579376 ↗
41. Lo, W.W., Layeghy, S., Sarhan, M., Gallagher, M., and Portmann, M. (2021). E-graphsage: A graph neural network based intrusion detection system for iot. arXiv.10.1109/NOMS54207.2022.9789878 ↗
42. Novel online network intrusion detection system for industrial IoT based on OI-SVDD and AS-ELM10.1109/JIOT.2022.3172393 ↗
43. A stacking ensemble of deep learning models for IoT intrusion detection10.1016/j.knosys.2023.110941 ↗
44. An adaptable deep learning-based intrusion detection system to zero-day attacks
45. E-xai: Evaluating black-box explainable ai frameworks for network intrusion detection10.1109/ACCESS.2024.3365140 ↗
46. Enhanced Network Intrusion Detection Systems With Explainable Artificial Intelligence for Network Security10.1002/dac.70209 ↗
47. A versatile XAI-based framework for efficient and explainable intrusion detection systems10.1007/s12243-025-01118-9 ↗
48. Kalakoti, R., Vaarandi, R., Bahsi, H., and Nõmm, S. (2025). Evaluating explainable AI for deep learning-based network intrusion detection system alert classification. arXiv.10.5220/0013180700003899 ↗
49. Random forests10.1023/A:1010933404324 ↗
50. An introduction to variable and feature selection
51. Principal component analysis10.1016/0169-7439(87)80084-9 ↗
52. Independent component analysis, a new concept?10.1016/0165-1684(94)90029-9 ↗
53. A study of the behavior of several methods for balancing machine learning training data10.1145/1007730.1007735 ↗
54. Han, H., Wang, W.Y., and Mao, B.H. (2005, January 23–26). Borderline-SMOTE: A new over-sampling method in imbalanced data sets learning. Proceedings of the International Conference on Intelligent Computing, Hefei, China.10.1007/11538059_91 ↗
55. He, H., Bai, Y., Garcia, E.A., and Li, S. (2008, January 1–8). ADASYN: Adaptive synthetic sampling approach for imbalanced learning. Proceedings of the 2008 IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence), Hong Kong, China.
56. A distributed intrusion detection system using machine learning for IoT based on ToN-IoT dataset
57. An ensemble learning and fog-cloud architecture-driven cyber-attack detection framework for IoMT networks10.1016/j.comcom.2020.12.003 ↗