arXiv:2506.18945cs.LGcs.CL2025-06被引 6

让专家模型逐轮协作,提升推理能力并节省显存。

Chain-of-Experts: Unlocking the Communication Power of Mixture-of-Experts Models

  • 专家按链式顺序迭代处理,每轮动态选不同专家。
  • 数学推理任务验证损失从1.20降至1.12,显存降低17.6%-42%。
  • 适合追求高效高表达力模型的开发者与研究者。

我们提出链式专家(Chain-of-Experts, CoE),一种新型混合专家(MoE)架构,在每一层内引入专家间的序列通信机制。与传统并行独立运作的专家不同,CoE 在层内对令牌进行逐轮迭代处理,通过每轮设置专用路由器实现动态专家选择。这使得令牌可在多轮中重新评估并切换专家,而非静态分配。该设计提升了专家组合多样性,增强了模型表征能力。在固定计算量下,CoE 在数学推理任务上将验证损失从1.20降至1.12。此外,它提供了一种新扩展维度:通过专家迭代增加深度,可替代3倍专家数量(宽度)实现相当性能,同时减少17.6%-42%内存占用。分析表明,其优势源于迭代残差结构和由迭代路由强化的专家专业化。代码已开源。

原文摘要 · Abstract (English)

We propose Chain-of-Experts (CoE), a new Mixture-of-Experts (MoE) architecture that introduces sequential expert communication within each layer. Unlike traditional MoE models, where experts operate independently in parallel, CoE processes tokens iteratively across a chain of experts inside a layer. To support dynamic expert selection across iterations, CoE employs a dedicated router at each iteration step within a layer. This design allows tokens to re-evaluate and select different experts during each iteration, rather than being statically assigned. As a result, CoE introduces a flexible routing mechanism that increases the diversity of expert combinations and enriches the model's representational capacity. CoE demonstrates improved performance under fixed compute: on math reasoning tasks, it reduces validation loss from 1.20 to 1.12 compared to a standard MoE. Beyond performance, CoE offers a new scaling axis: depth through expert iteration, which complements conventional width/depth scaling. For example, using 2x iterations matches the performance of 3x expert selections (in width), while reducing memory usage by 17.6-42% relative to other scaling strategies. Our analysis reveals that CoE's benefits stem from its iterative residual structure and enhanced expert specialization empowered by iterative routing, which together unlock more expressive representations. Code is available at https://github.com/ZihanWang314/coe.

MoE专家模型推理优化深度学习

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