Abstract:We consider a broadband over-the-air computation empowered model aggregation approach for wireless federated learning (FL) systems and propose to leverage an intelligent reflecting surface (IRS) to combat wireless fading and noise. We first investigate the conventional node-selection based framework, where a few edge nodes are dropped in model aggregation to control the aggregation error. We analyze the performance of this node-selection based framework and derive an upper bound on its performance loss, which is shown to be related to the selected edge nodes. Then, we seek to minimize the mean-squared error (MSE) between the desired global gradient parameters and the actually received ones by optimizing the selected edge nodes, their transmit equalization coefficients, the IRS phase shifts, and the receive factors of the cloud server. By resorting to the matrix lifting technique and difference-of-convex programming, we successfully transform the formulated optimization problem into a convex one and solve it using off-the-shelf solvers. To improve learning performance, we further propose a weight-selection based FL framework. In such a framework, we assign each edge node a proper weight coefficient in model aggregation instead of discarding any of them to reduce the aggregation error, i.e., amplitude alignment of the received local gradient parameters from different edge nodes is not required. We also analyze the performance of this weight-selection based framework and derive an upper bound on its performance loss, followed by minimizing the MSE via optimizing the weight coefficients of the edge nodes, their transmit equalization coefficients, the IRS phase shifts, and the receive factors of the cloud server. Furthermore, we use the MNIST dataset for simulations to evaluate the performance of both node-selection and weight-selection based FL frameworks.

Federated Edge Learning Via Reconfigurable Intelligent Surface with One-Bit Quantization

One Bit Aggregation for Federated Edge Learning with Reconfigurable Intelligent Surface: Analysis and Optimization

Federated Learning in Multi-RIS-Aided Systems

Reconfigurable Intelligent Surface Empowered Over-the-Air Federated Edge Learning

One-Bit Over-the-Air Aggregation for Communication-Efficient Federated Edge Learning: Design and Convergence Analysis

Reconfigurable Intelligent Surface Enabled Federated Learning: A Unified Communication-Learning Design Approach

Reconfigurable Intelligent Surface Empowered Federated Edge Learning with Statistical CSI

RIS-Aided Federated Edge Learning Exploiting Statistical CSI

Training Time Minimization for Federated Edge Learning with Optimized Gradient Quantization and Bandwidth Allocation

CSIT-Free Model Aggregation for Federated Edge Learning Via Reconfigurable Intelligent Surface

FedAQ: Communication-Efficient Federated Edge Learning via Joint Uplink and Downlink Adaptive Quantization

Federated Spectrum Learning for Reconfigurable Intelligent Surfaces-Aided Wireless Edge Networks

Over-the-Air Federated Edge Learning with Error-Feedback One-Bit Quantization and Power Control

Resource Allocation for Wireless Federated Edge Learning based on Data Importance

Joint Communication-Learning Design for RIS-Assisted Federated Learning

Joint Optimization of Bandwidth Allocation and Gradient Quantization for Federated Edge Learning

Gradient and Channel Aware Dynamic Scheduling for Over-the-Air Computation in Federated Edge Learning Systems

Adaptive Personalized Over-the-Air Federated Learning with Reflecting Intelligent Surfaces

IRS Assisted Federated Learning A Broadband Over-the-Air Aggregation Approach

IRS Assisted Federated Learning: A Broadband Over-the-Air Aggregation Approach

Energy-Efficient Radio Resource Allocation for Federated Edge Learning