User Scheduling in Federated Learning over Energy Harvesting Wireless Networks

Abstract

In this paper, the deployment of federated learning (FL) is investigated in an energy harvesting wireless network in which the base station (BS) is equipped with a massive multiple-input multiple-output (MIMO) system and a set of users powered by independent energy harvesting sources to cooperatively perform FL. Since a certain number of users may not be served due to interference and energy constraints, a joint energy management and user scheduling problem is considered. This problem is formulated as an optimization problem whose goal is to minimize the FL training loss via optimizing user scheduling. To determine the effect of various wireless factors (transmit power and number of scheduled users) on training loss, the convergence rate of the FL algorithm is analyzed. Given this analytical result, the original user scheduling and energy management optimization problem can be decomposed, simplified and solved. Simulation results show that the proposed algorithm can reduce training loss compared to a standard FL algorithm.