.png){kind=icon}
.png){kind=icon}
Return
FedMDS: An Efficient Model Discrepancy-Aware Semi-Asynchronous Clustered Federated Learning Framework
.png){kind=icon}
PRE
AI
DOI：10.1109/TPDS.2023.3237752.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Federated learning (FL) is an emerging distributed machine learning paradigm that protects privacy and tackles the problem of isolated data islands. At present, there are two main communication strategies of FL: synchronous FL and asyn-chronous FL. The advantages of synchronous FL are the high precision and easy convergence of the model. However, this synchronous communication strategy has the risk of the straggler effect. Asyn-chronous FL has a natural advantage in mitigating the straggler effect, but there are threats of model quality degradation and server crash. In this paper, we propose a model discrepancy-aware semi-asynchronous clustered FL framework, FedMDS, which al-leviates the straggler effect by 1) a clustered strategy based on the delay and direction of the model update and 2) a synchronous trigger mechanism that limits the model staleness. FedMDS lever-ages the clustered algorithm to reschedule the clients. Each group of clients performs asynchronous updates until the synchronous update mechanism based on the model discrepancy is triggered. We evaluate FedMDS based on four typical federated datasets in a non-IID setting and compare FedMDS to the baselines. The exper-imental results show that FedMDS significantly improves average test accuracy by more than +9.2% on the four datasets compared to TA-FedAvg. In particular, FedMDS improves absolute Top-1 test accuracy by +37.6% on FEMNIST compared to TA-FedAvg. The frequency of the average synchronization waiting time of FedMDS is significantly lower than that of TA-FedAvg on all datasets. More-over, FedMDS can improve the accuracy and alleviate the straggler effect.
Keywords:
Servers
Training
Federated learning
Task analysis
Training data
Optimization
Data models
Distributed machine learning
federated learning
neural network
Journal
IF:
6
Papers:
5.2K
Citations:
1.1W