cs.AIOct 21, 2026

Same Beliefs, Different Policies: Learning Reusable Representations for Early Stopping in LLM Reasoning

Authors: Jingwei Ni, Dmitry Abulkhanov, Vladislav Smirnov, Markus Leippold, Elliott Ash, Timothy Baldwin, Artem Shelmanov, Mrinmaya Sachan

Organizations: ETH Zürich · MBZUAI · University of Zürich · The University of Melbourne

Abstract

Reasoning LLMs often continue reasoning after reaching a correct answer, wasting computation and sometimes replacing it with an incorrect one. Deciding when to stop is a metareasoning problem: the model must weigh what further reasoning could achieve against its cost, given what it can already answer. Direct utility training learns a stopping representation at a chosen token price; supporting new prices can require additional controller training. We introduce BELT (BELief-based Termination), which separates outcome prediction from stopping control. It learns beliefs about current and future answers from a frozen LLM's hidden states, then fits a linear stopping policy on the frozen features. The prediction targets are independent of token price, so one probe supports different accuracy-cost trade-offs through CPU policy refitting or threshold adjustment. Averaged over five out-of-distribution (OOD) math benchmarks, BELT reduces generated tokens by 41% on Qwen3-1.7B and 51% on OLMo-3-7B-Think without loss of accuracy. The same math-trained BELT transfers to coding and agentic tasks without training on code or agentic data. In all three domains, its accuracy-cost trade-offs are competitive with or better than those of end-to-end controllers. On OLMo3, its probe has 21.2x fewer parameters than one checkpoint of the strongest baseline, OS-Pruner, and needs 55x fewer training GPU-hours than OS-Pruner's five price-specific checkpoints.

Figures & tables

Appendix figures & tables78 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Settle: Learning When to Stop Reasoning

    Sep 30, 2026Ryan Brown, Zihao Fu, Chris RussellEarly Stopping for Language Model ReasoningEfficient Language Model Reasoning

  2. When Does Learning to Stop Help? A Cost-Aware Study of Early Exits in Reasoning Models

    Jun 29, 2026Zhe Dong, Fang Qin, Manish ShahEfficient Language Model ReasoningCost-Aware Inference

  3. MetaCtrl: Your Large Language Models Can Reason Better and More Concisely with a Metacognitive Controller

    Sep 29, 2026Zhibin Wen, Tao Han, Lei Bai +2Adaptive Reasoning in Language ModelsRL for Language Model Reasoning