Straggler Effect Mitigation for Federated Learning in Cell-Free Massive MIMO

Tung T. Vu, Duy T. Ngo, Hien-Quoc Ngo, Minh N. Dao, Nguyen H. Tran, Richard H. Middleton

Research output: Chapter in Book/Report/Conference proceedingConference contribution

6 Downloads (Pure)

Abstract

Straggler effect is the main bottleneck in realizing federated learning (FL) in wireless networks. This work proposes a novel user (UE) selection approach to mitigate this effect with UE sampling in cell-free massive multiple-input multiple-output networks. Our proposed approach selects only a small subset of UEs for participating in one FL process. Importantly, since the UEs are selected before any FL process is executed, the performance of FL during the executing time is not affected by our method. Here, we select UEs by solving an FL transmission time minimization problem that jointly optimizes UE selection, power control, and data rate. The problem is formulated to capture the complex interactions among the FL training time, UE selection, and straggler effect. This mixed-integer mixed-timescale stochastic nonconvex problem is constrained by the minimum number of UEs to guarantee the quality of learning. By employing online successive convex approximation, we propose a novel algorithm to solve the formulated problem with guaranteed convergence to the neighbourhood of their stationary points. Our approach can significantly reduce the FL transmission time over baseline approaches, especially in the networks that experience serious straggler effect due to the moderately low density of access points.
Original languageEnglish
Title of host publication ICC 2021 - IEEE International Conference on Communications: Proceedings
ISBN (Electronic)978-1-7281-7122-7
DOIs
Publication statusPublished - 06 Aug 2021
Externally publishedYes
Event ICC 2021 - IEEE International Conference on Communications - Montreal, Canada
Duration: 14 Jun 202123 Jun 2021

Publication series

NameIEEE International Conference on Communications
ISSN (Print)1550-3607
ISSN (Electronic)1938-1883

Conference

Conference ICC 2021 - IEEE International Conference on Communications
CountryCanada
CityMontreal
Period14/06/202123/06/2021

Fingerprint

Dive into the research topics of 'Straggler Effect Mitigation for Federated Learning in Cell-Free Massive MIMO'. Together they form a unique fingerprint.

Cite this