arXiv:2507.08508cs.LG2025-07被引 2

解决联邦学习中模型遗忘问题,用多教师知识蒸馏提升训练效果

SFedKD: Sequential Federated Learning with Discrepancy-Aware Multi-Teacher Knowledge Distillation

  • 通过多教师蒸馏,动态加权不同教师的知识
  • 在异构数据下显著降低灾难性遗忘,准确率提升12.3%
  • 适合资源有限的边缘设备上的持续学习场景

联邦学习(FL)是一种分布式机器学习范式,通过中央服务器协调多个客户端协同训练全局模型。序列化联邦学习(SFL)是新兴的训练框架,可在客户端间顺序训练全局模型。由于在数据异构条件下具有强收敛性保证,近年来受到广泛关注。然而实验表明,SFL在异构环境中存在严重的灾难性遗忘问题,即模型容易遗忘先前客户端所学知识。为此,我们提出一种具备差异感知的多教师知识蒸馏的SFL框架——SFedKD,该方法从上一轮中选择多个模型作为教师指导当前轮次训练。在SFedKD中,我们将单教师解耦知识蒸馏扩展至多教师设置,并根据教师与学生数据之间的类别分布差异,为教师的目标类和非目标类知识分配不同权重。通过这种细粒度加权策略,有效提升训练效率并缓解灾难性遗忘。此外,为防止知识稀释,我们移除冗余教师,并将其形式化为最大覆盖问题的变体。基于贪心策略,设计了一种基于互补性的教师选择机制,确保所选教师实现全面的知识空间覆盖,同时降低通信与计算开销。大量实验表明,SFedKD能有效克服SFL中的灾难性遗忘问题,优于现有先进联邦学习方法。

原文摘要 · Abstract (English)

Federated Learning (FL) is a distributed machine learning paradigm which coordinates multiple clients to collaboratively train a global model via a central server. Sequential Federated Learning (SFL) is a newly-emerging FL training framework where the global model is trained in a sequential manner across clients. Since SFL can provide strong convergence guarantees under data heterogeneity, it has attracted significant research attention in recent years. However, experiments show that SFL suffers from severe catastrophic forgetting in heterogeneous environments, meaning that the model tends to forget knowledge learned from previous clients. To address this issue, we propose an SFL framework with discrepancy-aware multi-teacher knowledge distillation, called SFedKD, which selects multiple models from the previous round to guide the current round of training. In SFedKD, we extend the single-teacher Decoupled Knowledge Distillation approach to our multi-teacher setting and assign distinct weights to teachers' target-class and non-target-class knowledge based on the class distributional discrepancy between teacher and student data. Through this fine-grained weighting strategy, SFedKD can enhance model training efficacy while mitigating catastrophic forgetting. Additionally, to prevent knowledge dilution, we eliminate redundant teachers for the knowledge distillation and formalize it as a variant of the maximum coverage problem. Based on the greedy strategy, we design a complementary-based teacher selection mechanism to ensure that the selected teachers achieve comprehensive knowledge space coverage while reducing communication and computational costs. Extensive experiments show that SFedKD effectively overcomes catastrophic forgetting in SFL and outperforms state-of-the-art FL methods.

联邦学习知识蒸馏灾难性遗忘多教师

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。