Universal Transformer validation cost billions, but MoE scaling still favors layer-loop
teortaxesTex · x · 2026-07-21
Universal Transformer validation cost billions, but the approach fails to scale to billion-parameter MoE models
Zitian Gao says the team spent billions of USD validating the classic Universal Transformer approach, which they call the “model-loop.” The main takeaway:
- The model-loop does not scale well to MoE models with billions of parameters.
- In later pretraining stages, the conventional model-loop underperforms the layer-loop.
- It also needs more complicated pipeline parallelism as models get larger, which can slow training.
- Because non-loop models have more freedom in parameterization, comparing them purely on theoretical FLOPs is misleading.
- The paper therefore compares against a non-loop baseline matched on actual training FLOPs, which the authors считают more practical for industrial-scale pretraining.
The discussion frames the layer-loop as both more infrastructure-friendly and better performing in late-stage pretraining.
Related event: Loopie Looping Transformers Match Larger Models at Fraction of Cost(7 posts)→
More from Research
- Nature paper images cellular activity across all organs, revealing body-wide circuits — arjunrajlab · 2026-09-11
- SignNet 1M Dataset Released for Sign Language Research — ducha_aiki · 2026-09-11
- ECCV26 Oral: Flow Matching Enables Single-Stage Multi-View Point Cloud Registration — ducha_aiki · 2026-09-11
- InFlux++ Method Released — ducha_aiki · 2026-09-11
- Skyfall GS Uses Flux to Refine Gaussian Splatting, Accepted at ECCV 2026 — ducha_aiki · 2026-09-11
- Could 10k agents discover learning methods beyond backprop, or just tweak existing ones? — SeunghyunSEO7 · 2026-09-11