arXiv:2605.06663cs.CL2026-05被引 2

提出可自动生成模块化专家的MoE模型,实现高效内存部署。

EMO: Pretraining Mixture of Experts for Emergent Modularity

论文配图:EMO: Pretraining Mixture of Experts for Emergent Modularity
图 1 · 摘自论文原文
  • 通过文档边界约束,让相似内容共享专家池,自发形成模块化结构。
  • 仅保留25%专家时性能下降仅1%,远优于传统MoE。
  • 专家按语义领域分工,适合需要灵活组合能力的场景。

大型语言模型通常以整体系统部署,即使应用只需部分能力(如代码、数学或领域知识)也需加载全部模型。虽然混合专家(MoE)理论上可通过激活部分专家降低计算量,但实践中限制特定领域只使用部分专家会导致性能严重下降,限制其在内存受限环境的应用,尤其当模型规模和稀疏性增加时。本文提出EMO,一种面向模块化的MoE架构,支持专家子集的独立使用与组合,无需人工预设先验。核心思路是促使同一文档内相似内容依赖相似专家,不同文档可使用不同专家池。这一简单约束使专家分组在预训练中自然涌现。我们在1T tokens上预训练了一个1B活跃参数、14B总参数的EMO模型。作为完整模型,其性能与标准MoE相当;关键优势在于支持选择性专家使用:仅保留25%(12.5%)专家时,性能绝对下降仅1%(3%),而标准MoE在此设置下完全失效。进一步发现,EMO中的专家子集在语义层面专业化(如数学、代码),而非传统MoE的低级语法特征。结果表明,EMO为大而稀疏模型提供了模块化、内存高效的部署路径,并开启可组合架构的新可能。

原文摘要 · Abstract (English)

Large language models are typically deployed as monolithic systems, requiring the full model even when applications need only a narrow subset of capabilities, e.g., code, math, or domain-specific knowledge. Mixture-of-Experts (MoEs) seemingly offer a potential alternative by activating only a subset of experts per input, but in practice, restricting inference to a subset of experts for a given domain leads to severe performance degradation. This limits their practicality in memory-constrained settings, especially as models grow larger and sparser. We introduce EMO, an MoE designed for modularity-the independent use and composition of expert subsets-without requiring human-defined priors. Our key idea is to encourage tokens from similar domains to rely on similar experts. Since tokens within a document often share a domain, EMO restricts them to select experts from a shared pool, while allowing different documents to use different pools. This simple constraint enables coherent expert groupings to emerge during pretraining using document boundaries alone. We pretrain a 1B-active, 14B-total EMO on 1T tokens. As a full model, it matches standard MoE performance. Crucially, it enables selective expert use: retaining only 25% (12.5%) of experts incurs just a 1% (3%) absolute drop, whereas standard MoEs break under the same setting. We further find that expert subsets in EMO specialize at semantic levels (e.g., domains such as math or code), in contrast to the low-level syntactic specialization observed in standard MoEs. Altogether, our results demonstrate a path toward modular, memory-efficient deployment of large, sparse models and open new opportunities for composable architectures.

MoE模块化稀疏模型专家机制

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