发现MoE模型推理能力受活跃计算量与参数数据比共同决定。
Optimal Sparsity of Mixture-of-Experts Language Models for Reasoning Tasks
- 通过控制活跃计算和总参数,分离预训练损失与下游性能。
- 相同训练损失下,活跃计算越多,推理准确率越高;参数数据比最优时推理最佳。
- 该结论不依赖强化学习或测试时增加计算,适合构建高效推理系统。
经验缩放定律推动了大语言模型的发展,但其系数会随模型架构或数据流水线变化而改变。混合专家(MoE)模型已成为当前先进系统中的标准,引入了稀疏性这一新维度,而现有密集模型的前沿研究忽略了这一点。我们研究了MoE稀疏性对两类能力范式的影响:记忆能力和推理能力。在固定计算预算下,训练不同总参数量、活跃参数量和top-k路由的MoE家族,从而解耦预训练损失与下游准确率。结果揭示两条原则:第一,活跃浮点运算量(Active FLOPs)——具有相同训练损失但更高活跃计算的模型,推理准确率更高;第二,每参数总令牌数(TPP)——记忆任务随参数增多而提升,而推理任务受益于最优的TPP,表明推理是数据密集型的。无论采用强化学习后训练(GRPO)还是增加测试时计算,这些趋势均不变。因此我们主张,最优的MoE稀疏性应由活跃浮点运算量和TPP共同决定,重新定义了经典计算最优缩放图景。模型检查点、代码和日志已开源:https://github.com/rioyokotalab/optimal-sparsity。
原文摘要 · Abstract (English)
Empirical scaling laws have driven the evolution of large language models (LLMs), yet their coefficients shift whenever the model architecture or data pipeline changes. Mixture-of-Experts (MoE) models, now standard in state-of-the-art systems, introduce a new sparsity dimension that current dense-model frontiers overlook. We investigate how MoE sparsity influences two distinct capability regimes: memorization skills and reasoning skills. By training MoE families that vary total parameters, active parameters, and top-$k$ routing under fixed compute budgets, we disentangle pre-training loss from downstream accuracy. Our results reveal two principles. First, Active FLOPs: models with identical training loss but greater active compute achieve higher reasoning accuracy. Second, Total tokens per parameter (TPP): memorization tasks improve with more parameters, while reasoning tasks benefit from optimal TPP, indicating that reasoning is data-hungry. Neither reinforcement learning post-training (GRPO) nor increased test-time compute alters these trends. We therefore argue that optimal MoE sparsity must be determined jointly by active FLOPs and TPP, revising the classical picture of compute-optimal scaling. Our model checkpoints, code and logs are open-source at https://github.com/rioyokotalab/optimal-sparsity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。