Abstract:Federated Learning (FL) is a distributed machine learning paradigm that achieves a globally robust model through decentralized computation and periodic model synthesis, primarily focusing on the global model's accuracy over aggregated datasets of all participating clients. Personalized Federated Learning (PFL) instead tailors exclusive models for each client, aiming to enhance the accuracy of clients' individual models on specific local data distributions. Despite of their wide adoption, existing FL and PFL works have yet to comprehensively address the class-imbalance issue, one of the most critical challenges within the realm of data heterogeneity in PFL and FL research. In this paper, we propose FedReMa, an efficient PFL algorithm that can tackle class-imbalance by 1) utilizing an adaptive inter-client co-learning approach to identify and harness different clients' expertise on different data classes throughout various phases of the training process, and 2) employing distinct aggregation methods for clients' feature extractors and classifiers, with the choices informed by the different roles and implications of these model components. Specifically, driven by our experimental findings on inter-client similarity dynamics, we develop critical co-learning period (CCP), wherein we introduce a module named maximum difference segmentation (MDS) to assess and manage task relevance by analyzing the similarities between clients' logits of their classifiers. Outside the CCP, we employ an additional scheme for model aggregation that utilizes historical records of each client's most relevant peers to further enhance the personalization stability. We demonstrate the superiority of our FedReMa in extensive experiments.

Exploring Personalization Via Federated Representation Learning on Non-Iid Data.

FedDGP: Disentangling Global and Personal Models for Federated Learning

Group Personalized Federated Learning

FedPer++: Toward Improved Personalized Federated Learning on Heterogeneous and Imbalanced Data

Personalized Federated Learning via Feature Distribution Adaptation

Client-supervised Federated Learning: Towards One-model-for-all Personalization

Personalized federated learning based on feature fusion

Exploiting Shared Representations for Personalized Federated Learning

FedReMa: Improving Personalized Federated Learning via Leveraging the Most Relevant Clients

FedCRL: Personalized Federated Learning with Contrastive Shared Representations for Label Heterogeneity in Non-IID Data

FedL2P: Federated Learning to Personalize

FedCR: Personalized Federated Learning Based on Across-Client Common Representation with Conditional Mutual Information Regularization.

Personalizing or Not: Dynamically Personalized Federated Learning with Incentives

Learn What You Need in Personalized Federated Learning

Sparse Personalized Federated Learning

Personalized Federated Learning via Sequential Layer Expansion in Representation Learning

Federated Learning with Personalization Layers

Personalized Federated Learning with Adaptive Feature Aggregation and Knowledge Transfer

Personalized Federated Learning: A Clustered Distributed Co-Meta-Learning Approach

Sparse Federated Learning with Hierarchical Personalized Models

FedPHP: Federated Personalization with Inherited Private Models