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.

Joint User Scheduling and Resource Allocation for Federated Learning over Wireless Networks

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

Joint Device Scheduling and Resource Allocation for Latency Constrained Wireless Federated Learning

Mobility-Aware Joint User Scheduling and Resource Allocation for Low Latency Federated Learning

Asynchronous Federated Learning over Wireless Communication Networks.

Joint User Association and Resource Allocation for Wireless Hierarchical Federated Learning with IID and Non-IID Data

A Joint Learning and Communications Framework for Federated Learning over Wireless Networks

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

Scheduling Policies for Federated Learning in Wireless Networks:An Overview

Joint Scheduling of Participants, Local Iterations, and Radio Resources for Fair Federated Learning over Mobile Edge Networks

User Scheduling for Federated Learning Through Over-the-Air Computation

Federated Learning Over Wireless Channels: Dynamic Resource Allocation and Task Scheduling

Joint User Selection and Resource Allocation for Fast Federated Edge Learning

Federated Learning with Unsourced Random Access.

Device Scheduling and Resource Allocation for Federated Learning under Delay and Energy Constraints

Scheduling Policies for Federated Learning in Wireless Networks

User Assignment and Resource Allocation for Hierarchical Federated Learning over Wireless Networks

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

Federated Learning over Energy Harvesting Wireless Networks

Efficient Device Scheduling with Multi-Job Federated Learning

Joint Accuracy and Resource Allocation for Green Federated Learning Networks