MSSA-FL: High-Performance Multi-stage Semi-asynchronous Federated Learning with Non-IID Data.

Xiaohui Wei,Mingkai Hou,Chenghao Ren,Xiang Li,Hengshan Yue
DOI: https://doi.org/10.1007/978-3-031-10986-7_14
2022-01-01
Abstract:Federated Learning (FL) is an emerging distributed machine learning framework that allows edge devices to collaborative train a shared global model without transmitting their sensitive data to centralized servers. However, it is extremely challenging to apply FL in practical scenarios because the statistics of the data across edge devices are usually not independent and identically distributed (Non-HD), which will introduce the bias to global model. To solve the above data heterogeneity issue, we propose a novel Multi-Stage Semi-Asynchronous Federated Learning (MSSA-FL) framework. MSSA-FL benefits convergence accuracy through making the local model complete multi-stage training within the group guided by combination module. To improve the training efficiency of the framework, MSSA-FL adopts a semi-asynchronous update method. Meanwhile, proposed model assignment strategy and model aggregation method further boost the performance of MSSA-FL. Experiments on several public datasets show that MSSA-FL achieves higher accuracy and faster convergence than the comparison algorithms.
What problem does this paper attempt to address?