Scaling Fine-Grained MoE Beyond 50B Parameters: Empirical Evaluation and Practical Insights

Jakub Krajewski, Marcin Chochowski, Daniel Korzekwa

Keywords: Large Language Models, Mixture of Experts, MoE, granularity, scaling

TL;DR: The paper proposes a set of training recipes and provides a comprehensive empirical evaluation of fine-grained MoE.

Abstract: Mixture of Experts (MoE) architectures have emerged as pivotal for scaling Large Language Models (LLMs) efficiently. While fine-grained MoE approaches - utilizing more numerous, smaller experts - have shown promise, quantification of their efficiency gains at large scales remains crucial. This work proposes a set of training recipes and provides a comprehensive empirical evaluation of fine-grained MoE, directly comparing its scaling properties against standard MoE configurations for models with up to 56B total (17B active) parameters. We investigate convergence speed, model performance on downstream benchmarks, and practical training considerations across various setups. Overall, at the largest scale we observe that fine-grained MoE achieves better validation loss and higher accuracy across a set of downstream benchmarks. This study offers empirical grounding and practical insights for leveraging fine-grained MoE in the development of future large-scale models.

Submission Number: 19