Revisiting Ensembling in One-Shot Federated Learning
Federated learning (FL) is an appealing approach to training machine learning models without sharing raw data. However, standard FL algorithms are iterative and thus induce a significant communication cost. One-shot federated learning (OFL) trades the iterative exchange of models between clients and the server with a single round of communication, thereby saving substantially on communication costs. Not surprisingly, OFL exhibits a performance gap in terms of accuracy with respect to FL, especially under high data heterogeneity. We introduce FENS, a novel federated ensembling scheme that approaches the accuracy of FL with the communication efficiency of OFL. Learning in FENS proceeds in two phases: first, clients train models locally and send them to the server, similar to OFL; second, clients collaboratively train a lightweight prediction aggregator model using FL. We showcase the effectiveness of FENS through exhaustive experiments spanning several datasets and heterogeneity levels. In the particular case of heterogeneously distributed CIFAR-10 dataset, FENS achieves up to a 26.9% higher accuracy over state-of-the-art (SOTA) OFL, being only 3.1% lower than FL. At the same time, FENS incurs at most 4.3× more communication than OFL, whereas FL is at least 10.9× more communication-intensive than FENS.
Cite this item
Allouah, Y., Dhasade, A. B., Guerraoui, R., Gupta, N., Kermarrec, A.-M., Pinot, R., Pereira Pires, R., & Sharma, R. (2024, November 11). Revisiting Ensembling in One-Shot Federated Learning. Advances in Neural Information Processing Systems. 38th Annual Conference on Neural Information Processing Systems, Vancouver Convention Center, December 10, 2024. https://infoscience.epfl.ch/handle/20.500.14299/242008
Click Copy to copy the citation, or select the text manually.
main.pdf
Main Document
Published version
openaccess
CC BY
607.16 KB
Adobe PDF
0aee54df8439a4c5a2c20f12dce1e561