AirFL-Mem: Improving communication-learning trade-off by long-term memory

H Wen, H Xing, O Simeone - 2024 IEEE Wireless …, 2024 - ieeexplore.ieee.org
2024 IEEE Wireless Communications and Networking Conference (WCNC), 2024ieeexplore.ieee.org
Addressing the communication bottleneck inherent in federated learning (FL), over-the-air
FL (AirFL) has emerged as a promising solution, which is, however, hampered by deep
fading conditions. In this paper, we propose AirFL-Mem, a novel scheme designed to
mitigate the impact of deep fading by implementing a long-term memory mechanism.
Convergence bounds are provided that account for long-term memory, as well as for existing
AirFL variants with short-term memory, for general non-convex objectives. The theory …
Addressing the communication bottleneck inherent in federated learning (FL), over-the-air FL (AirFL) has emerged as a promising solution, which is, however, hampered by deep fading conditions. In this paper, we propose AirFL-Mem, a novel scheme designed to mitigate the impact of deep fading by implementing a long-term memory mechanism. Convergence bounds are provided that account for long-term memory, as well as for existing AirFL variants with short-term memory, for general non-convex objectives. The theory demonstrates that AirFL-Mem exhibits the same convergence rate of federated aver-aging (FedAvg) with ideal communication, while the performance of existing schemes is generally limited by error floors. The theoretical results are also leveraged to propose a novel convex optimization strategy for the truncation threshold used for power control in the presence of Rayleigh fading channels. Experimental results validate the analysis, confirming the advantages of a long-term memory mechanism for the mitigation of deep fading.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果