A Dual-Path Architecture for Scaling Compute and Capacity in LLMs
Not on the map. Why?
Abstract (arXiv)
Looped transformers apply a shared block multiple times and have emerged as a parameter-efficient route to scaling compute in language models. However, at fixed FLOPs a looped model has strictly less capacity than a baseline transformer. We propose a novel dual-path block that can flexibly scale compute, the number of sequential operations applied to a hidden state, and capacity, the parameters available at a single step. For this we expose both axes as parallel pathways within a single layer: a deep sublayer re-applied K times with shared parameters, and a wide sublayer with an enlarged feed-forward network applied once. Independent per-token gates combine both axes and allow detailed per-token routing analyses. We show that across two FLOP budgets, our dual-path model surpasses iso-FLOP matched models on language modeling and downstream evaluations, while using fewer parameters than the baseline at matched FLOPs. The learned gates are directly interpretable and show systematic per-token allocation with function words and lexical content trend wide, while punctuation, symbols, and arithmetic tokens trend deep.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 21 of its 27 references that are in the map. First posted on arXiv 28 May 2026.
Builds on
21 of 27 references.
References · 21
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2020GLU Variants Improve Transformercited by 1,724
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2023Scaling Vision Transformers to 22 Billion Parameterscited by 690
- 2018Universal Transformerscited by 613
- 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
- 2020Query-Key Normalization for Transformerscited by 266
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
- 2021PonderNet: Learning to Pondercited by 156
9 more
- 2024Nemotron-CC: Transforming Common Crawl into a Refined Long-Horizon Pretraining Datasetcited by 153
- 2025Scaling Latent Reasoning via Looped Language Modelscited by 145
- 2019Large Memory Layers with Product Keyscited by 120
- 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 120
- 2024OLMES: A Standard for Language Model Evaluationscited by 100
- 2019Augmenting Self-attention with Persistent Memorycited by 92
- 2026LoopFormer: Elastic-Depth Looped Transformers for Latent Reasoning via Shortcut Modulationcited by 48
- 2026Adaptive Loops and Memory in Transformers: Think Harder or Know More?cited by 11
- 2026Modalities, a PyTorch-native Framework For Large-scale LLM Training and Researchcited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.
- 2026Loop the Loopies!cited by 4
- 2026LoopMTP: A looped transformer guided by latent multi-token predictioncited by 3
- 2026LoopMTP: A looped transformer guided by latent multi-token prediction4 Aug 2026
- 2026Loop the Loopies!17 Jul 2026