A Random Access Scheme for Federated Learning over Massive MIMO Systems

Huimei Han,Jun Zhao,Xinyu Zhou
DOI: https://doi.org/10.1109/jiot.2023.3278256
IF: 10.6
2023-01-01
IEEE Internet of Things Journal
Abstract:In this paper, 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 multi-channel 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 RAFLMIMO scheme can improve the performance of FL. Simulation results based on L2-norm linear regression, and MNIST handwritten digits identification, Cifar-10 photo classification show that the proposed RAFL-MIMO scheme significantly outperforms the case of the RAFL-MIMO without ACB factor.
computer science, information systems,telecommunications,engineering, electrical & electronic
What problem does this paper attempt to address?