Federated Learning for Terahertz Wireless Communication
O. Tansel Baydas, Ozgur B. Akan
TL;DR
This work develops a multicarrier stochastic framework to analyze federated learning over wideband THz channels, explicitly linking local SGD updates to THz impairments such as beam squint, molecular absorption, jitter, and compression. It uncovers a harmonic-mean SNR bottleneck and a fundamental bandwidth limit that can degrade convergence, and it shows that SNR-weighted aggregation can recover convergence in high-squint regimes at the cost of a controlled bias. Theoretical results include a non-convex convergence bound and design inequalities that map physical-layer parameters to learning accuracy, complemented by experimental validation. The findings offer practical guidance for deploying THz-enabled FL, emphasizing channel estimation, equalization, and adaptive, distortion-aware aggregation to ensure reliable edge learning over challenging wideband links.
Abstract
The convergence of Terahertz (THz) communications and Federated Learning (FL) promises ultra-fast distributed learning, yet the impact of realistic wideband impairments on optimization dynamics remains theoretically uncharacterized. This paper bridges this gap by developing a multicarrier stochastic framework that explicitly couples local gradient updates with frequency-selective THz effects, including beam squint, molecular absorption, and jitter. Our analysis uncovers a critical diversity trap: under standard unbiased aggregation, the convergence error floor is driven by the harmonic mean of subcarrier SNRs. Consequently, a single spectral hole caused by severe beam squint can render the entire bandwidth useless for reliable model updates. We further identify a fundamental bandwidth limit, revealing that expanding the spectrum beyond a critical point degrades convergence due to the integration of thermal noise and gain collapse at band edges. Finally, we demonstrate that an SNR-weighted aggregation strategy is necessary to suppress the variance singularity at these spectral holes, effectively recovering convergence in high-squint regimes where standard averaging fails. Numerical results validate the expected impact of the discussed physical layer parameters' on performance of THz-FL systems.
