Software/Code for "federated Optimization under Intermittent Client Availability"

Yikai Yan,Chaoyue Niu,Yucheng Ding,Zhenzhe Zheng,Shaojie Tang,Qinya Li,Fan Wu,Chengfei Lyu,Yanghe Feng,Guihai Chen
DOI: https://doi.org/10.1287/ijoc.2022.0057.cd
IF: 3.288
2023-01-01
INFORMS Journal on Computing
Abstract:Federated learning is a new distributed machine learning framework, where numerous heterogeneous clients collaboratively train a model without sharing training data. In this work, we consider a practical and ubiquitous issue when deploying federated learning in mobile environments: intermittent client availability, where the set of eligible clients may change during the training process. Such intermittent client availability would seriously dete-riorate the performance of the classical federated averaging algorithm (FedAvg). Thus, we propose a simple distributed nonconvex optimization algorithm, called federated latest aver-aging (FedLaAvg), which leverages the latest gradients of all clients, even when the clients are not available, to jointly update the global model in each iteration. Our theoretical analysis shows that FedLaAvg achieves guaranteed convergence and a sublinear speedup with respect to the total number of clients. We implement FedLaAvg along with several baselines and evaluate them over the benchmarking MNIST and Sentiment140 data sets. The evalua-tion results demonstrate that FedLaAvg achieves more stable training than FedAvg in both convex and nonconvex settings and reaches a sublinear speedup.
What problem does this paper attempt to address?