Multiclass Cyber Attack Classification in Smart Home IoT Networks Using Ensemble Machine Learning with the ML-EdgeIIoT Dataset

PDF (1207KB), PP.152-177

Views: 0 Downloads: 0

Author(s)

Abhay Kumar Ray 1,* Rupak Sharma 1 Sunil Kumar Pandey 2

1. Department of Computer applications, SRM- Institute of Science & Technology, Delhi –NCR Campus, Ghaziabad, 201204, India

2. Department of IT, Institute of Technology & Science, Mohan Nagar, Ghaziabad, 201007, India

* Corresponding author.

DOI: https://doi.org/10.5815/ijwmt.2026.04.10

Received: 6 Jun. 2026 / Revised: 24 Jun. 2026 / Accepted: 8 Jul. 2026 / Published: 8 Aug. 2026

Index Terms

Smart Home Security, Edge Computing, Intrusion Detection System (IDS), IoT/IIoT Security, Machine Learning, Multiclass Classification, Cyberattack Detection, Ensemble Learning

Abstract

With the rapid adoption of smart home solutions and related technologies, edge computing has emerged as a key enabler by offering low-latency data processing, increased efficiency and improved scalability. However, this integration in IoT systems introduces complex security challenges in smart home edge environments, increasingly susceptible to cyber threats such as denial-of-service (DoS), malware injection, passive surveillance, and unauthorized access. This paper investigates intelligent intrusion detection and attack classification strategies specifically designed for smart home edge systems. Using the comprehensive ML-EdgeIIoT dataset, this study designs and evaluates a machine learning-based intrusion detection framework for multiclass classification of eight categories of IoT network attacks, namely Backdoor, MITM, DDoS, Ransomware, Password Attack, SQL Injection, Prob-attacks, and Normal traffic while minimizing false positives and false negatives. The framework incorporates data cleaning, correlation- and feature importance-based feature selection, hyperparameter optimization using gridsearchCV, model training, and ensemble learning. A set of machine learning models comprising Artificial Neural Network, Balanced Random Forest, K-Nearest Neighbours, Random Forest, and Logistic Regression was implemented and comparatively evaluated. Two ensemble techniques were subsequently developed using the three best-performing classifiers: (1) a stacking ensemble with Logistic Regression as the meta-learner and (2) a Top-3 majority voting ensemble. Model performance was evaluated using accuracy, precision, recall, F1-score, confusion matrix, and ROC-AUC. Robustness and generalization of the individual machine learning models were assessed through stratified 10-fold cross-validation for the three best-performing classifiers. The Top-3 voting ensemble subsequently achieved the highest performance on the independent test set, with accuracy of 99.24%, average precision of 98.75%, recall of 99.00%, and an F1-score of 99.00% for all attack classes, while reducing misclassification compared with individual classifiers. The findings of this study significantly enhance the understanding of smart home edge computing security, which will pave the way for more robust and intelligent threat detection frameworks.

Cite This Paper

Abhay Kumar Ray, Rupak Sharma, Sunil Kumar Pandey, "Multiclass Cyber Attack Classification in Smart Home IoT Networks Using Ensemble Machine Learning with the ML-EdgeIIoT Dataset", International Journal of Wireless and Microwave Technologies(IJWMT), Vol.16, No.4, pp. 152-177, 2026. DOI:10.5815/ijwmt.2026.04.10

Reference

[1]K. Maswadi, N. B. A. Ghani and S. B. Hamid, "Systematic Literature Review of Smart Home Monitoring Technologies Based on IoT for the Elderly," in IEEE Access, vol. 8, pp. 92244-92261, 2020, doi: 10.1109/ACCESS.2020.2992727.
[2]Yar, H.; Imran, A.S. Khan, Z.A. Sajjad, M. Kastrati, Z. " Towards Smart Home Automation Using IoT-Enabled Edge-Computing Paradigm" . Sensors 2021, 21, 4932. https://doi.org/10.3390/s21144932
[3]Vardakis, G. Hatzivasilis, G. Koutsaki, E. Papadakis, N. "Review of Smart-Home Security Using the Internet of Things" . Electronics 2024, 13, 3343. https://doi.org/10.3390/electronics13163343
[4]S. Shaukat "Intrusion Detection and Attack Classification Leveraging Machine Learning Technique," 2020 14th International Conference on Innovations in Information Technology (IIT), Al Ain, United Arab Emirates, 2020, pp. 198-202, doi: 10.1109/IIT50501.2020.9299093.
[5]Alani, M. M., Alshaer, J., Alrashdi, I., et al. (2023). XRecon: An explainable IoT reconnaissance attack detection framework using ensemble machine learning. Sensors, 23(11), 5298. https://doi.org/10.3390/s23115298
[6]R. A. Nugrahaeni and K. Mutijarsa, "Comparative analysis of machine learning KNN, SVM, and random forests algorithm for facial expression classification," 2016 International Seminar on Application for Technology of Information and Communication (ISemantic), Semarang, Indonesia, 2016, pp. 163-168, doi: 10.1109/ISEMANTIC.2016.7873831.
[7]K. Zhang, K. Yang, S. Li, D. Jing and H. -B. Chen, "ANN-Based Outlier Detection for Wireless Sensor Networks in Smart Buildings," in IEEE Access, vol. 7, pp. 95987-95997, 2019, doi: 10.1109/ACCESS.2019.2929550.
[8]G. Sharkov, W. Asif and I. Rehman, "Securing Smart Home Environment Using Edge Computing," 2022 IEEE International Smart Cities Conference (ISC2), Pafos, Cyprus, 2022, pp. 1-7, doi: 10.1109/ISC255366.2022.9921912.
[9]A. K. Ray and A. Bagwari, "IoT based Smart home: Security Aspects and security architecture," 2020 IEEE 9th International Conference on Communication Systems and Network Technologies (CSNT), Gwalior, India, 2020, pp. 218-222, doi: 10.1109/CSNT48778.2020.9115737
[10]Rani Molla,  "Amazon’s Ring blamed hacks on consumers reusing their passwords. A lawsuit says that’s not true." Reported on: Jan 18, 2020, URL:https://www.vox.com/recode/2020/1/17/21068703/amazon-ring-hacks-lawsuit-passwords last seen May ,2025
[11]Sandra E. Garcia, "Data Breach at Wyze Labs Exposes Information of 2.4 Million Customers" URL : https://www.nytimes.com/2019/12/30/business/wyze-security-camera-breach.html  last seen may-2025
[12]Eray Eliaçık, "Wyze camera breach impacts 13,000 users’ security" Published on : February 20, 2024 URL:https://www.dataconomy.com/2024/02/20/wyze-camera-breach-security/ last seen May-2025
[13]Rupanetti, D., Kaabouch, N.,” Combining Edge Computing-Assisted Internet of Things Security with Artificial Intelligence: Applications, Challenges, and Opportunities.” Appl. Sci. 2024, 14, 7104. https://doi.org/10.3390/app14167104
[14]Ansari, M.S., Alsamhi, S.H., Qiao, Y., Ye, Y., Lee, B. (2020). “Security of Distributed Intelligence in Edge Computing: Threats and Countermeasures.” In: Lynn, T., Mooney, J., Lee, B., Endo, P. (eds) The Cloud-to-Thing Continuum. Palgrave Studies in Digital Business & Enabling Technologies. Palgrave Macmillan, Cham. https://doi.org/10.1007/978-3-030-41110-7_6
[15]Mutleg, M.L., Mahmood, A.M., Al-Nayar, M.M.J. (2024). “A comprehensive review of cyber-attacks targeting IoT systems and their security measures” ,International Journal of Safety and Security Engineering, Vol. 14, No. 4, pp. 1073-1086. https://doi.org/10.18280/ijsse.140406
[16]Hassen Sallay , "Designing an Adaptive Effective Intrusion Detection System for Smart Home IoT A Device-Specific Approach with SDN Deployment "  (IJACSA) International Journal of Advanced Computer Science and Applications, Vol. 15, No. 1, 2024  URL : https://thesai.org/Downloads/Volume15No1/Paper_94- Designing_An_Adaptive_Effective_Intrusion_Detection_System.pdf
[17]Khraisat, A., Gondal, I., Vamplew, P. et al. “Survey of intrusion detection systems: techniques, datasets and challenges”, Cybersecur 2, 20 (2019). https://doi.org/10.1186/s42400-019-0038-7
[18]Jose Luis Hernandez-Ramos, Georgios Karopoulos, Efstratios Chatzoglou, Vasileios Kouliaridis, Enrique Marmol, Aurora Gonzalez-Vidal, and Georgios Kambourakis. 2025. “Intrusion Detection Based on Federated Learning: A Systematic Review”, ACM Comput. Surv. 57, 12, Article 309 (December 2025), 65 pages. https://doi.org/10.1145/3731596
[19]Ferrag, M. A., Friha, O., Hamouda, D., Maglaras, L., & Janicke, H. (2022). “Edge-IIoTset: A new comprehensive realistic cyber security dataset of IoT and IIoT applications for centralized and federated learning.” IEEE Access, 10, 40281–40306. https://doi.org/10.1109/ACCESS.2022.3165809 
[20]Gueriani, A., Kheddar, H., & Mazari, A. C. (2025) “Adaptive cyber-attack detection in IIoT using attention-based LSTM-CNN models” arXiv. Doi:https://doi.org/10.48550/arXiv.2501.13962 .
[21]S. Sohail, Z. Fan, X. Gu, and F. Sabrina, “Explainable and optimally configured artificial neural networks for attack detection in smart homes,” arXiv, May 2022.
[22]Malak Aljabri , Afrah Shaahid et. al.  “IoT Attacks Detection Using Supervised Machine Learning Techniques,” HighTech Innovation Journal, vol. 5, no. 3, Sept-2024 . DOI: https://doi.org/10.28991/HIJ-2024-05-03-01
[23]Ali. H. Farea and  Kerem Kucuk  “Machine learning-based intrusion detection technique for IoT: Simulation with Cooja,” International Journal of Computer Network and Information Security, vol. 16, no. 1, pp. 1–23, 2024.
[24] A. Aldaej, T. A. Ahanger, and I. Ullah, “Deep Learning‑Inspired IoT‑IDS Mechanism for Edge Computing Environments,” Sensors, vol. 23, no. 24, Art. no. 9869, Dec. 2023.
[25]Ismail, S., Dandan, S., & Qushou, A. (2025). “Intrusion detection in IoT and IIoT: Comparing lightweight machine learning techniques using TON_IoT, WUSTL-IIoT-2021, and Edge-IIoTset datasets” IEEE Access. Advance online publication. Doi: https://doi.org/10.1109/ACCESS.2025.3554083
[26] Rieger, P., Chilese, M., Mohamed, R., Miettinen, M., Fereidooni, H., & Sadeghi, A.-R. (2023). ARGUS: Context-based detection of stealthy IoT infiltration attacks. In Proceedings of the 32nd USENIX Security Symposium (USENIX Security '23) (pp. 4301–4318). USENIX Association. https://dl.acm.org/doi/10.5555/3620237.3620478
[27]Tanveer, M. U., Munir, K., Amjad, M., Alyamani, H. J., Bermak, A., & Ur Rehman, A. (2025). “LightEnsemble-Guard: An optimized ensemble learning framework for securing resource-constrained IoT systems”  IEEE Access, 13, 101764–101781. Doi: https://doi.org/10.1109/ACCESS.2025.3576218
[28]Thanh Noi, P.; Kappas, M. "Comparison of Random Forest, k-Nearest Neighbor, and Support Vector Machine Classifiers for Land Cover Classification Using Sentinel-2 Imagery." Sensors 2018, 18, 18. https://doi.org/10.3390/s18010018
[29]Ferrag, Mohamed Amine, Friha, Othmane, Hamouda, Djallel, Maglaras, Leandros Janicke, Helge (2022): “Edge-IIoTset: A New Comprehensive Realistic Cyber Security Dataset of IoT and IIoT Applications for Centralized and Federated Learning” TechRxiv. Preprint. https://doi.org/10.36227/techrxiv.18857336.v1, Web-link: https://www.kaggle.com/datasets/mohamedamineferrag/edgeiiotset-cyber-security-dataset-of-iot-iiotlast seen Dec 2025
[30]I. D. Mienye and Y. Sun, "A Survey of Ensemble Learning: Concepts, Algorithms, Applications, and Prospects," in IEEE Access, vol. 10, pp. 99129-99149, 2022, doi: 10.1109/ACCESS.2022.3207287.
[31]Ebo Jackson "Top 10 Python Libraries for AI and Machine Learning: A Comprehensive Guide with Code Examples" URL :  https://medium.com/@ebojacky/top-10-python-libraries-for-ai-and-machine-learning-a-comprehensive-guide-with-code-examples-e89bfa1ab7df last seen July 2025
[32]A. K. Ray, R. Sharma and S. K. Pandey, "Integrated Machine Learning Approach for Attack Detection in MQTT-Enabled Smart Home Systems," 2024 International Conference on Emerging Technologies and Innovation for Sustainability (EmergIN), Greater Noida, India, 2024, pp. 351-356, doi: 10.1109/EmergIN63207.2024.10960978.