arXiv:2606.01062cs.AI2026-06中稿 · ICML被引 2

通过结构化聚合提升专家模型表现,无需增加计算负担

DAG-MoE: From Simple Mixture to Structural Aggregation in Mixture-of-Experts

论文配图:DAG-MoE: From Simple Mixture to Structural Aggregation in Mixture-of-Experts
图 1 · 摘自论文原文
  • 用动态结构替代加权求和,自动学习最优专家组合方式
  • 在预训练和微调中均超越传统MoE基线,性能更优
  • 适合追求高效推理与多步推理能力的大型语言模型研究者

Mixture-of-Experts(MoE)模型已成为大语言模型中解耦参数量与计算成本的主流方法,但有效扩展其性能仍具挑战。已有研究表明,细粒度专家能扩大专家组合空间并增强灵活性,但也带来显著路由开销,形成新的可扩展瓶颈。本文探索另一条扩展路径——专家输出的聚合方式。我们理论证明,将标准加权求和替换为结构化聚合,在不改变专家或路由器的前提下,可扩大专家组合空间,并支持单层内实现多步推理。为此,我们提出DAG-MoE,一种轻量级稀疏MoE框架,通过一个可学习模块自动确定所选专家间的最优聚合结构。在标准语言建模设置下的大量实验表明,DAG-MoE在预训练和微调阶段均持续提升性能,优于传统MoE基线。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) models have become a leading approach for decoupling parameter count from computational cost in large language models, yet effectively scaling MoE performance remains a challenge. Prior work shows that fine-grained experts enlarge the space of expert combinations and improve flexibility, but they also impose substantial routing overhead, creating a new scalability bottleneck. In this paper, we explore a complementary axis for scaling -- how expert outputs are aggregated. We theoretically show that replacing the standard weighted-summation aggregation with structural aggregation expands the expert-combination space without altering the experts or router, and enables possible multi-step reasoning within a single MoE layer. To this end, we propose DAG-MoE, a sparse MoE framework that employs a lightweight module to automatically learn the optimal aggregation structure among the selected experts. Extensive experiments under standard language modeling settings show that DAG-MoE consistently improves performance in both pretraining and fine-tuning, surpassing traditional MoE baselines.

MoE专家模型结构聚合大模型

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