2023/12/18 by Nikita Kotelevskii, Samuel Horváth, Kotelevskii, Nikita +7 · 2 citations
Computer Science · Mathematics · Medicine · #Artificial intelligence #Computer science #Data mining #Data science #FOS: Computer and information sciences #Federated learning #MRI in cancer diagnosis #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Machine learning #Personalization #Point (geometry) #Privacy-Preserving Technologies in Data #Statistical Methods and Inference #World Wide Web
paper · pdf · doi:10.48550/arxiv.2312.11230
published in arXiv (Cornell University) (Cornell University)
openalex publication_date 2023/12/18 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28
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.