Synthetic Data Aided Federated Learning Using Foundation Models
摘要
In scenarios where the data distribution amongst Federated Learning (FL) participants is Non-Independent and Identically distributed (Non-IID), FL suffers from the well-known problem of data heterogeneity. This leads to significantly degraded FL performance, as the global model tends to struggle to converge. To solve this problem, we propose Differentially Private Synthetic Data Aided Federated Learning Using Foundation Models (DPSDA-FL) - a novel data augmentation strategy that aids in homogenizing the local data present on the clients’ side. DPSDA-FL improves the training of the local models by leveraging differentially private synthetic data generated from foundation models. We demonstrate the effectiveness of our approach by evaluating it on the benchmark image dataset: CIFAR-10. Our experimental results show that DPSDA-FL can improve the global model’s class recall and classification accuracy by up to 26% and 9%, respectively, in FL with Non-IID issues.