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

Publisher:
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
Publication Type:
Journal Article
Citation:
IEEE Wireless Communications Letters, 2024, 13, (12), pp. 3683-3687
Issue Date:
2024-01-01
Filename Description Size
1771133.pdfPublished version1.1 MB
Adobe PDF
Full metadata record
This letter presents a new over-the-air federated learning (OTA-FL) system supported by a user-centric cell-free (UCCF) network. We propose a two-level hierarchical deep reinforcement learning (HDRL) framework that minimizes mean squared error (MSE) derived from convergence analysis by jointly optimizing AP-device association (ADA) and power control (PC). Specifically, a soft actor-critic (SAC) with customized data pre-processing is designed for addressing ADA, and a multi-actor-attention-critic (MAAC) with tailored preprocessing and policy networks is designed for handling PC with complex state-action space. Simulations show that our method improves MSE by at least 31% and achieves better OTA-FL convergence than its benchmarks.
Please use this identifier to cite or link to this item: