arrow
Return

Coded Computing for Low-Latency Federated Learning Over Wireless Edge Networks

delete2021-01-01
delete68
delete
OA
AI
S
Saurav Prakash *
S
Sagar Dhakal
M
Mustafa Riza Akdeniz
Y
Yona, Yair
S
Shilpa Talwar
A
Avestimehr, Salman
N
Nageen Himayat
DOI:10.1109/JSAC.2020.3036961delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Federated learning enables training a global model from data located at the client nodes, without data sharing and moving client data to a centralized server. Performance of federated learning in a multi-access edge computing (MEC) network suffers from slow convergence due to heterogeneity and stochastic fluctuations in compute power and communication link qualities across clients. We propose a novel coded computing framework, CodedFedL, that injects structured coding redundancy into federated learning for mitigating stragglers and speeding up the training procedure. CodedFedL enables coded computing for non-linear federated learning by efficiently exploiting distributed kernel embedding via random Fourier features that transforms the training task into computationally favourable distributed linear regression. Furthermore, clients generate local parity datasets by coding over their local datasets, while the server combines them to obtain the global parity dataset. Gradient from the global parity dataset compensates for straggling gradients during training, and thereby speeds up convergence. For minimizing the epoch deadline time at the MEC server, we provide a tractable approach for finding the amount of coding redundancy and the number of local data points that a client processes during training, by exploiting the statistical properties of compute as well as communication delays. We also characterize the leakage in data privacy when clients share their local parity datasets with the server. Additionally, we analyze the convergence rate and iteration complexity of CodedFedL under simplifying assumptions, by treating CodedFedL as a stochastic gradient descent algorithm. Finally, for demonstrating gains that CodedFedL can achieve in practice, we conduct numerical experiments using practical network parameters and benchmark datasets, in which CodedFedL speeds up the overall training time by up to 15x in comparison to the benchmark schemes.
Keywords:
Servers
Training
Collaborative work
Encoding
Distributed databases
Convergence
Redundancy
Distributed computing
machine learning
edge computing
wireless communication
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Journal on Selected Areas in Communications cover
IEEE Journal on Selected Areas in Communications
IF:
17.2
Papers:
6.4K
Citations:
3.1W

Organization

U
university of southern california
Scholars:
4.6W
Papers: 3.8W
Citations: 51
I
Intel Corporation
Scholars:
2.7K
Papers: 2.0K
Citations: 6