Federated Learning aims to enable joint training of high-performance deep learning by multiple clients while preserving data privacy by avoiding local data upload. The efficiency of collaborative learning is compromised by the heterogeneity of data distribution on different clients. In order to address this, we propose FedDCP, a generalized framework for personalized federated learning, that can mitigate the negative impact of long-tail distribution of local data. The core idea involves using the prototypes to limits the drift of local’s feature extraction and allowing personalized models to assimilate global insights alongside local data adaptation by dual classifier. Furthermore, a series of comprehensive experiments on three distinct datasets show that FedDCP can significantly enhance personalization capabilities of existing federated learning methods. In comparison to eight other SOTA pFL algorithms, the FedDCP demonstrates notable enhancements in accuracy. The code for FedDCP is publicly available on GitHub at https://github.com/awsl0/FedDCP .

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

FedDCP: Personalized Federated Learning Based on Dual Classifiers and Prototypes

  • Xiangxiang Li,
  • Yang Hua,
  • Xiaoning Song,
  • Wenjie Zhang,
  • Xiao-jun Wu

摘要

Federated Learning aims to enable joint training of high-performance deep learning by multiple clients while preserving data privacy by avoiding local data upload. The efficiency of collaborative learning is compromised by the heterogeneity of data distribution on different clients. In order to address this, we propose FedDCP, a generalized framework for personalized federated learning, that can mitigate the negative impact of long-tail distribution of local data. The core idea involves using the prototypes to limits the drift of local’s feature extraction and allowing personalized models to assimilate global insights alongside local data adaptation by dual classifier. Furthermore, a series of comprehensive experiments on three distinct datasets show that FedDCP can significantly enhance personalization capabilities of existing federated learning methods. In comparison to eight other SOTA pFL algorithms, the FedDCP demonstrates notable enhancements in accuracy. The code for FedDCP is publicly available on GitHub at https://github.com/awsl0/FedDCP .