English
Related papers

Related papers: Personalized Bayesian Federated Learning with Wass…

200 papers

Personalized Federated Learning (PFL) enables a collection of agents to collaboratively learn individual models without sharing raw data. We propose a new PFL approach in which each agent optimizes a weighted combination of all agents'…

Machine Learning · Computer Science 2026-03-04 Jean-Baptiste Fermanian , Batiste Le Bars , Aurélien Bellet

Federated Learning (FL) offers a decentralized approach to model training, where data remains local and only model parameters are shared between the clients and the central server. Traditional methods, such as Federated Averaging (FedAvg),…

Machine Learning · Computer Science 2025-02-12 Jiahao Lai , Jiaqi Li , Jian Xu , Yanru Wu , Boshi Tang , Siqi Chen , Yongfeng Huang , Wenbo Ding , Yang Li

Federated learning aims to learn a global model that performs well on client devices with limited cross-client communication. Personalized federated learning (PFL) further extends this setup to handle data heterogeneity between clients by…

Machine Learning · Computer Science 2021-10-27 Idan Achituve , Aviv Shamsian , Aviv Navon , Gal Chechik , Ethan Fetaya

Traditional Federated Learning (FL) methods encounter significant challenges when dealing with heterogeneous data and providing personalized solutions for non-IID scenarios. Personalized Federated Learning (PFL) approaches aim to address…

Machine Learning · Computer Science 2025-11-11 Yasaman Saadati , Mohammad Rostami , M. Hadi Amini

Federated Learning (FL) has emerged as a promising method to collaboratively learn from decentralized and heterogeneous data available at different clients without the requirement of data ever leaving the clients. Recent works on FL have…

Machine Learning · Computer Science 2024-11-28 Shivam Pal , Aishwarya Gupta , Saqib Sarwar , Piyush Rai

In federated learning, participating clients typically possess non-i.i.d. data, posing a significant challenge to generalization to unseen distributions. To address this, we propose a Wasserstein distributionally robust optimization scheme…

Machine Learning · Computer Science 2022-06-06 Tung-Anh Nguyen , Tuan Dung Nguyen , Long Tan Le , Canh T. Dinh , Nguyen H. Tran

Federated learning's poor performance in the presence of heterogeneous data remains one of the most pressing issues in the field. Personalized federated learning departs from the conventional paradigm in which all clients employ the same…

Machine Learning · Computer Science 2024-02-27 Mengen Luo , Ercan Engin Kuruoglu

Bayesian Federated Learning (BFL) combines uncertainty modeling with decentralized training, enabling the development of personalized and reliable models under data heterogeneity and privacy constraints. Existing approaches typically rely…

Machine Learning · Computer Science 2026-01-22 Nour Jamoussi , Giuseppe Serra , Photios A. Stavrou , Marios Kountouris

Personalized federated learning (PFL) addresses a critical challenge of collaboratively training customized models for clients with heterogeneous and scarce local data. Conventional federated learning, which relies on a single consensus…

Machine Learning · Computer Science 2025-10-24 Ke Xing , Yanjie Dong , Xiaoyi Fan , Runhao Zeng , Victor C. M. Leung , M. Jamal Deen , Xiping Hu

Personalized federated learning (PFL) tailors models to clients' unique data distributions while preserving privacy. However, existing aggregation-weight-based PFL methods often struggle with heterogeneous data, facing challenges in…

Machine Learning · Computer Science 2025-02-18 Yuxia Sun , Aoxiang Sun , Siyi Pan , Zhixiao Fu , Jingcai Guo

In this paper, we first propose a novel algorithm for model fusion that leverages Wasserstein barycenters in training a global Deep Neural Network (DNN) in a distributed architecture. To this end, we divide the dataset into equal parts that…

Machine Learning · Computer Science 2025-06-23 Luiz Pereira , M. Hadi Amini

Personalized federated learning (PFL) aims to harness the collective wisdom of clients' data while building personalized models tailored to individual clients' data distributions. Existing works offer personalization primarily to clients…

Computer Vision and Pattern Recognition · Computer Science 2023-11-01 Hong-You Chen , Jike Zhong , Mingda Zhang , Xuhui Jia , Hang Qi , Boqing Gong , Wei-Lun Chao , Li Zhang

In federated learning, model personalization can be a very effective strategy to deal with heterogeneous training data across clients. We introduce WAFFLE (Weighted Averaging For Federated LEarning), a personalized collaborative machine…

Machine Learning · Computer Science 2021-12-14 Martin Beaussart , Felix Grimberg , Mary-Anne Hartley , Martin Jaggi

Federated Learning (FL) enables collaborative model training while preserving the privacy of raw data. A challenge in this framework is the fair and efficient valuation of data, which is crucial for incentivizing clients to contribute…

Machine Learning · Computer Science 2024-05-10 Wenqian Li , Shuran Fu , Fengrui Zhang , Yan Pang

Different from conventional federated learning, personalized federated learning (PFL) is able to train a customized model for each individual client according to its unique requirement. The mainstream approach is to adopt a kind of weighted…

Machine Learning · Computer Science 2023-07-18 Jiahao Liu , Jiang Wu , Jinyu Chen , Miao Hu , Yipeng Zhou , Di Wu

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 Bayesian…

Machine Learning · Computer Science 2022-04-19 Huili Chen , Jie Ding , Eric Tramel , Shuang Wu , Anit Kumar Sahu , Salman Avestimehr , Tao Zhang

Model-heterogeneous personalized federated learning (MHPFL) enables FL clients to train structurally different personalized models on non-independent and identically distributed (non-IID) local data. Existing MHPFL methods focus on…

Machine Learning · Computer Science 2024-04-30 Liping Yi , Han Yu , Chao Ren , Heng Zhang , Gang Wang , Xiaoguang Liu , Xiaoxiao Li

Federated Learning(FL) is popular as a privacy-preserving machine learning paradigm for generating a single model on decentralized data. However, statistical heterogeneity poses a significant challenge for FL. As a subfield of FL,…

Machine Learning · Computer Science 2024-10-22 Keting Yin , Jiayi Mao

Federated learning (FL) allows multiple clients to collaboratively learn a globally shared model through cycles of model aggregation and local model training, without the need to share data. Most existing FL methods train local models…

Machine Learning · Computer Science 2023-12-05 Liangxi Liu , Xi Jiang , Feng Zheng , Hong Chen , Guo-Jun Qi , Heng Huang , Ling Shao

Performance degradation owing to data heterogeneity and low output interpretability are the most significant challenges faced by federated learning in practical applications. Personalized federated learning diverges from traditional…

Machine Learning · Computer Science 2024-02-28 Mengen Luo , Chi Xu , Ercan Engin Kuruoglu