English

One-Shot Federated Learning with Classifier-Free Diffusion Models

Machine Learning 2026-01-30 v2

Abstract

Federated learning (FL) enables collaborative learning without data centralization but introduces significant communication costs due to multiple communication rounds between clients and the server. One-shot federated learning (OSFL) addresses this by forming a global model with a single communication round, often relying on the server's model distillation or auxiliary dataset generation - mostly through pre-trained diffusion models (DMs). Existing DM-assisted OSFL methods, however, typically employ classifier-guided DMs, which require training auxiliary classifier models at each client, introducing additional computation overhead. This work introduces OSCAR (One-Shot Federated Learning with Classifier-Free Diffusion Models), a novel OSFL approach that eliminates the need for auxiliary models. OSCAR uses foundation models to devise category-specific data representations at each client which are integrated into a classifier-free diffusion model pipeline for server-side data generation. In our experiments, OSCAR outperforms the state-of-the-art on four benchmark datasets while reducing the communication load by at least 99%.

Keywords

Cite

@article{arxiv.2502.08488,
  title  = {One-Shot Federated Learning with Classifier-Free Diffusion Models},
  author = {Obaidullah Zaland and Shutong Jin and Florian T. Pokorny and Monowar Bhuyan},
  journal= {arXiv preprint arXiv:2502.08488},
  year   = {2026}
}

Comments

Published in IEEE ICME 2025

R2 v1 2026-06-28T21:41:49.607Z