Over-the-Air Federated Learning in User-Centric Cell-Free Networks
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.
Authors
- Weicai Li (ORCID: https://orcid.org/0000-0001-5119-8844)
- Yingping Cui (ORCID: https://orcid.org/0000-0003-4342-6856)
- Tiejun Lv (ORCID: https://orcid.org/0000-0003-4127-2419)
- Ekram Hossain (ORCID: https://orcid.org/0000-0002-5932-6887)
- Wei Ni
Institutions
- University of Technology Sydney (AU)
- Beijing University of Posts and Telecommunications (CN)
- University of Manitoba (CA)
Publication Details
- Journal
- IEEE Wireless Communications Letters
- Published
- 2024-10-24
- DOI
- https://doi.org/10.1109/lwc.2024.3486201
- Citations
- 3
- Primary Topic
- Privacy-Preserving Technologies in Data
- Type
- article
- Field-Weighted Citation Impact
- 1.04