Abstract:Personalized federated learning (PFL) has gained increasing attention due to its success in handling the statistical heterogeneity of participants' local data by building distinct local models for participants. However, existing PFL schemes require the identical architecture and size of participants' models, e.g., the same number of layers in convolutional neural networks (CNN). In addition, the growing privacy issues (e.g., local update leakage to the curious server in model aggregation) have not been resolved in PFL. The utilization of identical model architectures among participants reduces the cost of privacy attacks since only one uniform attack method is required to extract private information, exacerbating the privacy threat. This paper proposes a novel privacy-preserving PFL framework that supports heterogeneous model architectures and sizes in delivering personalized models for different participants. Specifically, we utilize participants' knowledge, i.e., the soft predictions of local models on a public dataset, to effectively identify participants with similar data distributions regardless of the specific model architectures used. Based on the participants' knowledge, and their computing and storage capabilities, we employ the affinity propagation (AP) algorithm to implement a multi-level participant clustering mechanism for enabling heterogeneous PFL. Since knowledge is independent of original data, it is considered privacy-preserving during the clustering process. We also devise the ring aggregation algorithm to guarantee participants' privacy during the federated training process. In this way, each participant benefits from other participants with similar data distributions privately and obtains a satisfying personalized model. Furthermore, the cross-cluster knowledge transfer method boosts the personalization performance of weak participants. Sufficient theoretical analyses prove the effectiveness and privacy-preserving capacity of the proposed scheme. Extensive experiments on three benchmark datasets also demonstrate the superiority of our proposed scheme in various settings while maintaining privacy protection, outperforming other state-of-the-art schemes.

Efficient Hierarchical Personalized Federated Learning on Heterogeneous Data

Hierarchical Federated Learning: the Interplay of User Mobility and Data Heterogeneity

Group Personalized Federated Learning

Personalized Federated Learning on Heterogeneous and Long-Tailed Data via Expert Collaborative Learning

FedAPEN: Personalized Cross-silo Federated Learning with Adaptability to Statistical Heterogeneity

Hierarchical Personalized Federated Learning Over Massive Mobile Edge Computing Networks

Sparse Federated Learning with Hierarchical Personalized Models

Towards Fairer and More Efficient Federated Learning via Multidimensional Personalized Edge Models

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

Multi-level Personalized Federated Learning on Heterogeneous and Long-Tailed Data

Personalized Federated Learning with Adaptive Feature Aggregation and Knowledge Transfer

Personalized Federated Learning for Statistical Heterogeneity

Privacy-Preserving Heterogeneous Personalized Federated Learning with Knowledge

APFed: Attention-based Personalized Federated Learning in Model Heterogeneity Scenario.

Personalized federated learning for heterogeneous data: A distributed edge clustering approach

Personalized Federated Learning on Long-Tailed Data Via Adversarial Feature Augmentation.

Adaptive Personalized Federated Learning for Heterogeneous Data: a Method Based on Parameter Decomposition and Continual Learning

Hierarchical federated learning with global differential privacy

Federated Learning as a Service for Hierarchical Edge Networks with Heterogeneous Models

Optimizing Training Efficiency and Cost of Hierarchical Federated Learning in Heterogeneous Mobile-Edge Cloud Computing

HierarchyFL: Heterogeneous Federated Learning via Hierarchical Self-Distillation