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
- Stanford Data-Centric AI Workshop reopens site, extends poster deadline — jure · 2026-10-26
- 'Vibe verification': ~60% of AI-written formal specs prove the wrong property — basedjensen · 2026-10-26
- Ito: a browser that detects browsing intent and auto-organizes tabs into Flows — andrewkuznet · 2026-10-26
- Cornell study says brains process language as linear sequences, challenging Chomsky's hierarchy — anirbanbandyo · 2026-10-26
- Hi3D V1.0 open-sourced with 2M training assets and 90% slicing success rate — kimmonismus · 2026-10-26
- Ortet evaluates 34 open-weight models on 50 health tasks, finds general leaderboards insufficient — elmanmansimov · 2026-10-26