Tuned hyper-parameters in the CIFAR-10 experiment for the global cluster models, the local models and the gating model.

Tuned hyper-parameters in the CIFAR-10 experiment for the global cluster models, the local models and the gating model.

Source publication
Preprint
Full-text available
Federated Learning (FL) is a promising framework for distributed learning when data is private and sensitive. However, the state-of-the-art solutions in this framework are not optimal when data is heterogeneous and non-Independent and Identically Distributed (non-IID). We propose a practical and robust approach to personalization in FL that adjusts...

Context in source publication

Context 1
... show that our method is still robust in the fully non-IID case when p = 1. See Table 1 for tuned hyperparameters in the CIFAR-10 experiment. ...

Similar publications

Preprint
Full-text available
Federated learning faces huge challenges from model overfitting due to the lack of data and statistical diversity among clients. To address these challenges, this paper proposes a novel personalized federated learning method via Bayesian variational inference named pFedBayes. To alleviate the overfitting, weight uncertainty is introduced to neural...