Residential College | false |
Status | 已發表Published |
PFed-DBA: Distribution Bias Aware Personalized Federated Learning for Data Heterogeneity | |
Meihan Wu1; Li Li2; Tao Chang1; Jie Zhou1; Cui Miao1; Xiaodong Wang1; ChengZhong Xu2; Rigall, Eric3 | |
2024-05 | |
Conference Name | International Symposium on Quality of Service |
Source Publication | 32nd IEEE/ACM International Symposium on Quality of Service, IWQoS 2024 |
Pages | 202971 |
Conference Date | 2024-06 |
Conference Place | Guangdong |
Publisher | Institute of Electrical and Electronics Engineers Inc. |
Abstract | Personalized Federated Learning (PFL) aims to learn a custom model for each distributed client while benefiting from collaborative training in order to overcome the detrimental impact of data heterogeneity. Despite the promising benefits, the existing approaches often compromise the generalization performance of personalized models, as they solely focus on enhancing the personalization capability of models or merely aim to strike a balance between personalization and generalization. Indeed, increasing the personalization capability while preserving the strong generalization performance enabled by collaborative training remains a challenge for PFL, as the two objectives seem to compete with each other. To tackle this challenge, we investigate the relationship between model generalization and personalization under different degrees of heterogeneity. We find that besides the client-specific data distribution, the distribution bias between the unique data distribution of each client and that of the whole population is another critical factor that prominently impacts these two performances. Motivated by the above finding, we propose PFed-DBA, a novel PFL framework that effectively perceives this distribution bias to guide the training process. Concretely, we design the PFL models as a skip-connection network between a shared module for learning the shared representations delivering the common distribution of data across all clients and a personalized module for learning the personalized representations of the heterogeneous distribution bias. Then, we devise corresponding loss functions, aggregation strategy, and updating strategy in order to make the two modules intelligently complement each other. Moreover, we conduct extensive experiments to evaluate the effectiveness of PFed-DBA. The results show that PFed-DBA improves model accuracy to 12.34% at best compared with the state-of-the-art. |
Keyword | Contrastive Learning Data Heterogeneity Ersonalized Federated Learning Representation Learning |
DOI | 10.1109/IWQoS61813.2024.10682940 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85206352513 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) |
Corresponding Author | Li Li |
Affiliation | 1.National University of Defense Technology, Changsha, China 2.University of Macau, Macao 3.Ocean University of China, Qingdao, China |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Meihan Wu,Li Li,Tao Chang,et al. PFed-DBA: Distribution Bias Aware Personalized Federated Learning for Data Heterogeneity[C]:Institute of Electrical and Electronics Engineers Inc., 2024, 202971. |
APA | Meihan Wu., Li Li., Tao Chang., Jie Zhou., Cui Miao., Xiaodong Wang., ChengZhong Xu., & Rigall, Eric (2024). PFed-DBA: Distribution Bias Aware Personalized Federated Learning for Data Heterogeneity. 32nd IEEE/ACM International Symposium on Quality of Service, IWQoS 2024, 202971. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment