Loading [a11y]/accessibility-menu.js
Partial Synchronization to Accelerate Federated Learning Over Relay-Assisted Edge Networks | IEEE Journals & Magazine | IEEE Xplore

Partial Synchronization to Accelerate Federated Learning Over Relay-Assisted Edge Networks


Abstract:

Federated Learning (FL) is a promising machine learning paradigm to cooperatively train a global model with highly distributed data located on mobile devices. Aiming to o...Show More

Abstract:

Federated Learning (FL) is a promising machine learning paradigm to cooperatively train a global model with highly distributed data located on mobile devices. Aiming to optimize the communication efficiency for gradient aggregation and model synchronization among large-scale devices, we propose a relay-assisted FL framework. By breaking the traditional transmission-order constraint and exploiting the broadcast characteristic of relay nodes, we design a novel synchronization scheme named Partial Synchronization Parallel (PSP), in which models and gradients are transmitted simultaneously and aggregated at relay nodes, resulting in traffic reduction. We prove that PSP has the same convergence rate as the sequential synchronization approaches via rigorous analysis. To further accelerate the training process, we integrate PSP with any unbiased and error-bounded compression technologies and prove that the convergence properties of the resulting scheme still hold. Extensive experiments are conducted in a distributed cluster environment with real-world datasets and the results demonstrate that our proposed approach reduces the training time up to 37 percent compared to state-of-the-art methods.
Published in: IEEE Transactions on Mobile Computing ( Volume: 21, Issue: 12, 01 December 2022)
Page(s): 4502 - 4516
Date of Publication: 24 May 2021

ISSN Information:

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.