arXiv:2602.14159cs.LG2026-02被引 5

提出两种正则化损失,提升MoE专家专业化与路由效率。

Synergistic Intra- and Cross-Layer Regularization Losses for MoE Expert Specialization

  • 通过层内损失降低专家激活相似度,促进知识互补
  • 通过跨层损失增强相邻层路由一致性,稳定专家路径
  • 无需修改架构,适配多种MoE模型,加速推理

稀疏混合专家(MoE)模型可高效扩展Transformer,但存在专家重叠——即多个专家对相同输入产生冗余表示和路由模糊,导致模型容量严重浪费。现有结构改进如DeepSeekMoE虽能促进专业化,但需大幅修改架构且仅依赖层内信号。本文提出两种即插即用的正则化损失,无需改动路由器或模型架构即可提升MoE专业化与路由效率。第一,层内专业化损失惩罚相同输入下专家SwiGLU激活之间的余弦相似度,促使专家学习互补知识;第二,跨层耦合损失最大化相邻层共同选择的Top-$k$路由概率,建立网络深度上的连贯专家路径,同时强化层内专业化。两种损失与标准负载均衡损失正交,兼容DeepSeekMoE共享专家架构及原始top-$k$ MoE架构。我们在Megatron-LM中实现为模块化组件。大规模实验在预训练、微调和零样本基准上均显示持续性能提升,专家专业化更高,路由熵更低;整体提升带来更稳定的专家路径,从而加快推理速度。

原文摘要 · Abstract (English)

Sparse Mixture-of-Experts (MoE) models scale Transformers efficiently but suffer from expert overlap -- redundant representations across experts and routing ambiguity, resulting in severely underutilized model capacity. While architectural solutions like DeepSeekMoE promote specialization, they require substantial structural modifications and rely solely on intra-layer signals. In this paper, we propose two plug-and-play regularization losses that enhance MoE specialization and routing efficiency without modifying router or model architectures. First, an intra-layer specialization loss penalizes cosine similarity between experts' SwiGLU activations on identical tokens, encouraging experts to specialize in complementary knowledge. Second, a cross-layer coupling loss maximizes joint Top-$k$ routing probabilities across adjacent layers, establishing coherent expert pathways through network depth while reinforcing intra-layer expert specialization. Both losses are orthogonal to the standard load-balancing loss and compatible with both the shared-expert architecture in DeepSeekMoE and vanilla top-$k$ MoE architectures. We implement both losses as a drop-in Megatron-LM module. Extensive experiments across pre-training, fine-tuning, and zero-shot benchmarks demonstrate consistent task gains, higher expert specialization, and lower-entropy routing; together, these improvements translate into faster inference via more stable expert pathways.

MoE专家专业化路由优化正则化

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