提出特征蒸馏方法,解决异构联邦学习中模型差异导致的知识偏差问题。
Feature Distillation is the Better Choice for Model-Heterogeneous Federated Learning
- 通过正交投影对齐异构客户端的特征,实现更精准的知识融合。
- 在多个数据集上超越现有方法,提升全局模型性能。
- 适合异构模型协同训练场景,尤其适用于隐私保护下的分布式学习。
异构联邦学习(Hetero-FL)因其能在保护本地数据隐私的同时聚合异构模型知识而受到关注。为增强全局模型性能,常在全局聚合后采用集成蒸馏技术。然而,仅将Hetero-FL与集成蒸馏结合并不总能带来良好效果,且训练过程易不稳定。原因在于现有方法多依赖对数概率蒸馏,虽具备模型无关性,却无法补偿异构模型带来的知识偏差。为此,本文提出一种稳定高效的特征蒸馏方法FedFD,通过正交投影对齐特征信息,更好地整合异构模型知识。具体而言,提出基于特征的集成联邦知识蒸馏范式:服务器端需为每个客户端模型架构维护一个投影层以独立对齐特征;采用正交技术重参数化投影层,缓解异构模型引起的知识偏差,最大化可蒸馏知识。大量实验表明,FedFD在多个基准数据集上均优于现有最优方法。
原文摘要 · Abstract (English)
Model-Heterogeneous Federated Learning (Hetero-FL) has attracted growing attention for its ability to aggregate knowledge from heterogeneous models while keeping private data locally. To better aggregate knowledge from clients, ensemble distillation, as a widely used and effective technique, is often employed after global aggregation to enhance the performance of the global model. However, simply combining Hetero-FL and ensemble distillation does not always yield promising results and can make the training process unstable. The reason is that existing methods primarily focus on logit distillation, which, while being model-agnostic with softmax predictions, fails to compensate for the knowledge bias arising from heterogeneous models. To tackle this challenge, we propose a stable and efficient Feature Distillation for model-heterogeneous Federated learning, dubbed FedFD, that can incorporate aligned feature information via orthogonal projection to integrate knowledge from heterogeneous models better. Specifically, a new feature-based ensemble federated knowledge distillation paradigm is proposed. The global model on the server needs to maintain a projection layer for each client-side model architecture to align the features separately. Orthogonal techniques are employed to re-parameterize the projection layer to mitigate knowledge bias from heterogeneous models and thus maximize the distilled knowledge. Extensive experiments show that FedFD achieves superior performance compared to state-of-the-art methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。