Chen Ying, Wen Jing, Liang Shaoling, Chen Zhaofa, Huang Baohua
School of Computer and Electronic Information, Guangxi University, Nanning 530004, China.
Guangxi Key Laboratory of Digital Infrastructure, Guangxi Zhuang Autonomous Region Information Center, Nanning 530000, China.
Sensors (Basel). 2025 Aug 15;25(16):5076. doi: 10.3390/s25165076.
To address the limitation of generalization of federated learning under non-independent and identically distributed (Non-IID) data, we propose FedDFPA, a personalized federated learning framework that integrates dynamic parameter fusion and prototype alignment. We design a class-wise dynamic parameter fusion mechanism that adaptively fuses global and local classifier parameters at the class level. It enables each client to preserve its reliable local knowledge while selectively incorporating beneficial global information for personalized classification. We introduce a prototype alignment mechanism based on both global and historical information. By aligning current local features with global prototypes and historical local prototypes, it improves cross-client semantic consistency and enhances the stability of local features. To evaluate the effectiveness of FedDFPA, we conduct extensive experiments on various Non-IID settings and client participation rates. Compared to the average performance of state-of-the-art algorithms, FedDFPA improves the average test accuracy by 3.59% and 4.71% under practical and pathological heterogeneous settings, respectively. These results confirm the effectiveness of our dual-mechanism design in achieving a better balance between personalization and collaboration in federated learning.
为了解决联邦学习在非独立同分布(Non-IID)数据下泛化能力的局限性,我们提出了FedDFPA,这是一个集成了动态参数融合和原型对齐的个性化联邦学习框架。我们设计了一种逐类动态参数融合机制,该机制在类级别自适应地融合全局和局部分类器参数。它使每个客户端能够保留其可靠的局部知识,同时有选择地纳入有益的全局信息以进行个性化分类。我们引入了一种基于全局和历史信息的原型对齐机制。通过将当前局部特征与全局原型和历史局部原型对齐,它提高了跨客户端语义一致性,并增强了局部特征的稳定性。为了评估FedDFPA的有效性,我们在各种非独立同分布设置和客户端参与率下进行了广泛的实验。与现有算法的平均性能相比,FedDFPA在实际和病理异构设置下分别将平均测试准确率提高了3.59%和4.71%。这些结果证实了我们的双机制设计在联邦学习中实现个性化与协作之间更好平衡的有效性。