Residential Collegefalse
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 NameInternational Symposium on Quality of Service
Source Publication32nd IEEE/ACM International Symposium on Quality of Service, IWQoS 2024
Pages202971
Conference Date2024-06
Conference PlaceGuangdong
PublisherInstitute 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.

KeywordContrastive Learning Data Heterogeneity Ersonalized Federated Learning Representation Learning
DOI10.1109/IWQoS61813.2024.10682940
URLView the original
Language英語English
Scopus ID2-s2.0-85206352513
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionTHE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU)
Corresponding AuthorLi Li
Affiliation1.National University of Defense Technology, Changsha, China
2.University of Macau, Macao
3.Ocean University of China, Qingdao, China
Corresponding Author AffilicationUniversity 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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Meihan Wu]'s Articles
[Li Li]'s Articles
[Tao Chang]'s Articles
Baidu academic
Similar articles in Baidu academic
[Meihan Wu]'s Articles
[Li Li]'s Articles
[Tao Chang]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Meihan Wu]'s Articles
[Li Li]'s Articles
[Tao Chang]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.