A Random Access Scheme for Federated Learning Over Massive MIMO Systems
Huimei Han, Jun Zhao, Xinyu Zhou · IEEE Internet of Things Journal · 2023
In this article, we present a random access (RA) scheme for federated learning (FL) over massive multiple-input–multiple-output (MIMO) systems to tackle the issue of some local devices not being able to compute their local models. This scheme adopts a multichannel model and allows devices to randomly select their uploading channels, and then the base station (BS) aggregates the local models received from channels directly based on the over-the-air computation. We call this scheme as RA-based FL over massive MIMO (RAFL-MIMO). Furthermore, to enable more devices to be involved in the FL process, we propose to utilize an access class barring (ACB) method to select the uploading devices and formulate an optimization problem of the ACB factor. We also derive the expected asymptotic convergence rate of the proposed RAFL-MIMO scheme to analytically show that the proposed RAFL-MIMO scheme can improve the performance of FL. Simulation results based on${L2}$-norm linear regression, and MNIST handwritten digits identification, Cifar-10 photograph classification show that the proposed RAFL-MIMO scheme significantly outperforms the case of the RAFL-MIMO without the ACB factor.