Skip to main content

Wireless Network Optimization for Federated Learning with Model Compression in Hybrid VLC/RF Systems

Author(s): Huang, Wuwei; Yang, Yang; Chen, Mingzhe; Liu, Chuanhong; Feng, Chunyan; et al

Download
To refer to this page use: http://arks.princeton.edu/ark:/88435/pr1jh3d295
Full metadata record
DC FieldValueLanguage
dc.contributor.authorHuang, Wuwei-
dc.contributor.authorYang, Yang-
dc.contributor.authorChen, Mingzhe-
dc.contributor.authorLiu, Chuanhong-
dc.contributor.authorFeng, Chunyan-
dc.contributor.authorPoor, H Vincent-
dc.date.accessioned2024-02-04T01:12:55Z-
dc.date.available2024-02-04T01:12:55Z-
dc.identifier.citationHuang, Wuwei, Yang, Yang, Chen, Mingzhe, Liu, Chuanhong, Feng, Chunyan, Poor, H Vincent. (Wireless Network Optimization for Federated Learning with Model Compression in Hybrid VLC/RF Systems. Entropy, 23 (11), 1413 - 1413. doi:10.3390/e23111413en_US
dc.identifier.urihttp://arks.princeton.edu/ark:/88435/pr1jh3d295-
dc.description.abstractIn this paper, the optimization of network performance to support the deployment of federated learning (FL) is investigated. In particular, in the considered model, each user owns a machine learning (ML) model by training through its own dataset, and then transmits its ML parameters to a base station (BS) which aggregates the ML parameters to obtain a global ML model and transmits it to each user. Due to limited radio frequency (RF) resources, the number of users that participate in FL is restricted. Meanwhile, each user uploading and downloading the FL parameters may increase communication costs thus reducing the number of participating users. To this end, we propose to introduce visible light communication (VLC) as a supplement to RF and use compression methods to reduce the resources needed to transmit FL parameters over wireless links so as to further improve the communication efficiency and simultaneously optimize wireless network through user selection and resource allocation. This user selection and bandwidth allocation problem is formulated as an optimization problem whose goal is to minimize the training loss of FL. We first use a model compression method to reduce the size of FL model parameters that are transmitted over wireless links. Then, the optimization problem is 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 ultimate user selection and bandwidth allocation are obtained by iteratively compressing the model and solving these two subproblems. Simulation results show that the proposed FL algorithm can improve the accuracy of object recognition by up to 16.7% and improve the number of selected users by up to 68.7%, compared to a conventional FL algorithm using only RF.en_US
dc.languageenen_US
dc.language.isoen_USen_US
dc.relation.ispartofEntropyen_US
dc.rightsFinal published version. This is an open access article.en_US
dc.titleWireless Network Optimization for Federated Learning with Model Compression in Hybrid VLC/RF Systemsen_US
dc.typeJournal Articleen_US
dc.identifier.doidoi:10.3390/e23111413-
dc.date.eissued2021-10-27en_US
dc.identifier.eissn1099-4300-
pu.type.symplectichttp://www.symplectic.co.uk/publications/atom-terms/1.0/journal-articleen_US

Files in This Item:
File Description SizeFormat 
entropy-23-01413-v2.pdf648.17 kBAdobe PDFView/Download


Items in OAR@Princeton are protected by copyright, with all rights reserved, unless otherwise indicated.