Base Station Dataset-Assisted Broadband Over-the-Air Aggregation for Communication-Efficient Federated Learning | IEEE Journals & Magazine | IEEE Xplore

Base Station Dataset-Assisted Broadband Over-the-Air Aggregation for Communication-Efficient Federated Learning


Abstract:

This paper proposes an over-the-air aggregation framework for federated learning (FL) in broadband wireless networks where not only edge devices but also a base station (...Show More

Abstract:

This paper proposes an over-the-air aggregation framework for federated learning (FL) in broadband wireless networks where not only edge devices but also a base station (BS) has its own local dataset. The proposed framework leverages the BS dataset to improve communication efficiency of FL by reducing the number of channel uses required for the model convergence as well as avoiding the signaling overhead incurred by power scale coordination among edge devices. We analyze the convergence to a stationary point without convexity assumption on the objective function. The analysis result reveals that the utilization of BS dataset improves the convergence rate and the update distortion caused by the limited power budget is a crucial factor hindering the model convergence. To facilitate the convergence, we develop an optimized power control method by solving the distortion minimization problem without assumptions on power scale coordination and global CSI at BS. Our simulation results validate that BS dataset is beneficial to reducing the number of channel uses for the model convergence and the developed power control method outperforms the conventional method in terms of both convergence rate and converged test accuracy. Furthermore, we identify some scenarios where the compression of local update can be helpful to reduce communication resources for model training.
Published in: IEEE Transactions on Wireless Communications ( Volume: 22, Issue: 11, November 2023)
Page(s): 7259 - 7272
Date of Publication: 03 March 2023

ISSN Information:

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.