Sign InOpen Brain
arXivPaperNeeds Review

Learning When to Think: Adaptive Reasoning for Test-Time Compute Allocation

A model learned to choose its own reasoning budget, cutting average output tokens 41% with a small MATH500 accuracy drop. Agent builders could replace fixed effort settings with task-level routing.

arXiv
Open Source Open MarkdownOpen JSON
Source Summary

A **1.5B distilled model** learned to begin each response in NoThink, Short, or Long mode without a separate router. On MATH500, mean output fell from **4,796 to 2,811 tokens** while accuracy moved from 0.796 to 0.782.

Practical Implication

Builders running coding agents should reconsider one fixed reasoning budget for every task. A learned or explicit difficulty gate could reserve longer runs for harder work and shorten routine calls, reducing latency and token use.

Agent-Ready Context
A **1.5B distilled model** learned to begin each response in NoThink, Short, or Long mode without a separate router. On MATH500, mean output fell from **4,796 to 2,811 tokens** while accuracy moved from 0.796 to 0.782.

Builders running coding agents should reconsider one fixed reasoning budget for every task. A learned or explicit difficulty gate could reserve longer runs for harder work and shorten routine calls, reducing latency and token use.

The model was trained on MATH, not coding workloads, and the result averages only **three seeds**. Transfer was strongest on easier benchmarks, including a **76% token reduction on GSM8K**, so coding-agent gains still need direct evaluation.
Context Map
modelresearch#reasoning
Uncertainty
The model was trained on MATH, not coding workloads, and the result averages only **three seeds**. Transfer was strongest on easier benchmarks, including a **76% token reduction on GSM8K**, so coding-agent gains still need direct evaluation.