English
Related papers

Related papers: Personalized Bayesian Federated Learning with Wass…

200 papers

Personalized Federated Learning (PFL) aims to train a personalized model for each client that is tailored to its local data distribution, learning fails to perform well on individual clients due to variations in their local data…

Machine Learning · Computer Science 2025-03-10 Ziran Zhou , Guanyu Gao , Xiaohu Wu , Yan Lyu

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…

Machine Learning · Statistics 2023-12-19 Nikita Kotelevskii , Samuel Horváth , Karthik Nandakumar , Martin Takáč , Maxim Panov

Facing the challenge of statistical diversity in client local data distribution, personalized federated learning (PFL) has become a growing research hotspot. Although the state-of-the-art methods with model similarity-based pairwise…

Artificial Intelligence · Computer Science 2022-02-09 Leijie Wu , Song Guo , Yaohong Ding , Yufeng Zhan , Jie Zhang

Personalized Federated Learning (FL) is an emerging research field in FL that learns an easily adaptable global model in the presence of data heterogeneity among clients. However, one of the main challenges for personalized FL is the heavy…

Machine Learning · Computer Science 2022-06-14 Jaehun Song , Min-hwan Oh , Hyung-Sin Kim

Federated learning allows clients to collaboratively learn statistical models while keeping their data local. Federated learning was originally used to train a unique global model to be served to all clients, but this approach might be…

Machine Learning · Computer Science 2022-06-20 Othmane Marfoq , Giovanni Neglia , Laetitia Kameni , Richard Vidal

We introduce FedGVI, a probabilistic Federated Learning (FL) framework that is robust to both prior and likelihood misspecification. FedGVI addresses limitations in both frequentist and Bayesian FL by providing unbiased predictions under…

Machine Learning · Computer Science 2025-06-11 Terje Mildner , Oliver Hamelijnck , Paris Giampouras , Theodoros Damoulas

Conventional frequentist FL schemes are known to yield overconfident decisions. Bayesian FL addresses this issue by allowing agents to process and exchange uncertainty information encoded in distributions over the model parameters. However,…

Machine Learning · Computer Science 2022-09-20 Jinu Gong , Osvaldo Simeone , Joonhyuk Kang

In Federated Learning, heterogeneity in client data distributions often means that a single global model does not have the best performance for individual clients. Consider for example training a next-word prediction model for keyboards:…

Machine Learning · Computer Science 2025-05-06 Ljubomir Rokvic , Panayiotis Danassis , Boi Faltings

Statistical heterogeneity across clients in a Federated Learning (FL) system increases the algorithm convergence time and reduces the generalization performance, resulting in a large communication overhead in return for a poor model. To…

Machine Learning · Computer Science 2023-04-26 Mohamad Mestoukirdi , Matteo Zecchin , David Gesbert , Qianrui Li

A key challenge in federated learning (FL) is the statistical heterogeneity that impairs the generalization of the global model on each client. To address this, we propose a method Federated learning with Adaptive Local Aggregation (FedALA)…

Machine Learning · Computer Science 2023-09-19 Jianqing Zhang , Yang Hua , Hao Wang , Tao Song , Zhengui Xue , Ruhui Ma , Haibing Guan

Motivated by high resource costs of centralized machine learning schemes as well as data privacy concerns, federated learning (FL) emerged as an efficient alternative that relies on aggregating locally trained models rather than collecting…

Machine Learning · Computer Science 2023-12-22 Yiyue Chen , Haris Vikalo , Chianing Wang

Personalized Federated Learning (PFL) faces persistent challenges, including domain heterogeneity from diverse client data, data imbalance due to skewed participation, and strict communication constraints. Traditional federated learning…

Machine Learning · Computer Science 2025-11-25 Mincheol Jeon , Euinam Huh

Federated recommendation (FR) facilitates collaborative training by aggregating local models from massive devices, enabling client-specific personalization while ensuring privacy. However, we empirically and theoretically demonstrate that…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-19 Jundong Chen , Honglei Zhang , Chunxu Zhang , Fangyuan Luo , Yidong Li

In federated learning problems, data is scattered across different servers and exchanging or pooling it is often impractical or prohibited. We develop a Bayesian nonparametric framework for federated learning with neural networks. Each data…

Federated Learning (FL) faces challenges from client data heterogeneity and resource-constrained mobile devices, which can degrade model accuracy. Personalized Federated Learning (PFL) addresses this issue by adapting shared global…

Machine Learning · Computer Science 2026-04-21 Xingyan Chen , Tian Du , Changqiao Xu , Fuzhen Zhuang , Lujie Zhong , Gabriel-Miro Muntean , Enmao Diao

We consider the problem of computing a Wasserstein barycenter for a set of discrete probability distributions with finite supports, which finds many applications in areas such as statistics, machine learning and image processing. When the…

Optimization and Control · Mathematics 2020-12-29 Lei Yang , Jia Li , Defeng Sun , Kim-Chuan Toh

Personalized Federated Learning (pFL), which utilizes and deploys distinct local models, has gained increasing attention in recent years due to its success in handling the statistical heterogeneity of FL clients. However, standardized…

Machine Learning · Computer Science 2022-10-14 Daoyuan Chen , Dawei Gao , Weirui Kuang , Yaliang Li , Bolin Ding

Personalized federated learning (PFL) jointly trains a variety of local models through balancing between knowledge sharing across clients and model personalization per client. This paper addresses PFL via explicit disentangling latent…

Machine Learning · Computer Science 2023-07-14 Peng Yan , Guodong Long

Federated learning (FL) faces two structural tensions: gradient sharing enables data-reconstruction attacks, while non-IID client distributions degrade aggregation quality. We introduce PTOPOFL, a framework that addresses both challenges…

Machine Learning · Computer Science 2026-03-05 Kelly L Vomo-Donfack , Adryel Hoszu , Grégory Ginot , Ian Morilla

Federated foundation models represent a new paradigm to jointly fine-tune pre-trained foundation models across clients. It is still a challenge to fine-tune foundation models for a small group of new users or specialized scenarios, which…

Machine Learning · Computer Science 2025-09-17 Yiyuan Yang , Guodong Long , Qinghua Lu , Liming Zhu , Jing Jiang