Optimization of user selection and bandwidth allocation for federated learning in VLC/RF systems

Chuanhong Liu, Caili Guo, Yang Yang, Mingzhe Chen, H. Vincent Poor, Shuguang Cui

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

10 Scopus citations

Abstract

Limited radio frequency (RF) resources restrict the number of users that can participate in federated learning (FL) thus affecting FL convergence speed and performance. In this paper, we first introduce visible light communication (VLC) as a supplement to RF in FL and build a hybrid VLC/RF communication system, in which each indoor user can use both VLC and RF to transmit its FL model parameters. Then, the problem of user selection and bandwidth allocation is studied for FL implemented over a hybrid VLC/RF system aiming to optimize the FL performance. The problem is first separated into two subproblems. The first subproblem is a user selection problem with a given bandwidth allocation, which is solved by a traversal algorithm. The second subproblem is a bandwidth allocation problem with a given user selection, which is solved by a numerical method. The final user selection and bandwidth allocation are obtained by iteratively solving these two subproblems. Simulation results show that the proposed FL algorithm that efficiently uses VLC and RF for FL model transmission can improve the prediction accuracy by up to 10% compared with a conventional FL system using only RF.

Original languageEnglish (US)
Title of host publication2021 IEEE Wireless Communications and Networking Conference, WCNC 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781728195056
DOIs
StatePublished - 2021
Externally publishedYes
Event2021 IEEE Wireless Communications and Networking Conference, WCNC 2021 - Nanjing, China
Duration: Mar 29 2021Apr 1 2021

Publication series

NameIEEE Wireless Communications and Networking Conference, WCNC
Volume2021-March
ISSN (Print)1525-3511

Conference

Conference2021 IEEE Wireless Communications and Networking Conference, WCNC 2021
Country/TerritoryChina
CityNanjing
Period3/29/214/1/21

All Science Journal Classification (ASJC) codes

  • General Engineering

Fingerprint

Dive into the research topics of 'Optimization of user selection and bandwidth allocation for federated learning in VLC/RF systems'. Together they form a unique fingerprint.

Cite this