arXiv:2606.01607cs.LGcs.AI2026-06中稿 · IJCNN 2026

通过重要特征加权的多教师蒸馏,提升异构联邦学习性能

FedMTFI: Feature Importance Based Optimized Multi Teacher Knowledge Distillation in Heterogeneous Federated Learning Environment

论文配图:FedMTFI: Feature Importance Based Optimized Multi Teacher Knowledge Distillation in Heterogeneous Federated Learning Environment
图 1 · 摘自论文原文
  • 按设备类型分组训练,每组用本地数据更新模型
  • 利用SHAP值筛选关键特征,蒸馏后准确率显著提升
  • 适合设备差异大、数据非独立同分布的现实场景

联邦学习(FL)是一种去中心化协作训练方法,可在不暴露原始数据的情况下实现模型联合训练。在实际应用中,设备间数据分布不均,且计算能力与内存容量差异显著,导致系统性能难以保持一致。为此,我们提出FedMTFI,将多教师知识蒸馏(MTKD)与特征重要性结合,以优化异构环境下的联邦学习过程。客户端根据硬件和模型类型聚类,每组在非独立同分布(non-IID)数据上训练专属模型,仅使用本地私有数据更新。服务器通过FedAvg聚合各集群模型,形成多个原型模型作为教师,指导全局学生模型的训练。创新之处在于引入Shapley值(SHAP)识别关键特征,在蒸馏过程中加以强调,从而提升模型精度与可解释性。实验表明,相比传统方法,FedMTFI在非IID数据下表现更优,整体准确率更高。

原文摘要 · Abstract (English)

Federated learning (FL) is a decentralized approach that enables collaborative model training without exposing raw data. Instead of transferring sensitive data, it allows devices to share only model weights, keeping personal data locally and secure. However, in real world settings, the data held by devices is often not evenly distributed and devices mostly differ in computing power and memory capacity. These differences make FL harder to maintain consistent performance across the system. To address these issues, we propose FedMTFI, a novel architecture that combines multi-teacher knowledge distillation (MTKD) with feature importance to improve the FL process in heterogeneous environments. In FedMTFI, clients are clustered based on similar hardware and model types. Each cluster trains a specific model on not independently and identically distributed (non-IID) data. Within a cluster, every client updates that model using only its own local private data. The server then aggregates the locally trained models in each cluster using FedAvg to form multiple prototype models. Then these prototypes serve as teacher models to train a global generalized student model using MTKD. What makes FedMTFI more unique is the integration of Shapley values (SHAP) to emphasize important features during distillation, which enhances both accuracy and interpretability. Experimental results show that FedMTFI achieves higher accuracy than traditional FL algorithms and performs more effectively under non-IID data conditions.

联邦学习知识蒸馏特征重要性异构环境

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