Wireless Federated Learning over MIMO Networks: Joint Device Scheduling and Beamforming Design
Federated learning (FL) is recognized as a key enabling technology to support distributed artificial intelligence (AI) services in future 6G. By supporting decentralized data training and collaborative model training among devices, FL inherently tames privacy leakage and reduces transmission costs. Whereas, the performance of the wireless FL is typically restricted by the communication latency. Multiple-input multiple-output (MIMO) technique is one promising solution to build up a communication-efficient edge FL system with limited radio resources. In this paper, we propose a novel joint device scheduling and receive beamforming design approach to reduce the FL convergence gap over shared wireless MIMO networks. Specifically, we theoretically establish the convergence analysis of the FL process, and then apply the proposed device scheduling policy to maximize the number of weighted devices under the FL system latency and sum power constraints. Numerical results verify the theoretical analysis of the FL convergence and exhibit the appealing learning performance of the proposed approach.
Code (0)
등록된 구현이 없습니다.
Tasks
Federated LearningSchedulingSimilar Papers 제목 키워드 기반
Multi-Task Over-the-Air Federated Learning in Cell-Free Massive MIMO Systems
Wireless devices are expected to provide a wide range of AI services in 6G networks. The increasing computing capabilities of wireless devices and the surge of wireless data motivate the use of privacy-preserving federat…
Federated LearningPrivacy PreservingDigital Over-the-Air Federated Learning in Multi-Antenna Systems
In this paper, the performance optimization of federated learning (FL), when deployed over a realistic wireless multiple-input multiple-output (MIMO) communication system with digital modulation and over-the-air computat…
Federated LearningDecentralized Federated Learning via MIMO Over-the-Air Computation: Consensus Analysis and Performance Optimization
Decentralized federated learning (DFL), inherited from distributed optimization, is an emerging paradigm to leverage the explosively growing data from wireless devices in a fully distributed manner.DFL enables joint trai…
Distributed OptimizationFederated LearningEnergy-Efficient Over-the-Air Federated Learning via Pinching Antenna Systems
Pinching antennas systems (PASSs) have recently been proposed as a novel flexible-antenna technology. These systems are implemented by attaching low-cost pinching elements to dielectric waveguides. As the direct link is …
Federated LearningOver-the-Air Computation in Cell-Free Massive MIMO Systems
Over-the-air computation (AirComp) is considered as a communication-efficient solution for data aggregation and distributed learning by exploiting the superposition properties of wireless multi-access channels. However, …