Abstract: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% and 16%, respectively, compared to: 1) An optimal user selection algorithm with random resource allocation and 2) a standard FL algorithm with random user selection and resource allocation.

Joint User Cooperation and Scheduling for Federated Learning in CFmMIMO Networks

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

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

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

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

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

Joint User Scheduling and Channel Allocation for Cellular Networks with Full Duplex Base Stations.

Joint User Scheduling and Beamforming Design for Multiuser MISO Downlink Systems

Joint Receiver Design and User Scheduling for Over-the-Air Aggregation in Federated Learning

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

Joint User Scheduling and Beam Selection Optimization for Beam-Based Massive MIMO Downlinks

Joint User Pairing and Beamforming Design for NOMA-Aided CFMM-ISAC Systems

Neighbor-based joint spatial division and multiplexing in massive MIMO: user scheduling and dynamic beam allocation

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

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

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

A Dynamic Scheduling Scheme For Comp In Downlink Fd-Mimo Transmission

Cooperative Caching and Beamforming for CoMP-Enabled Mobile Edge Networks

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