Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Revisiting Ensembling in One-Shot Federated Learning
 
conference paper

Revisiting Ensembling in One-Shot Federated Learning

Allouah, Youssef  
•
Dhasade, Akash Balasaheb  
•
Guerraoui, Rachid  
Show more
November 11, 2024
Advances in Neural Information Processing Systems
38th Annual Conference on Neural Information Processing Systems

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.

  • Files
  • Details
  • Metrics
Type
conference paper
ArXiv ID

2411.07182

Author(s)
Allouah, Youssef  

EPFL

Dhasade, Akash Balasaheb  

EPFL

Guerraoui, Rachid  

EPFL

Gupta, Nirupam  
Kermarrec, Anne-Marie  

EPFL

Pinot, Rafael  
Pereira Pires, Rafael  

EPFL

Sharma, Rishi  
Date Issued

2024-11-11

Publisher

Curran Associates, Inc.

Published in
Advances in Neural Information Processing Systems
ISSN (of the series)

1049-5258

Subjects

Computer Science - Learning

•

Computer Science - Distributed; Parallel; and Cluster Computing

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
SACS  
DCL  
Event nameEvent acronymEvent placeEvent date
38th Annual Conference on Neural Information Processing Systems

NeurIPS 2024

Vancouver Convention Center

2024-12-10 - 2024-12-15

Available on Infoscience
November 12, 2024
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/242008
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés