English

Over-the-Air Federated Learning in Cell-Free MIMO with Long-term Power Constraint

Machine Learning 2024-10-24 v3 Artificial Intelligence

Abstract

Wireless networks supporting artificial intelligence have gained significant attention, with Over-the-Air Federated Learning emerging as a key application due to its unique transmission and distributed computing characteristics. This paper derives error bounds for Over-the-Air Federated Learning in a Cell-free MIMO system and formulates an optimization problem to minimize optimality gap via joint optimization of power control and beamforming. We introduce the MOP-LOFPC algorithm, which employs Lyapunov optimization to decouple long-term constraints across rounds while requiring only causal channel state information. Experimental results demonstrate that MOP-LOFPC achieves a better and more flexible trade-off between the model's training loss and adherence to long-term power constraints compared to existing baselines.

Keywords

Cite

@article{arxiv.2410.05354,
  title  = {Over-the-Air Federated Learning in Cell-Free MIMO with Long-term Power Constraint},
  author = {Yifan Wang and Cheng Zhang and Yuanndon Zhuang and Mingzeng Dai and Haiming Wang and Yongming Huang},
  journal= {arXiv preprint arXiv:2410.05354},
  year   = {2024}
}