Dense vs Sparse Pretraining at Tiny Scale: Active-Parameter vs Total-Parameter Matching
On the map: Mixture-of-experts → Expert routing strategies
Abstract (arXiv)
We study dense and mixture-of-experts (MoE) transformers in a tiny-scale pretraining regime under a shared LLaMA-style decoder training recipe. The sparse model replaces dense feed-forward blocks with Mixtral-style routed experts. Dense baselines are modestly width-resized to tightly match either active or total parameter budgets, while tokenizer, data, optimizer, schedule, depth, context length, normalization style, and evaluation protocol are held fixed. Our best sparse recipe uses four experts, top-2 routing, Switch-style load balancing, and router z-loss. In a three-seed full-data comparison, the dense active-match model reaches 1.6545 +/- 0.0012 best validation loss, the MoE reaches 1.5788 +/- 0.0020, and the dense total-match model reaches 1.5608 +/- 0.0025. This yields a matched-active gap of 0.0758 +/- 0.0021 in the MoE's favor and a matched-total gap of 0.0180 +/- 0.0020 in the dense model's favor. Across training, the matched-active advantage grows while the matched-total dense advantage narrows sharply. In this sub-25M-parameter regime, MoE therefore improves validation loss under active-parameter matching but does not surpass dense training at equal total stored capacity.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 17 of its 17 references that are in the map. First posted on arXiv 13 May 2026.
Builds on
17 of 17 references.
Closest ancestors · 1
Same direction, newest first
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 16
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2024Mixtral of Expertscited by 1,823
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,135
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 897
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 867
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 399
- 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
- 2022Unified Scaling Laws for Routed Language Modelscited by 253
- 2022MegaBlocks: Efficient Sparse Training with Mixture-of-Expertscited by 209
- 2021Efficient Large Scale Language Modeling with Mixtures of Expertscited by 175
4 more
- 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 173
- 2025Parameters vs FLOPs: Scaling Laws for Optimal Sparsity for Mixture-of-Experts Language Modelscited by 50
- 2025Joint MoE Scaling Laws: Mixture of Experts Can Be Memory Efficientcited by 31
- 2025Mixture-of-Experts Can Surpass Dense LLMs Under Strictly Equal Resourcecited by 8
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026DECO: Sparse Mixture-of-Experts with Dense-Comparable Performance on End-Side Devices
- 2025Dense Backpropagation Improves Training for Sparse Mixture-of-Experts
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers
- 2025Continual Pre-training of MoEs: How robust is your router?
- 2025Ban&Pick: Enhancing Performance and Efficiency of MoE-LLMs via Smarter Routing
- 2026MoE Routing Testbed: Studying Expert Specialization and Routing Behavior at Small Scale
- 2025Sigma-MoE-Tiny Technical Report
- 2025BlockFFN: Towards End-Side Acceleration-Friendly Mixture-of-Experts with Chunk-Level Activation Sparsity