Multiple Adaptive Prototypes Learning in Personalized Federated Learning

Xiaozhen Xu,Chunhui Zhao,Liming Zhu,Jun Zhang
DOI: https://doi.org/10.1109/cac59555.2023.10451033
2023-01-01
Abstract:Personalized Federated Learning (pFL) is a method that allows local models to leverage common prior knowledge obtained from a broad range of distributed data, in order to enhance their ability to fit local data. However, most research over the past few years has mainly applied weighted aggregation of entire model parameters, which limits the heterogeneity of models across clients and reduces communication efficiency. In this paper, we propose a noval pFL training framework named Multiple Adaptive Prototypes Personalized Federated Learning (mapFL), which can simultaneously solve statistical heterogeneity and model heterogeneity problems. In this framework, Clients share common knowledge using a memory network consisting of multiple prototypes that are trainable and contain high-level semantic information about categories in different scenarios. Fur-thermore, we introduce the infoNCeloss in contrastive learning to separate the distance between prototypes of different categories in feature space. The Hungarian algorithm is employed to select prototypes of the same category for knowledge fusion. We conducte extensive experiments on different models and learning tasks, which demonstrate that mapFL shows great performance in both model accuracy and communication efficiency.
What problem does this paper attempt to address?