Abstract
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 may be 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. © 2025 Elsevier B.V.
| Original language | English |
|---|---|
| Article number | 113073 |
| Journal | Knowledge-Based Systems |
| Volume | 311 |
| Online published | 30 Jan 2025 |
| DOIs | |
| Publication status | Published - 28 Feb 2025 |
Funding
This work was supported in part by the National Natural Science Foundation of China (NSFC) under Grants 62203308, 62376163 and 62306180, in part by the grant from the Research Grants Council of the Hong Kong Special Administrative Region [CityU 11203723], in part by the Guangdong Regional Joint Foundation Key Project under Grant 2022B1515120076; in part by the Natural Science Foundation of Guangdong Province under Grant 2023A1515011238, and in part by the Shenzhen Science and Technology Program under Grant JCYJ20220531101411027.
Research Keywords
- Data heterogeneity
- Multiple classifier aggregation
- Personalized feature extraction
- Personalized federated learning
RGC Funding Information
- RGC-funded
Fingerprint
Dive into the research topics of 'Personalized federated learning with multiple classifier aggregation'. Together they form a unique fingerprint.Projects
- 1 Active
-
GRF: DNA Motif Knowledge Extraction and Distillation from Big Deep Learning Models in Regulatory Genomics
WONG, K. C. (Principal Investigator / Project Coordinator)
1/01/24 → …
Project: Research
Cite this
- APA
- Author
- BIBTEX
- Harvard
- Standard
- RIS
- Vancouver