Using analog beamforming in mmWave frequency bands we can focus the energy towards a receiver to achieve high throughput. However, this requires the network to quickly find the best downlink beam configuration in the face of non-IID data. We propose a personalized Federated Learning (FL) method to address this challenge, where we learn a mapping between uplink Sub-6GHz channel estimates and the best downlink beam in heterogeneous scenarios with non-IID characteristics. We also devise FedLion, a FL implementation of the Lion optimization algorithm. Our approach reduces the signalling overhead and provides superior performance, up to 33.6 % higher accuracy than a single FL model and 6 % higher than a local model.
Suggested Citation
@misc{isaksson2023mmwave,
title={mmWave Beam Selection in Analog Beamforming Using Personalized Federated Learning},
author={Martin Isaksson and Filippo Vannella and David Sandberg and Rickard Cöster},
year={2023},
eprint={2310.00406},
archivePrefix={arXiv},
}