Residential College | false |
Status | 已發表Published |
Dynamic User-Scheduling and Power Allocation for SWIPT Aided Federated Learning: A Deep Learning Approach | |
Li Yang1,4; Wu Yuan1,4,5; Song Yuxiao1,4; Qian Liping2; Jia Weijia3 | |
2023-12 | |
Source Publication | IEEE Transactions on Mobile Computing |
ISSN | 1536-1233 |
Volume | 22Issue:12Pages:6956-6969 |
Abstract | Federated learning (FL) has been considered as a promising paradigm for enabling distributed machine learning (ML) in wireless networks. To address the limited energy capacity of wireless devices, we propose a simultaneous wireless information and power transfer (SWIPT) aided FL, in which one FL server (FLS) co-located at a cellular base station (BS) uses SWIPT to simultaneously broadcast the global model to wireless user-devices (UDs) and provide wireless power transfer to them. The UDs then use the harvested energy to train their local models and further transmit the local models to the FLS for aggregation. To improve the spectrum efficiency, we consider that the UDs form a non-orthogonal multiple access (NOMA) group for simultaneously sending their local models over the same spectrum channel. Taking the UDs' time-varying available energy and channel conditions into account, we propose a dynamic optimization of the UDs-scheduling, the BS's transmit-power allocation, and the UDs' power-splitting factors for SWIPT, with the objective of minimizing the long-term energy consumption while ensuring the FL convergence. The optimization problem, however, is challenging to solve since it is a finite-horizon dynamic programming problem but with an unknown stopping time, and moreover, the action space covers both discrete and continuous variables. To address these difficulties, we first execute a series of equivalent transformations to reduce the number of decision variables and then formulate the problem as a stochastic shortest path problem, based on which we propose an actor-critic deep reinforcement learning algorithm with the proximal policy optimization to efficiently learn the policy that dynamically adjusts the UDs-scheduling for FL as well as the BS's transmit-power for SWIPT. Numerical results validate the effectiveness and performance of our proposed algorithm. The results demonstrate that our proposed algorithm can effectively reduce the long-term energy consumption in comparison with two baseline algorithms. |
Keyword | Federated Learning Swipt Dynamic User-scheduling Stochastic Shortest Path Deep Reinforcement Learning Proximal Policy Optimization |
DOI | 10.1109/TMC.2022.3201622 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science ; Telecommunications |
WOS Subject | Computer Science, Information Systems ; Telecommunications |
WOS ID | WOS:001098818300007 |
Publisher | IEEE COMPUTER SOC10662 LOS VAQUEROS CIRCLE, PO BOX 3014, LOS ALAMITOS, CA 90720-1314 |
Scopus ID | 2-s2.0-85137594239 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) |
Corresponding Author | Wu Yuan |
Affiliation | 1.State Key Laboratory of Internet of Things for Smart City, University of Macau, Macau, China 2.College of Information Engineering, Zhejiang University of Technology, Hangzhou, China 3.Institute of Artificial Intelligence and Future Networks, Beijing Normal University, Zhuhai, China 4.Department of Computer Information Science, University of Macau 5.Zhuhai-UM Science and Technology Research Institute, Zhuhai 519031, China. |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Li Yang,Wu Yuan,Song Yuxiao,et al. Dynamic User-Scheduling and Power Allocation for SWIPT Aided Federated Learning: A Deep Learning Approach[J]. IEEE Transactions on Mobile Computing, 2023, 22(12), 6956-6969. |
APA | Li Yang., Wu Yuan., Song Yuxiao., Qian Liping., & Jia Weijia (2023). Dynamic User-Scheduling and Power Allocation for SWIPT Aided Federated Learning: A Deep Learning Approach. IEEE Transactions on Mobile Computing, 22(12), 6956-6969. |
MLA | Li Yang,et al."Dynamic User-Scheduling and Power Allocation for SWIPT Aided Federated Learning: A Deep Learning Approach".IEEE Transactions on Mobile Computing 22.12(2023):6956-6969. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment