AIセキュリティポータル K Program
FedMADE: Robust Federated Learning for Intrusion Detection in IoT Networks Using a Dynamic Aggregation Method
Share
Abstract
The rapid proliferation of Internet of Things (IoT) devices across multiple sectors has escalated serious network security concerns. This has prompted ongoing research in Machine Learning (ML)-based Intrusion Detection Systems (IDSs) for cyber-attack classification. Traditional ML models require data transmission from IoT devices to a centralized server for traffic analysis, raising severe privacy concerns. To address this issue, researchers have studied Federated Learning (FL)-based IDSs that train models across IoT devices while keeping their data localized. However, the heterogeneity of data, stemming from distinct vulnerabilities of devices and complexity of attack vectors, poses a significant challenge to the effectiveness of FL models. While current research focuses on adapting various ML models within the FL framework, they fail to effectively address the issue of attack class imbalance among devices, which significantly degrades the classification accuracy of minority attacks. To overcome this challenge, we introduce FedMADE, a novel dynamic aggregation method, which clusters devices by their traffic patterns and aggregates local models based on their contributions towards overall performance. We evaluate FedMADE against other FL algorithms designed for non-IID data and observe up to 71.07% improvement in minority attack classification accuracy. We further show that FedMADE is robust to poisoning attacks and incurs only a 4.7% (5.03 seconds) latency overhead in each communication round compared to FedAvg, without increasing the computational load of IoT devices.
A supervised intrusion detection system for smart home iot devices
E. Anthi, L. Williams, M. Słowinska, G. Theodorakopoulos, P. Burnap
Published: 2019
Fs3: Few-shot and self-supervised framework for efficient intrusion detection in Internet of Things networks
Ayesha S., D., A.B., S., D., M.
Published: 2023
Federated deep learning for intrusion detection in IoT networks
Belarbi, O., Spyridopoulos, T., Anthi, E., Mavromatis, I., Carnelli, P., Khan, A.
Published: 2023
Smote: synthetic minority over-sampling technique
N. V. Chawla, K. W. Bowyer, L. O. Hall, W. P. Kegelmeyer
Published: 2002
A density-based algorithm for discovering clusters in large spatial databases with noise
M. Ester, H.-P. Kriegel, J. Sander, X. Xu, et al.
Published: 1996
Clustered sampling: Low-variance and improved representativity for clients selection in federated learning
Fraboni, Y., Vidal, R., Kameni, L., Lorenzi, M.
Published: 2021
Evaluating gradient inversion attacks and defenses in federated learning
Yangsibo Huang, Samyak Gupta, Zhao Song, Kai Li, Sanjeev Arora
Published: 2021
Collaborative learning model for cyberattack detection systems in iot industry 4.0
Tran Viet Khoa, Yuris Mulya Saputra, Dinh Thai Hoang, Nguyen Linh Trung, Diep Nguyen, Nguyen Viet Ha, Eryk Dutkiewicz
Published: 2020
Survey of intrusion detection systems: techniques, datasets, and challenges
Khraisat, A., et al.
Published: 2019
Towards the Development of Realistic Botnet Dataset in the Internet of Things for Network Forensic Analytics: Bot-IoT Dataset
Nickolaos Koroniotis, Nour Moustafa, Elena Sitnikova, Benjamin Turnbull
Published: 2018.11.2
Model-contrastive federated learning
Li, Q., He, B., Song, D.
Published: 2021
Federated Optimization in Heterogeneous Networks
Tian Li, Anit Kumar Sahu, Manzil Zaheer, Maziar Sanjabi, Ameet Talwalkar, Virginia Smith
Published: 2018.12.15
Deep anomaly detection for time-series data in industrial IoT: A communication-efficient on-device federated learning approach
Liu, Y., Garg, S., Nie, J., Zhang, Y., Xiong, Z., Kang, J., Hossain, M.S.
Published: 2021
Fedsld: Federated learning with shared label distribution for medical image classification
Luo, J., Wu, S.
Published: 2022
A review of recent approaches on wrapper feature selection for intrusion detection
Maldonado, J., Riff, M.C., Neveu, B.
Published: 2022
A survey and taxonomy of the fuzzy signature-based intrusion detection systems
M. Masdari, H. Khezri
Published: 2020
Communication-Efficient Learning of Deep Networks from Decentralized Data
H. Brendan McMahan, Eider Moore, Daniel Ramage, Seth Hampson, Blaise Agüera y Arcas
Published: 2016.2.18
N-baiot—network-based detection of iot botnet attacks using deep autoencoders
Y. Meidan, et al.
Published: 2018
Federated-learning-based anomaly detection for iot security attacks
Viraaji Mothukuri, Prachi Khare, Reza M Parizi, Seyedamin Pouriyeh, Ali Dehghantanha, Gautam Srivastava
Published: 2021
A new distributed architecture for evaluating ai-based security systems at the edge: Network ton iot datasets
N. Moustafa
Published: 2021
Unsw-nb15: a comprehensive data set for network intrusion detection systems (unsw-nb15 network data set)
N. Moustafa, J. Slay
Published: 2015
Ciciot2023: A real-time dataset and benchmark for large-scale attacks in iot environment
Euclides Carlos Pinto Neto, Sajjad Dadkhah, Raphael Ferreira, Alireza Zohourian, Rongxing Lu, Ali A. Ghorbani
Published: 2023
Dïot: A federated self-learning anomaly detection system for iot
Thien Duc Nguyen, Samuel Marchal, Markus Miettinen, Hossein Fereidooni, N Asokan, Ahmad-Reza Sadeghi
Published: 2019
Poisoning attacks on federated learning-based iot intrusion detection system
Nguyen, T.D., Rieger, P., Miettinen, M., Sadeghi, A.R.
Published: 2020
Federated PCA on grassmann manifold for anomaly detection in IoT networks
Nguyen, T.A., He, J., Le, L.T., Bao, W., Tran, N.H.
Published: 2023
Internet of Things (IoT): Opportunities, issues and challenges towards a smart and sustainable future
Nižetić, S., Šolić, P., Gonzalez-De, D.L.d.I., Patrono, L.
Published: 2020
A two-layer dimension reduction and two-tier classification model for anomaly-based intrusion detection in IoT backbone networks
Pajouh, H., Javidan, R., Khayami, R., Dehghantanha, A., Choo, K.
Published: 2019
Federated deep learning for zero-day botnet attack detection in IoT-edge devices
Popoola, S.I., Ande, R., Adebisi, B., Gui, G., Hammoudeh, M., Jogunola, O.
Published: 2022
Federated learning for malware detection in iot devices
V. Rey, P. M. S. Sanchez, A. H. Celdrán, G. Bovet
Published: 2022
Harvesting private medical images in federated learning systems with crafted models
Shi, S., Haque, M.S., Parida, A., Linguraru, M.G., Hou, Y.T., Anwar, S.M., Lou, W.
Published: 2024
Scale-mia: A scalable model inversion attack against secure federated learning via latent space reconstruction
Shi, S., Wang, N., Xiao, Y., Zhang, C., Shi, Y., Hou, Y.T., Lou, W.
Published: 2023
Partner in crime: Boosting targeted poisoning attacks against federated learning
Sun, S., Sugrim, S., Stavrou, A., Wang, H.
Published: 2024
A detailed analysis of the kdd cup 99 data set
Mahbod Tavallaee, Ebrahim Bagheri, Wei Lu, Ali A Ghorbani
Published: 2009
Addressing class imbalance in federated learning
Wang, L., Xu, S., Wang, X., Zhu, Q.
Published: 2021
FeCo: Boosting intrusion detection capability in IoT networks via contrastive learning
Wang, N., Chen, Y., Hu, Y., Lou, W., Hou, Y.T.
Published: 2022
Node selection toward faster convergence for federated learning on non-iid data
Wu, H., Wang, P.
Published: 2022
Personalized federated learning with feature alignment and classifier collaboration
Xu, J., Tong, X., Huang, S.L.
Published: 2023
Fedmix: Approximation of mixup under mean augmented federated learning
Yoon, T., Shin, S., Hwang, S.J., Yang, E.
Published: 2021
Mindfl: Mitigating the impact of imbalanced and noisy-labeled data in federated learning with quality and fairness-aware client selection
Zhang, C., Wang, N., Shi, S., Du, C., Lou, W., Hou, Y.T.
Published: 2023
Federated learning with non-iid data
Zhao, Y., Li, M., Lai, L., Suda, N., Civin, D., Chandra, V.
Published: 2018
Machine Learning Based Network Vulnerability Analysis of Industrial Internet of Things
Maede Zolanvari, Marcio A. Teixeira, Lav Gupta, Khaled M. Khan, Raj Jain
Published: 2019.11.14
Share