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 proceeding contributionpeer-review

13 Citations (Scopus)

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 publicationICC 2021 - IEEE International Conference on Communications
Subtitle of host publicationproceedings
Place of PublicationPiscataway, NJ
PublisherInstitute of Electrical and Electronics Engineers (IEEE)
Number of pages6
ISBN (Electronic)9781728171227
ISBN (Print)9781728171234
DOIs
Publication statusPublished - 2021
Externally publishedYes
Event2021 IEEE International Conference on Communications, ICC 2021 - Virtual, Online, Canada
Duration: 14 Jun 202123 Jun 2021

Publication series

Name
ISSN (Print)1550-3607
ISSN (Electronic)1938-1883

Conference

Conference2021 IEEE International Conference on Communications, ICC 2021
Country/TerritoryCanada
CityVirtual, Online
Period14/06/2123/06/21

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