Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Federated Learning in Non-IID Settings Aided by Differentially Private Synthetic Data
(Submitted on 1 Jun 2022 (v1), last revised 20 Apr 2023 (this version, v2))
Abstract: Federated learning (FL) is a privacy-promoting framework that enables potentially large number of clients to collaboratively train machine learning models. In a FL system, a server coordinates the collaboration by collecting and aggregating clients' model updates while the clients' data remains local and private. A major challenge in federated learning arises when the local data is heterogeneous -- the setting in which performance of the learned global model may deteriorate significantly compared to the scenario where the data is identically distributed across the clients. In this paper we propose FedDPMS (Federated Differentially Private Means Sharing), an FL algorithm in which clients deploy variational auto-encoders to augment local datasets with data synthesized using differentially private means of latent data representations communicated by a trusted server. Such augmentation ameliorates effects of data heterogeneity across the clients without compromising privacy. Our experiments on deep image classification tasks demonstrate that FedDPMS outperforms competing state-of-the-art FL methods specifically designed for heterogeneous data settings.
Submission history
From: Huancheng Chen [view email][v1] Wed, 1 Jun 2022 18:00:48 GMT (1771kb,D)
[v2] Thu, 20 Apr 2023 01:50:34 GMT (2670kb,D)
Link back to: arXiv, form interface, contact.