P-FedAvg: Parallelizing Federated Learning with Theoretical Guarantees

التفاصيل البيبلوغرافية
العنوان: P-FedAvg: Parallelizing Federated Learning with Theoretical Guarantees
المؤلفون: Quan Z. Sheng, Min Chen, Yipeng Zhou, Xu Chen, Zhicong Zhong, Chao Li, Di Wu
المصدر: INFOCOM
بيانات النشر: IEEE, 2021.
سنة النشر: 2021
مصطلحات موضوعية: Mathematical optimization, Rate of convergence, Computer science, Heuristic, Robustness (computer science), Convergence (routing), Fraction (mathematics), Construct (python library), Limit (mathematics), Network topology
الوصف: With the growth of participating clients, the centralized parameter server (PS) will seriously limit the scale and efficiency of Federated Learning (FL). A straightforward approach to scale up the FL system is to construct a Parallel FL (PFL) system with multiple PSes. However, it is unclear whether PFL can really achieve a faster convergence rate or not. Even if the answer is yes, it is non-trivial to design a highly efficient parameter average algorithm for a PFL system. In this paper, we propose a completely parallelizable FL algorithm called P-FedAvg under the PFL architecture. P-FedAvg extends the well-known FedAvg algorithm by allowing multiple PSes to cooperate and train a learning model together. In P-FedAvg, each PS is only responsible for a fraction of total clients, but PSes can mix model parameters in a dedicatedly designed way so that the FL model can well converge. Different from heuristic-based algorithms, P-FedAvg is with theoretical guarantees. To be rigorous, we conduct theoretical analysis on the convergence rate of P-FedAvg, and derive the optimal weights for each PS to mix parameters with its neighbors. We also examine how the overlay topology formed by PSes affects the convergence rate and robustness of a PFL system. Lastly, we perform extensive experiments with real datasets to verify our analysis and demonstrate that P-FedAvg can significantly improve convergence rates than traditional FedAvg and other competitive baselines. We believe that our work can help to lay a theoretical foundation for building more efficient PFL systems.
DOI: 10.1109/infocom42981.2021.9488877
URL الوصول: https://explore.openaire.eu/search/publication?articleId=doi_________::f216f5311685f4fe8d9e433eb9a41c81
https://doi.org/10.1109/infocom42981.2021.9488877
Rights: CLOSED
رقم الانضمام: edsair.doi...........f216f5311685f4fe8d9e433eb9a41c81
قاعدة البيانات: OpenAIRE
الوصف
DOI:10.1109/infocom42981.2021.9488877