Universal YOCO for Efficient Depth Scaling
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
- 2025Encode, Think, Decode: Scaling test-time reasoning with recursive latent thoughts8 Oct
- 2025Parallel Scaling Law for Language Models15 May
Other references · 28
- 2017Attention Is All You Needcited by 59,932
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2021Are NLP Models really able to Solve Simple Math Word Problems?cited by 861
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
16 more
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
- 2018Universal Transformerscited by 613
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 594
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
- 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 390
- 2022Dynamic Prompt Learning via Policy Gradient for Semi-structured Mathematical Reasoningcited by 347
- 2021A Diverse Corpus for Evaluating and Developing English Math Word Problem Solverscited by 314
- 2024MathScale: Scaling Instruction Tuning for Mathematical Reasoningcited by 144
- 2024You Only Cache Once: Decoder-Decoder Architectures for Language Modelscited by 135
- 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 120
- 2024MARIO: MAth Reasoning with code Interpreter Output -- A Reproducible Pipelinecited by 81
- 2023Evaluating and Improving Tool-Augmented Computation-Intensive Math Reasoningcited by 49
- 2025Rope to Nope and Back Again: A New Hybrid Attention Strategycited by 46
- 2025Parallel Loop Transformer for Efficient Test-Time Computation Scalingcited by 23
- 2025Recursive Inference Scaling: A Winning Path to Scalable Inference in Language and Multimodal Systemscited by 8
- 2025Efficient Pretraining Length Scalingcited by 5
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026What Makes Recurrence Effective in Looped Language Models?29 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026YOCO++: Enhancing YOCO with KV Residual Connections for Efficient LLM Inference
- 2025Accelerating Large Language Model Inference via Early-Exiting Algorithms
- 2026Shallow Prefill, Deep Decoding: Efficient Long-Context Inference via Layer-Asymmetric KV Visibility
- 2026BUDDY: BUdget-Driven DYnamic Depth Routing for Adaptive Large Language Model Inference
- 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depth
- 2026River-LLM: Large Language Model Seamless Exit Based on KV Share
- 2024Accelerating Large Language Model Inference with Self-Supervised Early Exits
- 2025Beyond Parameters: Exploring Virtual Logic Depth for Scaling Laws