Search papers, labs, and topics across Lattice.
The paper introduces FedKT-CSD, a novel framework for one-shot federated learning that effectively balances communication efficiency, robustness to data heterogeneity, and formal privacy guarantees. By utilizing pretrained autoencoders to create a shared latent space, clients can encode their private data and transmit class-conditional statistics to a server, which then constructs a synthetic dataset while ensuring $(\varepsilon,未)$-differential privacy. The results show that FedKT-CSD not only meets privacy requirements but also outperforms non-private baselines across various datasets and client distributions.
Achieving formal privacy in federated learning without sacrificing model performance, FedKT-CSD outperforms traditional methods even under stringent privacy constraints.
One-shot federated learning (OSFL) addresses the communication overhead of federated learning by limiting training to a single round, but doing so without sacrificing model quality is non-trivial, particularly when client data distributions diverge. Recent work has addressed this challenge by aggregating client knowledge on the server through the construction of transferable synthetic datasets or distillates. However, most of these methods lack formal privacy guarantees, leaving a gap in jointly achieving low communication, robustness to heterogeneity, and rigorous privacy. We propose FedKT-CSD (Federated Knowledge Transfer via Collaborative Synthetic Data), a framework inspired by neural image compression that closes this gap by leveraging publicly pretrained autoencoders as a shared latent space. Each client encodes its private data in a single forward pass, computes class-conditional latent statistics, and transmits these to the server. The server aggregates these statistics via secure aggregation, adds calibrated differential privacy noise, and decodes a synthetic dataset for training a global model and further downstream tasks. This design provides formal $(\varepsilon,未)$-differential privacy by construction, while keeping client-side computation and communication lightweight. Despite operating under privacy constraints, FedKT-CSD is competitive with and even outperforms non-private baselines across diverse datasets and heterogeneity settings, and scales to a large number of clients. Our code is available at: https://github.com/an7123/FedKT-CSD