Communication-Learning Co-Design for Pinching-Antenna-Assisted Federated Learning in Heterogeneous Wireless Networks
Shuai Wang, Wenxuan Sun, Mingjie Shao, Wantao Jiang, Yanqing Xu, Lingxiang Li, Zhi Chen
原文摘要
Wireless federated learning (FL) suffers from two coupled bottlenecks: statistical heterogeneity across clients and communication-constrained model synchronization over unreliable wireless links. Instead of focusing on algorithmic design and/or wireless resource optimization, this paper provides a systematic study of pinching antenna (PA)-empowered FL for heterogeneous wireless networks and develops a unified communication-and-learning framework. Specifically, we integrate a segmented waveguide-enabled pinching antenna (SWAN) architecture with a heterogeneity-robust FL algorithm based on client variance reduction and dual quantization, so as to jointly mitigate communication bottlenecks and client drift. On this basis, we further develop a two-stage SWAN-assisted downlink protocol and formulate fairness-oriented uplink optimization problems for PA positioning, bandwidth allocation, and quantization design. Efficient solution frameworks are then derived for the resulting downlink and uplink optimization problems. Experimental results show the superiority of the proposed design in convergence speed and final accuracy under non-independent and identically distributed data and communication bottlenecks.
cs.IT