mlsanityData to 7 Oct 2026

Evaluating Tiny Recursive Models Across Training for Code Generation

arXiv 2608.29376 ↗First posted 29 Aug 2026cs.AI cs.LG cs.SEcited by 0

Not on the map. Why?

Abstract (arXiv)

Code generation increasingly relies on large transformer models, whose capability advances with scale. Yet such a scale is costly, creating demand for small models, especially where data is limited. Recursive models address this by reusing a single block to add depth rather than stacking independent layers. Such models are typically evaluated by teacher-forced fit (next-token loss on ground-truth prefixes) or task accuracy, at a single checkpoint, whereas code is produced by free-running generation, where the model extends its own output. Whether a teacher-forced advantage survives free-running generation, and whether it holds across training, remains open. To study both, we compare a ~28M-parameter autoregressive Tiny Recursive Model (TRM-AR) on natural-language-to-Python code generation against parameter-matched and depth-matched controls, tracking fit and generation across 40 epochs and three seeds. The fit ranking between the recursive model and the depth-matched control reverses twice. Selecting each checkpoint by validation loss and examining the trajectory yields a consistent comparison. At equal parameters, TRM-AR fits, generates, and generalizes better than the parameter-matched control while recovering approximately 45% of the validation-loss gap and 57% of the generation-quality gap between the two controls, at roughly 175 times the per-step cost of the parameter-matched control. However, at equal effective depth, the larger transformer fits and generates better at its validation optimum, suggesting TRM-AR's advantage lies in resistance to overfitting, not greater capability. These findings suggest that recursive code generation models should be evaluated jointly on fit and generation across the training trajectory rather than at a single checkpoint.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 41 of its 50 references that are in the map. First posted on arXiv 29 Aug 2026.

Builds on

41 of 50 references.

References · 41

  1. 2017Attention Is All You Needcited by 59,932
  2. 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
  3. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  4. 2020Scaling Laws for Neural Language Modelscited by 6,411
  5. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  6. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  7. 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
  8. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  9. 2020GLU Variants Improve Transformercited by 1,724
  10. 2022Competition-Level Code Generation with AlphaCodecited by 1,427
  11. 2019Root Mean Square Layer Normalizationcited by 1,425
  12. 2022CodeGen: An Open Large Language Model for Code with Multi-Turn Program Synthesiscited by 876
29 more
  1. 2023StarCoder: may the source be with you!cited by 790
  2. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 779
  3. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
  4. 2024A Survey on Large Language Models for Code Generationcited by 646
  5. 2019Green AIcited by 620
  6. 2018Universal Transformerscited by 613
  7. 2015Sequence Level Training with Recurrent Neural Networkscited by 568
  8. 2023Textbooks Are All You Needcited by 561
  9. 2020CodeBLEU: a Method for Automatic Evaluation of Code Synthesiscited by 468
  10. 2023Scaling Data-Constrained Language Modelscited by 391
  11. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
  12. 2023CodeT5+: Open Code Large Language Models for Code Understanding and Generationcited by 327
  13. 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 223
  14. 2016Professor Forcing: A New Algorithm for Training Recurrent Networkscited by 214
  15. 2025Hierarchical Reasoning Modelcited by 167
  16. 2024The pitfalls of next-token predictioncited by 156
  17. 2025Scaling Latent Reasoning via Looped Language Modelscited by 145
  18. 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 137
  19. 2022Why Exposure Bias Matters: An Imitation Learning Perspective of Error Accumulation in Language Generationcited by 106
  20. 2021The Efficiency Misnomercited by 98
  21. 2024Relaxed Recursive Transformers: Effective Parameter Sharing with Layer-wise LoRAcited by 72
  22. 2020Train Large, Then Compress: Rethinking Model Size for Efficient Training and Inference of Transformerscited by 67
  23. 2023Paloma: A Benchmark for Evaluating Language Model Fitcited by 53
  24. 2024Instruction Tuning With Loss Over Instructionscited by 40
  25. 2025Parallel Loop Transformer for Efficient Test-Time Computation Scalingcited by 23
  26. 2024Position: Why We Must Rethink Empirical Research in Machine Learningcited by 19
  27. 2025On the Effectiveness of LLM-as-a-judge for Code Generation and Summarizationcited by 17
  28. 2026Anytime Pretraining: Horizon-Free Learning-Rate Schedules with Weight Averagingcited by 8
  29. 2026Tiny Autoregressive Recursive Modelscited by 6

Built on by

Not cited yet.