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

by   Hang Liu, et al.

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 improvements compared with the state-of-the-art approaches, especially when channel conditions vary dramatically across edge devices.


page 1

page 2

page 3

page 4


Over-the-Air Multi-Task Federated Learning Over MIMO Interference Channel

With the explosive growth of data and wireless devices, federated learni...

Federated Learning for Physical Layer Design

Model-free techniques, such as machine learning (ML), have recently attr...

Performance-Oriented Design for Intelligent Reflecting Surface Assisted Federated Learning

To efficiently exploit the massive raw data that is pervading generated ...

Robust Design of Federated Learning for Edge-Intelligent Networks

Mass data traffics, low-latency wireless services and advanced artificia...

CSIT-Free Federated Edge Learning via Reconfigurable Intelligent Surface

We study over-the-air model aggregation in federated edge learning (FEEL...

Over-the-Air Federated Learning via Second-Order Optimization

Federated learning (FL) is a promising learning paradigm that can tackle...

Relay-Assisted Cooperative Federated Learning

Federated learning (FL) has recently emerged as a promising technology t...

Please sign up or login with your details

Forgot password? Click here to reset