从多个聚类中提炼通用专家模型,提升联邦学习的共享知识利用效率。
Distilling A Universal Expert from Clustered Federated Learning
- 通过三步迭代:本地训练、聚类聚合、通用专家蒸馏,融合个性化与共享知识。
- 在多个数据非独立同分布场景下,性能优于传统梯度聚合方法。
- 适合需要兼顾个性化和全局共享知识的联邦学习应用。
聚类联邦学习(CFL)通过为不同群组训练特定专家模型来应对非独立同分布(non-IID)数据的挑战。然而,现有方法常忽略各聚类间的共享信息,这些信息代表了对所有参与方都有价值的可泛化知识。为此,本文提出一种新型联邦学习框架,从多个聚类的知识中蒸馏出一个通用专家模型。该通用专家捕捉所有客户端之间的全局共享信息,并作为下一回合模型训练的初始化分配给每个客户端。所提框架包含三个迭代步骤:(1) 客户端本地模型训练,(2) 聚类特定模型聚合,(3) 通用专家蒸馏。这一三步学习范式在保留细粒度非IID特性的同时,有效融入跨聚类共享知识。相比传统的基于梯度的聚合方法,基于蒸馏的模型聚合在处理模型异质性方面更具灵活性,并减少了聚类专家之间的冲突。大量实验结果表明,该方法在多种场景下均表现出优越性能,展现出更有效地平衡个性化与共享知识的潜力。
原文摘要 · Abstract (English)
Clustered Federated Learning (CFL) addresses the challenges posed by non-IID data by training multiple group- or cluster-specific expert models. However, existing methods often overlook the shared information across clusters, which represents the generalizable knowledge valuable to all participants in the Federated Learning (FL) system. To overcome this limitation, this paper introduces a novel FL framework that distills a universal expert model from the knowledge of multiple clusters. This universal expert captures globally shared information across all clients and is subsequently distributed to each client as the initialization for the next round of model training. The proposed FL framework operates in three iterative steps: (1) local model training at each client, (2) cluster-specific model aggregation, and (3) universal expert distillation. This three-step learning paradigm ensures the preservation of fine-grained non-IID characteristics while effectively incorporating shared knowledge across clusters. Compared to traditional gradient-based aggregation methods, the distillation-based model aggregation introduces greater flexibility in handling model heterogeneity and reduces conflicts among cluster-specific experts. Extensive experimental results demonstrate the superior performance of the proposed method across various scenarios, highlighting its potential to advance the state of CFL by balancing personalized and shared knowledge more effectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。