arrow
Return

Personalized federated learning with multiple classifier aggregation

delete2025-02-01
delete0
PRE
AI
林秋镇 (Qiuzhen Lin)
S
Songbai Liu
L
Li, Jianqiang
DOI:10.1016/j.knosys.2025.113073delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Personalized federated learning (PFL) has garnered attention due to its capability to address statistical heterogeneity among clients. Typically, prevailing PFL methods aggregate a single global model for personalization, which maybe inadequate for clients with diverse data distributions. Furthermore, in the local update, each private dataset is used to optimize the model independently, which increases the risk of overfitting the current data distribution and losing previously acquired knowledge, resulting in knowledge forgetting. In this study, a personalized federated learning with multiple classifier aggregation (FedMCA) method is proposed. FedMCA splits the client model into its head and base, optimizing them respectively using an alternating strategy that sequentially targets the head and base. Initially, to address the suboptimal model problem, the proposed method aggregates multiple classifiers using data distribution and employs knowledge distillation to impart positive and negative classifier knowledge for learning the most suitable personalized model head. Additionally, to mitigate knowledge forgetting, a learnable personalization layer is introduced, and hidden loss is utilized to learn the knowledge of the global base and prevent overfitting of the model base. The experimental results demonstrate that the proposed method achieves competitive performance across various benchmarks, outperforming most state-of-the-art PFL algorithms. The source code is publicly available at https://github.com/xiaye-maker/FedMCA.
Keywords:
Personalized federated learning
Data heterogeneity
Multiple classifier aggregation
Personalized feature extraction

Journal

K
Knowledge-Based Systems
IF:
7.6
Papers:
1.2W
Citations:
4.5W

Organization

No organization information available
Cited Papers

Cited Papers

errShare
errSave
Federated Machine Learning: Concept and Applications
err2019-01-28
err5.5K
PREAI
errYang, Qiang; Liu, Yang; Chen, Tianjian; Tong, Yongxin
errShare
errSave
Scalable and accurate deep learning with electronic health records
err2018-05-08
err1.4K
errOAAI
errRajkomar, Alvin; Oren, Eyal; Chen, Kai; Dai, Andrew M.; Hajaj, Nissan; Hardt, Michaela; Liu, Peter J.; Liu, Xiaobing; Marcus, Jake; Sun, Mimi; Sundberg, Patrik; Yee, Hector; Zhang, Kun; Zhang, Yi; Flores, Gerardo; Duggan, Gavin E.; Irvine, Jamie; Quoc Le; Litsch, Kurt; Mossin, Alexander; Tansuwan, Justin; Wang, De; Wexler, James; Wilson, Jimbo; Ludwig, Dana; Volchenboum, Samuel L.; Chou, Katherine; Pearson, Michael; Madabushi, Srinivasan; Shah, Nigam H.; Butte, Atul J.; Howell, Michael D.; Cui, Claire; Corrado, Greg S.; Dean, Jeffrey
errShare
errSave
FedALA: Adaptive Local Aggregation for Personalized Federated Learning
err2023-06-26
err0
errOAAI
errJianqing Zhang; Yang Hua; Hao Wang; Tao Song; Zhengui Xue; Ruhui Ma; Haibing Guan
errShare
errSave
MedMNIST v2 - A large-scale lightweight benchmark for 2D and 3D biomedical image classification
err2023-01-19
err0
errOAAI
errJiancheng Yang; Rui Shi; Donglai Wei; Zequan Liu; Lin Zhao; Bilian Ke; Hanspeter Pfister; Bingbing Ni
errShare
errSave
Knowledge-based Clustering Federated Learning for fault diagnosis in robotic assembly
err2024-06-01
err2
PREAI
errXiao, Peng; Wang, Chuang; Lin, Ze; Hao, Ying; Chen, Gang; Xie, Longhan
errShare
errSave
Efficient one-off clustering for personalized federated learning
err2023-10-01
err3
PREAI
errLiang, Tingting; Yuan, Cheng; Lu, Cheng; Li, Youhuizi; Yuan, Junfeng; Yin, Yuyu
errShare
errSave
researcher View more