arXiv:2609.08690cs.LGcs.AI2026-09

发现稀疏MoE模型超参数需随激活比例调整,打破传统缩放规律。

Hyperparameter Scaling Laws Across MoE Sparsity

论文配图:Hyperparameter Scaling Laws Across MoE Sparsity
图 1 · 摘自论文原文
  • 提出新型缩放规律,将激活比例作为关键因子
  • 在20万亿令牌上验证,120亿参数模型预测准确
  • 适合大规模稀疏模型设计与跨架构迁移研究者

Mixture-of-Experts(MoE)模型可在不显著增加训练算力的前提下扩展模型容量,但提升稀疏度会使超参数迁移变得不可靠。本文表明,传统超参数缩放规律无法适用于超稀疏MoE:最优学习率与批次大小随激活比例变化,且这种变化无法仅由总参数量或激活参数量解释。我们进行了1800次预训练实验,覆盖六种激活参数规模,最大模型达60亿非嵌入参数,处理约20万亿令牌,耗时约20万等效H800 GPU小时。结果揭示两种缩放模式:在固定稀疏度下,最优批次大小与训练令牌数$D$呈幂律关系;最优学习率随训练算力$C$变化,对模型规模与数据分配保持鲁棒性。跨稀疏度时,激活比例$A$以乘法幂律形式同时影响两者。该统一规律可跨稀疏度迁移。大规模评估显示其优于其他函数形式。在120亿参数、仅1/64专家激活的超稀疏模型上,预测超参数接近实测最优,支持模型规模与稀疏度联合外推。进一步实验验证了跨专家粒度迁移,并分离出激活比例的影响。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) models expand model capacity without a proportional increase in training compute, but increasing sparsity makes reliable hyperparameter transfer challenging. In this work, we show that conventional hyperparameter scaling laws are insufficient for ultra-sparse MoEs: the optimal learning rate and batch size vary with activation ratio, and these shifts cannot be explained by either total or activated parameter count alone. To characterize this dependence, we conduct 1,800 pre-training runs spanning six activated-parameter scales and models with up to 6B total non-embedding parameters, processing approximately 20 trillion tokens at a cost of 200,000 equivalent H800 GPU-hours. Our results reconcile conflicting findings in prior work by revealing two scaling regimes. At fixed sparsity, the optimal batch size follows a power-law relationship with training tokens $D$, whereas the optimal learning rate scales with training compute $C$ and remains robust to the allocation between model size and data. Across sparsity levels, the activation ratio $A$ enters both relationships as an additional multiplicative power-law factor. These observations lead to unified hyperparameter scaling laws that transfer across MoE sparsity levels. Large-scale evaluation shows that the scaling form outperforms alternative functional forms. On a held-out ultra-sparse MoE with 12B total parameters and only 1/64 of its experts activated, the predicted hyperparameters remain close to the observed optima, supporting joint extrapolation across model scale and sparsity. Further experiments demonstrate transfer across expert granularities and isolate the effect of activation ratio from that of total expert count.

MoE超参数缩放规律稀疏性

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