mlsanityData to 14 Oct 2026

Squeezing More from Limited Data with Recursive Transformers

arXiv 2608.26973 ↗First posted 27 Aug 2026cs.CL cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Pre-training under limited data requires a different view of scaling than web-scale language modeling. With a fixed data budget but relatively abundant compute, increasing parameter count helps only up to an optimal scale; beyond that point, models overfit and generalization worsens. We study this behavior across 10M-100M word pre-training budgets, two corpora, and multiple downstream evaluations, and find that optimal size depends strongly on both the data budget and the downstream target. We argue that standard Transformers scale down poorly to this setting, because embeddings consume a large fraction of the parameter budget and per-token computation is tied to representational capacity. To address this coupling, we study recursive Transformers, reusing a shared block across depth to scale compute, together with factorized embeddings to reduce vocabulary-map parameters. We train three recursive models and find that they outperform standard Transformers at 10M and 100M words, while remaining competitive with BabyLM Challenge 2025 winners.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 39 references that are in the map. First posted on arXiv 27 Aug 2026.

Builds on

25 of 39 references.

References · 25

  1. 2017Attention Is All You Needcited by 60,361
  2. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  3. 2020Scaling Laws for Neural Language Modelscited by 6,586
  4. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  5. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,790
  6. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  7. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  8. 2018Universal Transformerscited by 647
  9. 2019Deep Equilibrium Modelscited by 615
  10. 2019BLiMP: The Benchmark of Linguistic Minimal Pairs for Englishcited by 564
  11. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
  12. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
13 more
  1. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 283
  2. 2020Query-Key Normalization for Transformerscited by 280
  3. 2020Understanding the Difficulty of Training Transformerscited by 210
  4. 2025NorMuon: Making Muon more efficient and scalablecited by 84
  5. 2024Relaxed Recursive Transformers: Effective Parameter Sharing with Layer-wise LoRAcited by 78
  6. 2025Nemotron-CLIMB: CLustering-based Iterative Data Mixture Bootstrapping for Language Model Pre-trainingcited by 66
  7. 2024Elements of World Knowledge (EWoK): A Cognition-Inspired Framework for Evaluating Basic World Knowledge in Language Modelscited by 55
  8. 2026Parcae: Scaling Laws For Stable Looped Language Modelscited by 55
  9. 2024Findings of the Second BabyLM Challenge: Sample-Efficient Pretraining on Developmentally Plausible Corporacited by 48
  10. 2025Understanding Transformer from the Perspective of Associative Memorycited by 34
  11. 2024GPT or BERT: why not both?cited by 27
  12. 2025AbbIE: Autoregressive Block-Based Iterative Encoder for Efficient Sequence Modelingcited by 7
  13. 2025Mask and You Shall Receive: Optimizing Masked Language Modeling For Pretraining BabyLMscited by 3

Built on by

Not cited yet.