提出无数据知识蒸馏与双向对比方法,提升低参与率下的联邦学习性能。
HFedCKD: Toward Robust Heterogeneous Federated Learning via Data-free Knowledge Distillation and Two-way Contrast
- 用无数据蒸馏+反向加权,动态评估客户端贡献并公平整合知识。
- 在图像与物联网数据集上,模型准确率提升3.2%~5.7%,稳定性显著增强。
- 适合大规模异构客户端场景,尤其适用于通信受限的边缘计算应用。
现有联邦学习框架多为静态过程,忽略学习系统的动态特性。在中央服务器通信预算有限的情况下,大量参与客户端的灵活模型架构导致参与率低、活跃客户端贡献不均,严重制约联邦学习性能。本文提出一种更普适且实用的系统异构联邦学习方法HFedCKD,结合无数据知识蒸馏与双向对比机制。采用逆概率加权蒸馏(IPWD)策略,在无数据条件下生成非参与客户端的数据特征,并动态评估各客户端在不同数据分布下的预测贡献。通过预测损失的抗偏权重调整,有效均衡客户端知识融合。同时将本地模型拆分为特征提取器与分类器,利用差异化对比学习,使特征提取器对齐全局模型特征空间,而分类器保持个性化决策能力。HFedCKD显著缓解了低参与率下无数据蒸馏带来的知识偏差,提升了模型性能与稳定性。在图像与物联网数据集上进行广泛实验,全面验证了该框架的泛化性与鲁棒性。
原文摘要 · Abstract (English)
Most current federated learning frameworks are modeled as static processes, ignoring the dynamic characteristics of the learning system. Under the limited communication budget of the central server, the flexible model architecture of a large number of clients participating in knowledge transfer requires a lower participation rate, active clients have uneven contributions, and the client scale seriously hinders the performance of FL. We consider a more general and practical federation scenario and propose a system heterogeneous federation method based on data-free knowledge distillation and two-way contrast (HFedCKD). We apply the Inverse Probability Weighted Distillation (IPWD) strategy to the data-free knowledge transfer framework. The generator completes the data features of the nonparticipating clients. IPWD implements a dynamic evaluation of the prediction contribution of each client under different data distributions. Based on the antibiased weighting of its prediction loss, the weight distribution of each client is effectively adjusted to fairly integrate the knowledge of participating clients. At the same time, the local model is split into a feature extractor and a classifier. Through differential contrast learning, the feature extractor is aligned with the global model in the feature space, while the classifier maintains personalized decision-making capabilities. HFedCKD effectively alleviates the knowledge offset caused by a low participation rate under data-free knowledge distillation and improves the performance and stability of the model. We conduct extensive experiments on image and IoT datasets to comprehensively evaluate and verify the generalization and robustness of the proposed HFedCKD framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。