mlsanityData to 7 Oct 2026

A Dual-Path Architecture for Scaling Compute and Capacity in LLMs

arXiv 2605.30202 ↗First posted 28 May 2026cs.CLcited by 2

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

  1. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  2. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  3. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  4. 2020GLU Variants Improve Transformercited by 1,724
  5. 2019Root Mean Square Layer Normalizationcited by 1,425
  6. 2023Scaling Vision Transformers to 22 Billion Parameterscited by 690
  7. 2018Universal Transformerscited by 613
  8. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
  9. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
  10. 2020Query-Key Normalization for Transformerscited by 266
  11. 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
  12. 2021PonderNet: Learning to Pondercited by 156
9 more
  1. 2024Nemotron-CC: Transforming Common Crawl into a Refined Long-Horizon Pretraining Datasetcited by 153
  2. 2025Scaling Latent Reasoning via Looped Language Modelscited by 145
  3. 2019Large Memory Layers with Product Keyscited by 120
  4. 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 120
  5. 2024OLMES: A Standard for Language Model Evaluationscited by 100
  6. 2019Augmenting Self-attention with Persistent Memorycited by 92
  7. 2026LoopFormer: Elastic-Depth Looped Transformers for Latent Reasoning via Shortcut Modulationcited by 48
  8. 2026Adaptive Loops and Memory in Transformers: Think Harder or Know More?cited by 11
  9. 2026Modalities, a PyTorch-native Framework For Large-scale LLM Training and Researchcited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.