Pipelined Training and Transmission for Federated Learning

Yihui Shen,Hong Zhang,Juncheng Jia,Tao Deng,Zhenhua Wang,Zhiwei Teng,Mianxiong Dong
DOI: https://doi.org/10.1109/icpads60453.2023.00072
2023-01-01
Abstract:Federated learning is a distributed machine learning framework that enables distributed edge devices to jointly train machine learning models without exchanging private data. Because federated learning can leverage edge devices’ data and computing resources, the research on federated learning has set off an upsurge. However, one of major challenges faced by federated learning is the limited computation and communication resources of edge devices, which leads to low efficiency of federated learning. In this paper, we propose a federated learning framework FedPipeline, which focuses on the local training and uploading process to improve the convergence efficiency. Different from the existing federated learning algorithms with disjoint sequential steps of training and uploading for devices, FedPipeline exploits the parallelism of training and uploading that devices upload part of the model updates in advance while they are conducting local training simultaneously. In this way, the computation and communication resources of devices can be better utilized. We further design strategies for the scheduling of early uploading and the selection of uploaded updates to maximize the benefit of FedPipeline. Experimental results show that FedPipeline can effectively accelerate the convergence speed of federated learning and preserve the accuracy of the final model.
What problem does this paper attempt to address?