PERSOTA FL: A Robust-To-Noise Personalized Over The Air Federated Learning For Human Activity Recognition

Arash Rasti-Meymandi1, Pai Chet Ng1, Huan Liu2, Yuanhao Yu2, Konstantinos N. Plataniotis1,
University of Toronto1, Huawei Noah’s Ark Lab, Canada2
ICASSP 2024

Abstract

This paper presents federated learning with over-the-air (OTA) aggregation over a noisy communication channel for human activity recognition (HAR) problem. OTA aggregation permits simultaneous transmission of model weights through waveform superposition, enabling federated computation over the air. However, noise susceptibility in the channel can degrade weight signal aggregation quality. To this end, we propose a novel PerSonalized OTA FL (PerSOTA FL), which employs QR decomposition to estimate model weight. Our PerSOTA FL leverages multiple antennas for robust OTA-based aggregation, particularly in environments with low signal-to-noise ratio (SNR). Our experiments, with two real-world HAR datasets exhibiting practical data and label heterogeneity, indicate that our PerSOTA FL approach achieves comparable performance to vanilla FL while outperforms the conventional OTA FL. Our experimental results verify the efficiency of our PerSOTA FL, demonstrating its capacity for on-device personalized training while delivering a generalized HAR model through OTA FL.

BibTeX

BibTex Code Here