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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Krajewski, Jakub, Chochowski, Marcin, Korzekwa, Daniel
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915321471827968
author Krajewski, Jakub
Chochowski, Marcin
Korzekwa, Daniel
author_facet Krajewski, Jakub
Chochowski, Marcin
Korzekwa, Daniel
contents Mixture of Experts (MoE) architectures have emerged as pivotal for scaling Large Language Models (LLMs) efficiently. Fine-grained MoE approaches - utilizing more numerous, smaller experts - have demonstrated potential in improving model convergence and quality. 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 show 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.
format Preprint
id arxiv_https___arxiv_org_abs_2506_02890
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Scaling Fine-Grained MoE Beyond 50B Parameters: Empirical Evaluation and Practical Insights
Krajewski, Jakub
Chochowski, Marcin
Korzekwa, Daniel
Machine Learning
Artificial Intelligence
Computation and Language
Mixture of Experts (MoE) architectures have emerged as pivotal for scaling Large Language Models (LLMs) efficiently. Fine-grained MoE approaches - utilizing more numerous, smaller experts - have demonstrated potential in improving model convergence and quality. 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 show 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.
title Scaling Fine-Grained MoE Beyond 50B Parameters: Empirical Evaluation and Practical Insights
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2506.02890