Practical One-Shot Federated Learning for Cross-Silo Setting

Practical One-Shot Federated Learning for Cross-Silo Setting

Qinbin Li, Bingsheng He, Dawn Song

Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence
Main Track. Pages 1484-1490. https://doi.org/10.24963/ijcai.2021/205

Federated learning enables multiple parties to collaboratively learn a model without exchanging their data. While most existing federated learning algorithms need many rounds to converge, one-shot federated learning (i.e., federated learning with a single communication round) is a promising approach to make federated learning applicable in cross-silo setting in practice. However, existing one-shot algorithms only support specific models and do not provide any privacy guarantees, which significantly limit the applications in practice. In this paper, we propose a practical one-shot federated learning algorithm named FedKT. By utilizing the knowledge transfer technique, FedKT can be applied to any classification models and can flexibly achieve differential privacy guarantees. Our experiments on various tasks show that FedKT can significantly outperform the other state-of-the-art federated learning algorithms with a single communication round.
Keywords:
Data Mining: Federated Learning
Machine Learning: Classification