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

by   Mingzhe Chen, et al.

In this paper, the problem of training federated learning (FL) algorithms over a realistic wireless network is studied. In particular, in the considered model, wireless users execute an FL algorithm while training their local FL models using their own data and transmitting the trained local FL models to a base station (BS) that will generate a global FL model and send it back to the users. Since all training parameters are transmitted over wireless links, the quality of the training will be affected by wireless factors such as packet errors and the availability of wireless resources. Meanwhile, due to the limited wireless bandwidth, the BS must select an appropriate subset of users to execute the FL algorithm so as to build a global FL model accurately. This joint learning, wireless resource allocation, and user selection problem is formulated as an optimization problem whose goal is to minimize an FL loss function that captures the performance of the FL algorithm. To address this problem, a closed-form expression for the expected convergence rate of the FL algorithm is first derived to quantify the impact of wireless factors on FL. Then, based on the expected convergence rate of the FL algorithm, the optimal transmit power for each user is derived, under a given user selection and uplink resource block (RB) allocation scheme. Finally, the user selection and uplink RB allocation is optimized so as to minimize the FL loss function. Simulation results show that the proposed joint federated learning and communication framework can reduce the FL loss function value by up to 10 16 random resource allocation and 2) a standard FL algorithm with random user selection and resource allocation.


page 1

page 2

page 3

page 4


Convergence Time Optimization for Federated Learning over Wireless Networks

In this paper, the convergence time of federated learning (FL), when dep...

Federated Learning over Energy Harvesting Wireless Networks

In this paper, the deployment of federated learning (FL) is investigated...

Joint Age-based Client Selection and Resource Allocation for Communication-Efficient Federated Learning over NOMA Networks

In federated learning (FL), distributed clients can collaboratively trai...

Optimization of User Selection and Bandwidth Allocation for Federated Learning in VLC/RF Systems

Limited radio frequency (RF) resources restrict the number of users that...

Federated Learning over Wireless IoT Networks with Optimized Communication and Resources

To leverage massive distributed data and computation resources, machine ...

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

As an efficient distributed machine learning approach, Federated learnin...

Age of Information in Federated Learning over Wireless Networks

In this paper, federated learning (FL) over wireless networks is investi...

Please sign up or login with your details

Forgot password? Click here to reset