mlsanityData to 14 Oct 2026

R2R: Efficiently Navigating Divergent Reasoning Paths with Small-Large Model Token Routing

arXiv 2505.21600 ↗First posted 27 May 2025cs.CL cs.AI cs.LGcited by 29

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

Abstract (arXiv)

Large Language Models (LLMs) achieve impressive reasoning capabilities at the cost of substantial inference overhead, posing substantial deployment challenges. Although distilled Small Language Models (SLMs) significantly enhance efficiency, their performance suffers as they fail to follow LLMs' reasoning paths. Luckily, we reveal that only a small fraction of tokens genuinely diverge reasoning paths between LLMs and SLMs. Most generated tokens are either identical or exhibit neutral differences, such as minor variations in abbreviations or expressions. Leveraging this insight, we introduce **Roads to Rome (R2R)**, a neural token routing method that selectively utilizes LLMs only for these critical, path-divergent tokens, while leaving the majority of token generation to the SLM. We also develop an automatic data generation pipeline that identifies divergent tokens and generates token-level routing labels to train the lightweight router. We apply R2R to combine R1-1.5B and R1-32B models from the DeepSeek family, and evaluate on challenging math, coding, and QA benchmarks. With an average activated parameter size of 5.6B, R2R surpasses the average accuracy of R1-7B by 1.6x, outperforming even the R1-14B model. Compared to R1-32B, it delivers a 2.8x wall-clock speedup with comparable performance, advancing the Pareto frontier of test-time scaling efficiency. Our code is available at https://github.com/thu-nics/R2R.

Key facts. Cited by 29 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 37 of its 62 references that are in the map. First posted on arXiv 27 May 2025.

Builds on

37 of 62 references.

Closest ancestors · 10

Same direction, newest first

  1. 2025PM-KVQ: Progressive Mixed-precision KV Cache Quantization for Long-CoT LLMs24 May
  2. 2025SplitReason: Learning To Offload Reasoning23 Apr
  3. 2025Speculative Thinking: Enhancing Small-Model Reasoning with Large Model Guidance at Inference Time12 Apr
  4. 2025SpecReason: Fast and Accurate Inference-Time Compute via Speculative Reasoning10 Apr
  5. 2025A Survey of Efficient Reasoning for Large Reasoning Models: Language, Multimodality, and Beyond27 Mar
  6. 2025Stop Overthinking: A Survey on Efficient Reasoning for Large Language Models20 Mar
  7. 2025Reward-Guided Speculative Decoding for Efficient LLM Reasoning31 Jan
  8. 2025Thoughts Are All Over the Place: On the Underthinking of o1-Like LLMs30 Jan
  9. 2024Token-Budget-Aware LLM Reasoning24 Dec
  10. 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertainty26 Jan

Other references · 27

  1. 2023GPT-4 Technical Reportcited by 18,396
  2. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  3. 2024Qwen2.5 Technical Reportcited by 4,942
  4. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
  5. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  6. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  7. 2024Mixtral of Expertscited by 1,823
  8. 2024OpenAI o1 System Cardcited by 1,244
  9. 2023Gemini: A Family of Highly Capable Multimodal Modelscited by 872
  10. 2024DeepSeek-V3 Technical Reportcited by 811
  11. 2024From Crowdsourced Data to High-Quality Benchmarks: Arena-Hard and BenchBuilder Pipelinecited by 500
  12. 2025Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attentioncited by 407
15 more
  1. 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 370
  2. 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 331
  3. 2025Chain of Draft: Thinking Faster by Writing Lesscited by 297
  4. 2024DeepSeek-Prover: Advancing Theorem Proving in LLMs through Large-Scale Synthetic Datacited by 234
  5. 2024A Survey on Efficient Inference for Large Language Modelscited by 205
  6. 2025MoBA: Mixture of Block Attention for Long-Context LLMscited by 198
  7. 2024DeepSeek-Prover-V1.5: Harnessing Proof Assistant Feedback for Reinforcement Learning and Monte-Carlo Tree Searchcited by 197
  8. 2024Are We Done with MMLU?cited by 194
  9. 2023Routing to the Expert: Efficient Reward-guided Ensemble of Large Language Modelscited by 145
  10. 2024Learning Harmonized Representations for Speculative Samplingcited by 72
  11. 2025SageAttention3: Microscaling FP4 Attention for Inference and An Exploration of 8-Bit Trainingcited by 66
  12. 2025MixLLM: Dynamic Routing in Mixed Large Language Modelscited by 59
  13. 2024Mixture of Attention Spans: Optimizing LLM Inference Efficiency with Heterogeneous Sliding-Window Lengthscited by 44
  14. 2025CITER: Collaborative Inference for Efficient Large Language Model Decoding with Token-Level Routingcited by 30
  15. 2025Estimating LLM Uncertainty with Evidencecited by 26

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 7; 2026: 22.

29 papers.

Sort:

By area:

  1. 2025Cache-to-Cache: Direct Semantic Communication Between Large Language Modelscited by 61
  2. 2025Don't Overthink It: A Survey of Efficient R1-style Large Reasoning Modelscited by 37
  3. 2026Dynamic Model Routing and Cascading for Efficient LLM Inference: A Surveycited by 19
  4. 2025Towards Concise and Adaptive Thinking in Large Reasoning Models: A Surveycited by 18
  5. 2025Semantic Energy: Detecting LLM Hallucination Beyond Entropycited by 18
  6. 2025PAROAttention: Pattern-Aware ReOrdering for Efficient Sparse and Quantized Attention in Visual Generation Modelscited by 17
  7. 2026GlimpRouter: Efficient Collaborative Inference by Glimpsing One Token of Thoughtscited by 17
  8. 2025Think-at-Hard: Dynamic Looped Transformers for Improved Reasoningcited by 16
  9. 2026Token-Level LLM Collaboration via FusionRoutecited by 9
  10. 2026RelayLLM: Efficient Reasoning via Collaborative Decodingcited by 8
  11. 2025A Survey on Collaborating Small and Large Language Models for Performance, Cost-effectiveness, Cloud-edge Privacy, and Trustworthinesscited by 7
  12. 2026FROST: Filtering Reasoning Outliers with Attention for Efficient Reasoningcited by 5
17 more
  1. 2026Switch Attention: Towards Dynamic and Fine-grained Hybrid Transformerscited by 4
  2. 2026Select to Think: Unlocking SLM Potential with Local Sufficiencycited by 3
  3. 2026MentorCollab: Large-to-Small Inference-Time Mentorship for Concise Reasoning in Language Modelscited by 2
  4. 2026Rethinking Model Efficiency: Multi-Agent Inference with Large Modelscited by 2
  5. 2026Dissecting Failure Dynamics in Large Language Model Reasoningcited by 2
  6. 2026Learn What's Left, Not What's Mastered: Saturation Aware Advantage Reweighting for Multi-Reward Policy Optimizationcited by 2
  7. 2026TriSpec: Ternary Speculative Decoding via Lightweight Proxy Verificationcited by 1
  8. 2026RelayGen: Intra-Generation Model Switching for Efficient Reasoningcited by 1
  9. 2026RankGuide: Tensor-Rank-Guided Routing and Steering for Efficient Reasoningcited by 1
  10. 2026UniScale: Adaptive Unified Inference Scaling via Online Joint Optimization of Model Routing and Test-Time Scalingcited by 1
  11. 2026FutureBridge: Token Selection Beyond Local Preference in Collaborative Decodingcited by 1
  12. 2026Improving Test-Time Scaling with Adaptive Looped Transformerscited by 1
  13. 2026Calibrated Speculative Decoding: Frequency-Guided Candidate Selection for Efficient Inferencecited by 0
  14. 2026Compute Where it Counts: Self Optimizing Language Modelscited by 0
  15. 2026Before Thinking, Learn to Decide: Proactive Routing for Efficient Visual Reasoningcited by 0
  16. 2026What Does a Token Cost? A Mixture-of-Agents Measurement of Sufficient Per-Token Computecited by 0
  17. 2026TokenRouter: Efficient Serving System for Token-Level LLM Routingcited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2025Token Assorted: Mixing Latent and Text Tokens for Improved Language Model Reasoning
  2. 2025Route-and-Reason: Scaling Large Language Model Reasoning with Reinforced Model Router
  3. 2025Route to Reason: Adaptive Routing for LLM and Reasoning Strategy Selection
  4. 2026Think Deep, Not Just Long: Measuring LLM Reasoning Effort via Deep-Thinking Tokens
  5. 2025On the Bias of Next-Token Predictors Toward Systematically Inefficient Reasoning: A Shortest-Path Case Study
  6. 2025Confidence-Guided Stepwise Model Routing for Cost-Efficient Reasoning
  7. 2025MobileLLM-R1: Exploring the Limits of Sub-Billion Language Model Reasoners with Open Training Recipes
  8. 2026The Path of Least Resistance: Guiding LLM Reasoning Trajectories with Prefix Consensus