Grouped Federated Learning Algorithm Based on Non-IID Data

Zhongxin Li,Jianfei Zhang
DOI: https://doi.org/10.1145/3650215.3650331
2024-01-01
Abstract:Federated learning is a new machine learning paradigm in which multiple clients collaborate to train a machine learning model while protecting local data privacy. Client-side non-identically and non-independently distributed (Non-IID) data is one of the most important challenges for federated learning algorithms. The classification accuracy of the federated learning model drops drastically due to the weight dispersion caused by data heterogeneity. In this work, we adopt a way of dividing client data that is closer to real applications. We propose a federated learning algorithm based on rearranging all clients(FedRC). FedRC sets up a novel within-group training approach and a server-side pre-aggregation policy to address the impact of Non-IID on the model. In addition, we test FedRC on a simulated dataset and the experimental results demonstrate that FedRC outperforms the baseline algorithm and significantly improves the performance on Non-IID data.
What problem does this paper attempt to address?