让专家跨层复用,提升大模型效率与表现。
ReXMoE: Reusing Experts with Minimal Overhead in Mixture-of-Experts
- 设计跨层复用专家机制,突破单层专家池限制。
- 在0.5B至7B参数模型上均实现性能提升。
- 适合追求高效扩展的大模型研究者使用。
Mixture-of-Experts (MoE) 架构已成为扩展大型语言模型(LLMs)的有前景方法。通过每标记仅激活部分专家,MoE 提升了效率。近期研究显示,细粒度专家显著增强了活跃专家的组合灵活性并提升了模型表达能力。然而,这种设计受制于层内路由机制:每层仅限于自身专家池,导致在固定参数预算下,需在专家维度与路由多样性间权衡。本文提出 ReXMoE,一种新 MoE 架构,通过允许路由器跨相邻层复用专家,超越现有层内路由方式。ReXMoE 将专家维度与每层预算解耦,可在不牺牲单个专家容量或增加总体参数的情况下,实现更丰富的专家组合。为此,我们提出渐进式扩展路由(PSR)策略,在训练中逐步扩大候选专家池。实验表明,ReXMoE 在语言建模与下游任务中均取得改进。在0.5B至7B参数、多种架构的模型上进行的广泛实验显示,固定架构维度下,ReXMoE 均持续提升性能,证实其为参数高效且可扩展的 MoE 模型新范式。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) architectures have emerged as a promising approach to scale Large Language Models (LLMs). MoE boosts the efficiency by activating a subset of experts per token. Recent works show that fine-grained experts substantially enriches the combinatorial flexibility of active experts and enhances model expressiveness. However, such a design is fundamentally limited by the layer-local routing mechanism: each layer is restricted to its own expert pool. This requires a careful trade-off between expert dimensionality and routing diversity given fixed parameter budgets. We describe ReXMoE, a novel MoE architecture that improves routing beyond the existing layer-local approaches by allowing routers to reuse experts across adjacent layers. ReXMoE decouples expert dimensionality from per-layer budgets, enabling richer expert combinations without sacrificing individual expert capacity or inflating overall parameters. To this end, we propose a new progressive scaling routing (PSR) strategy to gradually increase the candidate expert pool during training. As a result, ReXMoE improves both language modeling and downstream task performance. Extensive experiments on models ranging from 0.5B to 7B parameters across different architectures demonstrate that ReXMoE consistently improves performance under fixed architectural dimensions, confirming ReXMoE as new design paradigm for parameter-efficient and scalable MoE-based LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。