arrow
Return

Joint Client Selection and Bandwidth Allocation Algorithm for Federated Learning

delete2023-06-01
delete31
PRE
AI
H
Haneul Ko
J
Jaewook Lee
S
Sangwon Seo
S
Sangheon Pack *
V
Victor C. M. Leung
DOI:10.1109/TMC.2021.3136611delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In federated learning (FL), if the participating mobile devices have low computing power and poor wireless channel conditions and/or they do not have sufficient data for various classes, a long convergence time is required to achieve the desired model accuracy. To address this problem, we first formulate a constrained Markov decision process (CMDP) problem that aims to minimize the average time of rounds while maintaining the numbers of trained data and trained data classes above certain numbers. To obtain the optimal scheduling policy, the formulated CMDP problem is converted into an equivalent linear programming (LP). Additionally, to overcome the problem of the curse of dimensionality in CMDP, we develop a joint client selection and bandwidth allocation algorithm (JCSBA) that jointly selects appropriate mobile devices and allocates suitable amount of bandwidth to them at each round by considering their data information, computing power, and channel gain. Evaluation results validate that J-CSBA can reduce the convergence time by up to 49% compared to a conventional random scheme.
Keywords:
Mobile handsets
Servers
Data models
Convergence
Computational modeling
Channel allocation
Bandwidth
Federated learning (FL)
joint optimization
client selection
bandwidth allocation
convergence time
constrained Markov decision process (CMDP)

Journal

IEEE Transactions on Mobile Computing cover
IEEE Transactions on Mobile Computing
IF:
9.2
Papers:
5.9K
Citations:
1.8W

Organization

K
Korea University
Scholars:
3.6W
Papers: 3.8W
Citations: 4.4W
S
shenzhen university
Scholars:
4.6W
Papers: 3.4W
Citations: 72
Cited Papers

Cited Papers

Ferroelectric Behavior of Thiourea
err1959-11-01
err0
PREAI
errG. J. Goldsmith; J. G. White
errShare
errSave
Intelligence and Learning in O-RAN for Data-Driven NextG Cellular Networks
err2021-10-01
err144
errOAAI
errBonati, Leonardo; D'Oro, Salvatore; Polese, Michele; Basagni, Stefano; Melodia, Tommaso
errShare
errSave
Model Pruning Enables Efficient Federated Learning on Edge Devices
err2023-12-01
err160
errOAAI
errJiang, Yuang; Wang, Shiqiang; Valls, Victor; Ko, Bong Jun; Lee, Wei-Han; Leung, Kin K.; Tassiulas, Leandros
errShare
errSave
Jointly Optimizing Client Selection and Resource Management in Wireless Federated Learning for Internet of Things
err2022-03-15
err73
PREAI
errYu, Liangkun; Albelaihi, Rana; Sun, Xiang; Ansari, Nirwan; Devetsikiotis, Michael
errShare
errSave
Binding theory and grammatical specific language impairment in children
err1997-03-01
err0
PREAI
errHeather K.J van der Lely; Linda Stollwerck
errShare
errSave
researcher View more