arXiv:2603.00054cs.LGcs.AI2026-03被引 1

通过引导专家分流,让大模型专家分工更明确。

Expert Divergence Learning for MoE-based Language Models

  • 用领域标签设计辅助损失,强制不同领域专家路由分化。
  • 150亿参数模型训练后,语言建模损失更低,下游任务提升显著。
  • 几乎不增加计算开销,适合大规模MoE模型预训练。

Mixture-of-Experts(MoE)架构是扩展语言模型的有效手段,但常出现专家同质化问题,即各专家学习到相似功能,限制了其潜力。为此,我们提出专家分歧学习(Expert Divergence Learning),一种新型预训练策略,显式促进专家间功能分化。该方法引入基于标签的辅助损失,利用预训练语料中固有的领域标签,最大化不同数据领域下专家路由分布之间的Jensen-Shannon散度。此优化目标促使模型对不同领域发展出差异化的路由策略,同一领域则趋于一致,从而实现涌现且有序的专家专业化。我们在从头预训练高达150亿参数的MoE模型上验证该方法。实验表明,采用专家分歧学习的模型不仅语言建模损失更低,还在多样化的下游基准测试中表现显著提升。进一步分析证实,该方法有效缓解专家同质化,增强功能分化,且训练期间计算开销几乎可忽略。

原文摘要 · Abstract (English)

The Mixture-of-Experts (MoE) architecture is a powerful technique for scaling language models, yet it often suffers from expert homogenization, where experts learn redundant functionalities, thereby limiting MoE's full potential. To address this, we introduce Expert Divergence Learning, a novel pre-training strategy that explicitly encourages functional specialization among experts. Our method incorporates a label-driven auxiliary loss that leverages domain labels inherent in pre-training corpora to maximize the Jensen-Shannon Divergence between the expert routing distributions of different data domains. This optimization objective guides the model to develop diverged routing policies for varied domains and closer routing policies for the same domain, which leads to emergent and organized expert specialization. We validate our approach by pre-training MoE models of up to 15 billion parameters from scratch. Experimental results demonstrate that models trained with Expert Divergence Learning not only achieve a lower language modeling loss but also exhibit significant performance improvements across a diverse range of downstream benchmarks. Further analysis confirms that our method effectively mitigates expert homogenization and brings greater functional specialization, all with negligible computational overhead during training.

MoE专家分化语言模型预训练

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