Dirichlet-based Uncertainty Quantification for Personalized Federated Learning with Improved Posterior Networks
In modern federated learning, one of the main challenges is to account for inherent heterogeneity and the diverse nature of data distributions for different clients. This problem is often addressed by introducing personalization of the models towards the data distribution of the particular client. However, a personalized model might be unreliable when applied to the data that is not typical for this client. Eventually, it may perform worse for these data than the non-personalized global model trained in a federated way on the data from all the clients. This paper presents a new approach to federated learning that allows selecting a model from global and personalized ones that would perform better for a particular input point. It is achieved through a careful modeling of predictive uncertainties that helps to detect local and global in- and out-of-distribution data and use this information to select the model that is confident in a prediction. The comprehensive experimental evaluation on the popular real-world image datasets shows the superior performance of the model in the presence of out-of-distribution data while performing on par with state-of-the-art personalized federated learning algorithms in the standard scenarios.
Code (0)
등록된 구현이 없습니다.
Tasks
Federated LearningPersonalized Federated LearningUncertainty QuantificationSimilar Papers 제목 키워드 기반
Self-Aware Personalized Federated Learning
In the context of personalized federated learning (FL), the critical challenge is to balance local model improvement and global model tuning when the personal and global objectives may not be exactly aligned. Inspired by…
Federated LearningPersonalized Federated LearningUncertainty QuantificationCUQ-GNN: Committee-based Graph Uncertainty Quantification using Posterior Networks
In this work, we study the influence of domain-specific characteristics when defining a meaningful notion of predictive uncertainty on graph data. Previously, the so-called Graph Posterior Network (GPN) model has been pr…
Node ClassificationUncertainty QuantificationActPerFL: Active Personalized Federated Learning
In the context of personalized federated learning (FL), the critical challenge is to balance local model improvement and global model tuning when the personal and global objectives may not be exactly aligned. Inspired by…
Federated LearningPersonalized Federated LearningUncertainty QuantificationMulti-Agent Conformal Prediction with Personalized Statistical Validity
Uncertainty quantification is essential in high-stakes machine learning tasks. However, one of the principled solutions, conformal prediction, faces challenges under limited local calibration data, privacy constraints, a…
FedSI: Federated Subnetwork Inference for Efficient Uncertainty Quantification
While deep neural networks (DNNs) based personalized federated learning (PFL) is demanding for addressing data heterogeneity and shows promising performance, existing methods for federated learning (FL) suffer from effic…
Federated LearningPersonalized Federated LearningUncertainty Quantification