The invention discloses an air
federated learning realization method for collaborative optimization of
client scheduling and
model compression. The invention realizes the air
federated learning realization method for collaborative optimization of
client scheduling and
model compression. Along with rapid development of
federated learning in a
wireless network, air computing based on a multiple-input-multiple-output technology is widely concerned due to high communication efficiency of the air computing. However, in a resource-limited large-scale device scene, channel interference, device heterogeneity and limited spectrum resources significantly
restrict the convergence rate and model performance of federated learning. In order to deal with the challenges, the invention provides an air federated learning framework combining
compressed sensing and
client scheduling, and by
collaborative design of
model parameter compression, multi-antenna beam forming and dynamic equipment selection strategies, the total communication overhead of each round of training is minimized, and meanwhile, the convergence of a
global model is guaranteed. In order to
balance training cost and
model quality, a joint
optimization problem based on calculation-communication cost and model precision loss is provided. In order to solve the non-convex
optimization problem, an original problem is decoupled into two sub-problems. Firstly, an
optimization problem of a pre-coding and post-
processing matrix is designed for a given user scheduling result to minimize a gradient aggregation error. Then, a novel user scheduling
algorithm based on channels and data is provided to obtain an air aggregation result.