Abstract:To exploit massive amounts of data generated at mobile edge networks, federated learning (FL) has been proposed as an attractive substitute for centralized machine learning (ML). By collaboratively training a shared learning model at edge devices, FL avoids direct data transmission and thus overcomes high communication latency and privacy issues as compared to centralized ML. To improve the communication efficiency in FL model aggregation, over-the-air computation has been introduced to support a large number of simultaneous local model uploading by exploiting the inherent superposition property of wireless channels. However, due to the heterogeneity of communication capacities among edge devices, over-the-air FL suffers from the straggler issue in which the device with the weakest channel acts as a bottleneck of the model aggregation performance. This issue can be alleviated by device selection to some extent, but the latter still suffers from a tradeoff between data exploitation and model communication. In this paper, we leverage the reconfigurable intelligent surface (RIS) technology to relieve the straggler issue in over-the-air FL. Specifically, we develop a learning analysis framework to quantitatively characterize the impact of device selection and model aggregation error on the convergence of over-the-air FL. Then, we formulate a unified communication-learning optimization problem to jointly optimize device selection, over-the-air transceiver design, and RIS configuration. Numerical experiments show that the proposed design achieves substantial learning accuracy improvement compared with the state-of-the-art approaches, especially when channel conditions vary dramatically across edge devices.

Over-The-Air Federated Learning in User-Centric Cell-Free Networks

Deep Learning Based Coded Over-the-Air Computation for Personalized Federated Learning

Over-the-Air Federated Learning with Joint Adaptive Computation and Power Control

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

Over-The-Air Federated Learning Over Scalable Cell-free Massive MIMO

Federated Learning with Unsourced Random Access.

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

Over-The-Air Federated Learning: Status Quo, Open Challenges, and Future Directions

RIS-Assisted Over-the-Air Adaptive Federated Learning with Noisy Downlink

Enhancement of Over-the-Air Federated Learning by Using AI-based Fluid Antenna System

Hierarchical Over-the-Air Federated Edge Learning

ROAR-Fed: RIS-Assisted Over-the-Air Adaptive Resource Allocation for Federated Learning

Deep Reinforcement Learning for Multi-Functional RIS-Aided Over-the-Air Federated Learning in Internet of Robotic Things

Deep Reinforcement Learning for Energy Efficiency Maximization in SWIPT-Based Over-the-Air Federated Learning

Over-the-Air Federated Edge Learning with Hierarchical Clustering

Over-the-air Learning Rate Optimization for Federated Learning

Personalized Over-the-Air Federated Learning with Personalized Reconfigurable Intelligent Surfaces

Over-the-Air Hierarchical Personalized Federated Learning

Over-the-Air Federated Learning and Optimization

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

Over-The-Air Clustered Wireless Federated Learning