Federated Learning Over Energy Harvesting Wireless Networks

Abstract

In this article, the deployment of federated learning (FL) is investigated in an energy harvesting wireless network in which the base stations (BSs) employs massive multiple-input–multiple-output (MIMO) to serve a set of users powered by independent energy harvesting sources. Since a certain number of users may not be able to participate in FL due to interference and energy constraints, a joint energy management and user scheduling problem in FL over wireless systems is formulated. This problem is formulated as an optimization problem whose goal is to minimize the FL training loss via optimizing user scheduling. To find how the transmit power, the number of scheduled users and user association, affect the training loss, the FL convergence rate is first analyzed. Given this analytical result, the original optimization problem can be decomposed, simplified, and solved. Simulation results show that the proposed user scheduling and user association algorithm can reduce training loss compared to a standard FL algorithm.