arXiv:2602.14301cs.LGcs.AI2026-02

让边缘设备协作训练大模型,通信少、效果接近集中式训练。

DeepFusion: Accelerating MoE Training via Federated Knowledge Distillation from Heterogeneous Edge Devices

  • 边缘设备各自训练小模型,通过知识蒸馏融合异构知识。
  • 提出视图对齐注意力模块,解决模型架构差异导致的蒸馏失效问题。
  • 在医疗金融数据上实现71%通信节省,性能接近中心化训练。

基于混合专家(MoE)的大语言模型(如Qwen-MoE和DeepSeek-MoE)正推动自然语言生成的发展。然而,这些模型需要大量且多样化的训练数据。联邦学习(FL)通过利用异构边缘设备的私有数据,在保护隐私的前提下支持MoE训练。但传统方法要求设备运行本地MoE模型,这对资源受限设备不现实。为此,我们提出DeepFusion,首个可扩展的联邦MoE训练框架,通过联邦知识蒸馏融合异构设备上的语言模型知识,构建知识丰富的全局MoE模型。具体而言,每个设备可独立配置并训练适配自身需求与硬件限制的本地LLM。此外,我们提出新型视图对齐注意力(VAA)模块,整合全局MoE模型的多阶段特征表示,构建与本地LLM预测视角一致的表示,从而实现跨架构的有效知识蒸馏。通过显式对齐预测视角,VAA解决了传统联邦知识蒸馏中因模型架构与预测行为差异导致的视图不匹配问题。在行业级MoE模型(Qwen-MoE和DeepSeek-MoE)及真实数据集(医疗与金融)上的实验表明,DeepFusion性能接近集中式训练。相比关键联邦MoE基线,其通信成本降低最高达71%,词元困惑度改善最高达5.28%。

原文摘要 · Abstract (English)

Recent Mixture-of-Experts (MoE)-based large language models (LLMs) such as Qwen-MoE and DeepSeek-MoE are transforming generative AI in natural language processing. However, these models require vast and diverse training data. Federated learning (FL) addresses this challenge by leveraging private data from heterogeneous edge devices for privacy-preserving MoE training. Nonetheless, traditional FL approaches require devices to host local MoE models, which is impractical for resource-constrained devices due to large model sizes. To address this, we propose DeepFusion, the first scalable federated MoE training framework that enables the fusion of heterogeneous on-device LLM knowledge via federated knowledge distillation, yielding a knowledge-abundant global MoE model. Specifically, DeepFusion features each device to independently configure and train an on-device LLM tailored to its own needs and hardware limitations. Furthermore, we propose a novel View-Aligned Attention (VAA) module that integrates multi-stage feature representations from the global MoE model to construct a predictive perspective aligned with on-device LLMs, thereby enabling effective cross-architecture knowledge distillation. By explicitly aligning predictive perspectives, VAA resolves the view-mismatch problem in traditional federated knowledge distillation, which arises from heterogeneity in model architectures and prediction behaviors between on-device LLMs and the global MoE model. Experiments with industry-level MoE models (Qwen-MoE and DeepSeek-MoE) and real-world datasets (medical and finance) demonstrate that DeepFusion achieves performance close to centralized MoE training. Compared with key federated MoE baselines, DeepFusion reduces communication costs by up to 71% and improves token perplexity by up to 5.28%.

联邦学习MoE模型知识蒸馏边缘计算

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