Atindra Jha, Margaret Li, Jure Leskovec, Percy Liang, Luke Zettlemoyer
Mixture-of-Experts language models overfit faster than dense models when training data is repeated, with overfitting severity governed by total rather than active parameters.
Compute-matched controlled sweeps training dense and MoE transformers (80M, 200M, 1B active parameters; sparsity 2x to 32x) on OLMoE data domains (DCLM web crawl, StarCoder code, peS2o, Wikipedia), fixing total token budget T=20*N_active and varying repetition rates R from 1 to 1024, with held-out cross-entropy, downstream tasks, and mechanistic router/expert analyses.
80M dense models tolerate up to 8x repetition with minimal degradation; MoEs begin degrading at 4x and lose their unique-data advantage entirely at 32x, underperforming dense. With masking regularization, MoEs outperform dense even past 64x repetition, but no regularizer matches all-unique-data performance. Effects are consistent across data domains and data-quality filtering, and depend on total parameters: 200M dense sits between 80M-active MoEs of 158M and 244M total parameters.
The authors concede no regularization method fully matches all-unique-data performance and that regularization only mitigates, not eliminates, overfitting. Unstated limits: models are tiny by frontier standards (max 1B active, 8.5B total), results are pre-training cross-entropy oriented with downstream tasks relegated to appendix, compute-matching uses Chinchilla-style ratios that may not transfer to frontier-scale training, and the reversal of validation loss at extreme repetition suggests the degradation story is non-monotonic and not fully understood.
Small-scale lab models trained on academic corpora may not predict behavior of frontier MoEs trained once on hundreds of billions of unique tokens, where repetition rates are far below 4x. The interaction could be an artifact of the fixed-token compute-matching regime rather than an intrinsic property of sparsity. Validation loss recovery at extreme repetition complicates the clean overfitting narrative.
Modest near-term relevance: if unique-text supply is a binding constraint, MoE compute-efficiency gains may be partially clawed back by higher data requirements, slightly raising the effective data cost of sparse frontier training; this is an incremental cost-curve adjustment, not a regime change in the next 12 to 24 months.
Technical Capability: Practitioners choosing between dense and MoE architectures under data constraints learn that sparsity's efficiency benefit is conditional on unique-data budget and can be partly restored with dropout or output masking.
Market & Valuation: To the extent MoE compute-efficiency underpins frontier cost claims, a data-side penalty on sparsity slightly complicates the compute-efficiency advantage cited in lab cost narratives.