vix.ing · top · new · best · stats

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

2021/08/05 by Xiang Ma, Haijian Sun, Ma, Xiang +5
Computer Science · Engineering · #Advanced Wireless Communication Technologies #Artificial Intelligence (cs.AI) #FOS: Computer and information sciences #FOS: Electrical engineering #Machine Learning (cs.LG) #Privacy-Preserving Technologies in Data #Signal Processing (eess.SP) #Stochastic Gradient Optimization Techniques #cs.AI #cs.LG #eess.SP #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2108.02891

arxiv created 2021/08/05 · openalex publication_date 2021/08/05 · arxiv updated 2021/08/09 · openalex created_date 2025/10/10 · openalex updated_date 2026/08/01

Abstract

A new machine learning (ML) technique termed as federated learning (FL) aims to preserve data at the edge devices and to only exchange ML model parameters in the learning process. FL not only reduces the communication needs but also helps to protect the local privacy. Although FL has these advantages, it can still experience large communication latency when there are massive edge devices connected to the central parameter server (PS) and/or millions of model parameters involved in the learning process. Over-the-air computation (AirComp) is capable of computing while transmitting data by allowing multiple devices to send data simultaneously by using analog modulation. To achieve good performance in FL through AirComp, user scheduling plays a critical role. In this paper, we investigate and compare different user scheduling policies, which are based on various criteria such as wireless channel conditions and the significance of model updates. Receiver beamforming is applied to minimize the mean-square-error (MSE) of the distortion of function aggregation result via AirComp. Simulation results show that scheduling based on the significance of model updates has smaller fluctuations in the training process while scheduling based on channel condition has the advantage on energy efficiency.

Related