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.

User Cooperation with Power Control for Federated Learning in CFmMIMO Networks.

Joint User Cooperation and Scheduling for Federated Learning in CFmMIMO Networks

Delay-Aware Power Control for Downlink Multi-User MIMO Via Constrained Deep Reinforcement Learning.

Joint BS-User Association, Power Allocation, and User-Side Interference Cancellation in Cell-free Heterogeneous Networks

Joint User Association and Power Control for Cell-Free Massive MIMO

Joint Device Selection and Power Control for Wireless Federated Learning

Resource Allocation for Multi-Task Federated Learning Algorithm over Wireless Communication Networks

Joint User Pairing, Resource Block Allocation, And Power Control For Full-Duplex Cellular Networks

Joint Energy and Latency Optimization in Federated Learning over Cell-Free Massive MIMO Networks

Joint Time Allocation and Power Control in Multicell Networks with Load Coupling: Energy Saving and Rate Improvement.

Power-Efficient Transmission for User-Centric Networks with Limited Fronthaul Capacity and Computation Resource

Joint Power Control and Data Size Selection for Over-the-Air Computation Aided Federated Learning

Energy-Efficient Optimization in User-Centric Cell-Free Massive MIMO Systems for URLLC with Finite Blocklength Communications

Joint Client Selection and CPU Frequency Control in Wireless Federated Learning Networks with Power Constraints

Dynamic User-Scheduling and Power Allocation for SWIPT Aided Federated Learning: A Deep Learning Approach

Theoretical Analysis of Federated Learning Supported by Cell-Free Massive MIMO Networks with Enhanced Power Allocation

Joint Duplex Mode Selection, Channel Allocation, and Power Control for Full-Duplex Cognitive Femtocell Networks

Deep reinforcement learning based joint cooperation clustering and downlink power control for cell-free massive MIMO

Joint Power Control and User Association for NOMA-Based Full-Duplex Systems

Optimized Device Selection and Power Control for Wireless Federated Learning.