arXiv Machine Learning

When Should LLMs Reason? Adaptive Inference Routing via Entropy Dynamics

The paper introduces EDRM, an entropy‑dynamics‑based routing framework that decides whether to use direct, standard, or chain‑of‑thought decoding for large language models. By analyzing early decoding entropy, EDRM identifies tasks that benefit from explicit reasoning and routes queries accordingly, achieving 16–55% token savings while maintaining accuracy across 15 benchmarks and 7 LLMs. The study also shows that a short direct fallback branch can further improve accuracy by up to 4.9 percentage points compared to routing alone.

arXiv AI
Sep 3

SCX Router: Streaming Zero-Shot Model Selection with a Decoder-KV Classifier and a Real-World Task Ontology

The paper introduces SCX Router, a lightweight GLiClass-based model selector that assigns suitability scores to inference-time language models without autoregressive generation. It uses a 0.6B-parameter Qwen3 decoder with a shallow bidirectional scorer, preserving a text-only key–value cache across sessions and predicting task attributes such as type, difficulty, and expected output length. The authors build a comprehensive task ontology with 23 families, 115 types, and 1,173 synthetic examples, generating 150,000 verifier-scored tasks to train the router, which outperforms baseline models on LiveBench subsets with a top‑1 score of 0.707 versus 0.696 for the strongest fixed model.

By Ihor Stepanov, Aleksandr Smechov, Mykhailo Shtopko, Dmytro Vodianytskyi, Oleksandr Lukashov
arXiv Computation and Language
Sep 30

Beyond the Context Window: An Adaptive Entropy-Based Routing Framework for Hybrid Retrieval and Long-Context Language Models

arXiv:2609.35831v1 Announce Type: new Abstract: Modern large language models now support context windows of more than one million tokens, which has raised the question of whether retrieval-augmented...

By Isaac Olufadewa, Miracle Adesina, Ezekiel Oladejo, Owen Adeniyi, Fadare Fadekemi, Olamide Oso, Uthman Babatunde, Matthew Olawoyin
arXiv AI
Oct 15

Efficiently Learning To Reason or Not to Reason: Root-token Policy Optimization for Adaptive Thinking

The paper introduces Root-token Policy Optimization (RPO), a framework that lets large reasoning models decide when to engage in explicit reasoning by training only the initial root token. RPO uses group-relative rewards and advantages to focus on this branching point, dramatically cutting training overhead and VRAM usage. Experiments across various model families show that RPO achieves difficulty-aware adaptive thinking with roughly 2% of the compute required by previous adaptive-reasoning methods.

By Taehyeon Kim, Hyunsoo Lee, Youngsoo Jang, Moontae Lee
arXiv Machine Learning
Jun 5

IR3DE: A Linear Router for Large Language Models

arXiv:2606. 06098v1 Announce Type: cross Abstract: Foundational Large Language Models (LLMs) demonstrate proficiency on a wide range of general tasks, and achieve remarkable results on various specialized tasks via domain-expert LLMs.

By Eros Fan\`i, O\u{g}uzhan Ersoy
arXiv Machine Learning
Oct 23

Route-and-Generate: KL-Regularized Router Improvement for Reasoning Routing in LLMs

Route-and-Generate (Rouge) is a hierarchical training framework that lets large language models decide how much reasoning to apply to each input. It separates reasoning effort from answer generation by using a routing policy that distributes a fixed sampling budget across different reasoning formats, while the base model is trained with GRPO and per-sample rewards. The method employs a rank‑based reward favoring shorter correct solutions and updates the router through a closed‑form KL‑regularized policy improvement, leading to more concise answers on easy tasks and longer reasoning on harder ones, thereby improving accuracy‑efficiency trade‑offs on multiple reasoning benchmarks.

By Tien-Phat Nguyen, Truong Nguyen, Thanh-Thien Le, Hoang Phan, Thien Huu Nguyen, Linh Ngo Van, Trung Le