mlsanityData to 7 Oct 2026

Universal YOCO for Efficient Depth Scaling

arXiv 2604.01220 ↗First posted 1 Apr 2026cs.CLcited by 1

On the map: LLM reasoning → Test-time compute and efficient thinking

also in 2 more
Abstract (arXiv)

The rise of test-time scaling has remarkably boosted the reasoning and agentic proficiency of Large Language Models (LLMs). Yet, standard Transformers struggle to scale inference-time compute efficiently, as conventional looping strategies suffer from high computational overhead and a KV cache that inflates alongside model depth. We present Universal YOCO (YOCO-U), which combines the YOCO decoder-decoder architecture with recursive computation to achieve a synergistic effect greater than either alone. Built on the YOCO framework, YOCO-U implements a Universal Self-Decoder that performs multiple iterations via parameter sharing, while confining the iterative process to shallow, efficient-attention layers. This combination yields a favorable capability-efficiency tradeoff that neither YOCO nor recursion achieves independently. The YOCO architecture provides a constant global KV cache and linear pre-filling, while partial recursion enhances representational depth with limited overhead. Together, YOCO-U improves token utility and scaling behavior while maintaining efficient inference. Empirical results confirm that YOCO-U remains highly competitive in general and long-context benchmarks, demonstrating that the integration of efficient-attention architectures and recursive computation is a promising direction for scalable LLMs.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 30 of its 39 references that are in the map. First posted on arXiv 1 Apr 2026.

Builds on

30 of 39 references.

Closest ancestors · 2

Same direction, newest first

  1. 2025Encode, Think, Decode: Scaling test-time reasoning with recursive latent thoughts8 Oct
  2. 2025Parallel Scaling Law for Language Models15 May

Other references · 28

  1. 2017Attention Is All You Needcited by 59,932
  2. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  3. 2020Scaling Laws for Neural Language Modelscited by 6,411
  4. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  6. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  7. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  8. 2019Generating Long Sequences with Sparse Transformerscited by 1,523
  9. 2019Root Mean Square Layer Normalizationcited by 1,425
  10. 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
  11. 2021Are NLP Models really able to Solve Simple Math Word Problems?cited by 861
  12. 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
16 more
  1. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
  2. 2018Universal Transformerscited by 613
  3. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
  4. 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
  5. 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 390
  6. 2022Dynamic Prompt Learning via Policy Gradient for Semi-structured Mathematical Reasoningcited by 347
  7. 2021A Diverse Corpus for Evaluating and Developing English Math Word Problem Solverscited by 314
  8. 2024MathScale: Scaling Instruction Tuning for Mathematical Reasoningcited by 144
  9. 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 135
  10. 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 120
  11. 2024MARIO: MAth Reasoning with code Interpreter Output -- A Reproducible Pipelinecited by 81
  12. 2023Evaluating and Improving Tool-Augmented Computation-Intensive Math Reasoningcited by 49
  13. 2025Rope to Nope and Back Again: A New Hybrid Attention Strategycited by 46
  14. 2025Parallel Loop Transformer for Efficient Test-Time Computation Scalingcited by 23
  15. 2025Recursive Inference Scaling: A Winning Path to Scalable Inference in Language and Multimodal Systemscited by 8
  16. 2025Efficient Pretraining Length Scalingcited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026YOCO++: Enhancing YOCO with KV Residual Connections for Efficient LLM Inference
  2. 2025Accelerating Large Language Model Inference via Early-Exiting Algorithms
  3. 2026Shallow Prefill, Deep Decoding: Efficient Long-Context Inference via Layer-Asymmetric KV Visibility
  4. 2026BUDDY: BUdget-Driven DYnamic Depth Routing for Adaptive Large Language Model Inference
  5. 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depth
  6. 2026River-LLM: Large Language Model Seamless Exit Based on KV Share
  7. 2024Accelerating Large Language Model Inference with Self-Supervised Early Exits
  8. 2025Beyond Parameters: Exploring Virtual Logic Depth for Scaling Laws