A Robust Federated Learning Approach for Combating Attacks Against IoT Systems Under non-IID Challenges

TL;DR

This paper evaluates federated learning algorithms like FedAvg, FedProx, and Scaffold for IoT attack detection under non-IID data challenges, providing insights into their effectiveness in resource-constrained, security-sensitive environments.

Contribution

It offers a comprehensive comparison of federated learning methods for IoT attack detection under non-IID data distributions, addressing a gap in existing research.

Findings

FedProx outperforms other methods under non-IID conditions.

Scaffold shows improved convergence in heterogeneous data scenarios.

Federated learning effectively detects IoT attacks while preserving privacy.

Abstract

In the context of the growing proliferation of user devices and the concurrent surge in data volumes, the complexities arising from the substantial increase in data have posed formidable challenges to conventional machine learning model training. Particularly, this is evident within resource-constrained and security-sensitive environments such as those encountered in networks associated with the Internet of Things (IoT). Federated Learning has emerged as a promising remedy to these challenges by decentralizing model training to edge devices or parties, effectively addressing privacy concerns and resource limitations. Nevertheless, the presence of statistical heterogeneity in non-Independently and Identically Distributed (non-IID) data across different parties poses a significant hurdle to the effectiveness of FL. Many FL approaches have been proposed to enhance learning effectiveness under statistical heterogeneity. However, prior studies have uncovered a gap in the existing research landscape, particularly in the absence of a comprehensive comparison between federated methods addressing statistical heterogeneity in detecting IoT attacks. In this research endeavor, we delve into the exploration of FL algorithms, specifically FedAvg, FedProx, and Scaffold, under different data distributions. Our focus is on achieving a comprehensive understanding of and addressing the challenges posed by statistical heterogeneity. In this study, We classify large-scale IoT attacks by utilizing the CICIoT2023 dataset. Through meticulous analysis and experimentation, our objective is to illuminate the performance nuances of these FL methods, providing valuable insights for researchers and practitioners in the domain.