arXiv Computation and Language

Adaptive Stopping of Streamed Reasoning for Language Model Agents

The paper introduces StreamStop, a lightweight stopping policy for reasoning-enabled language model agents that learns offline from prefix-level action agreement. By encoding available reasoning text with a frozen prefix reader and a stop head, StreamStop can decide when to stop reasoning without needing internal signals from the model. Experiments on WebShop and ScienceWorld show that StreamStop achieves higher success rates than many fixed-budget baselines while using fewer reasoning tokens, notably reducing token usage by 42% on WebShop with Qwen3.8-27B.

arXiv Computation and Language
Aug 28

TRACES: Tagging Reasoning Steps for Adaptive Cost-Efficient Early-Stopping

TRACES (Tagging Reasoning Steps for Adaptive Cost‑Efficient Early‑Stopping) is a lightweight framework that tags reasoning steps of large‑language models in real time, enabling adaptive, cost‑efficient early stopping during inference. By monitoring the types of steps generated, the method identifies when models shift their reasoning after arriving at a correct answer, allowing for interpretable stopping criteria. Experiments on mathematical reasoning benchmarks (MATH500, GSM8K, AIME) and knowledge benchmarks (MMLU, GPQA) show token reductions of 20–50% while preserving accuracy, with more conservative thresholds needed for harder tasks such as BeyondAIME and IMO AnswerBench.

By Yannis Belkhiter, Seshu Tirupathi, Giulio Zizzo, John D. Kelleher
arXiv AI
Oct 9

When Should Agents Think? Adaptive Reasoning via Cross-Turn Estimation

Large language model agents often reason before every action, but this may be unnecessary if earlier reasoning still supports later actions. The paper introduces RACE, a method that uses likelihood-based signals to detect when reasoning can be omitted, called LoGiC. Experiments on four benchmarks show RACE cuts reasoning cost while keeping or improving performance.

By Yiruo Cheng, Shen Huang, Xiaoshuai Song, Jiejun Tan, Guanting Dong, Pengjun Xie, Ji-Rong Wen, Zhicheng Dou
arXiv Machine Learning
Sep 22

StepKV: Step-Aware KV Cache Compression for LLM Agents

StepKV introduces a step-aware approach to compressing the key-value cache used during large language model inference, treating reasoning steps as primary units of retention rather than individual tokens. By linking cache entries to the steps that generated them and estimating each step’s utility from trajectory signals, StepKV assigns a combined token‑ and step‑level score to guide pruning. Experiments on multi‑hop question answering and long‑horizon web reasoning show that StepKV maintains accuracy even under tight cache budgets, outperforming token‑level baselines that suffer sharp performance drops.

By Boyu Feng, Jiahong Liu, Yifan Li, Wenhao Yu, Zexuan Qiu, Yuliang Sun, Ming Shen, Xiang Li, Quanyu Dai, Irwin King
arXiv AI
Sep 28

Learning to Stop without Learning to Stop: Self-Supervised Confidence Training Improves Reasoning Efficiency

The paper demonstrates that fine‑tuning reasoning models to predict their own confidence at intermediate steps—using only 600 self‑supervised examples—substantially improves inference efficiency. Without adding any explicit stopping or length penalties, the models generate up to 25 % fewer tokens while maintaining accuracy on mathematical, scientific, and coding benchmarks across several architectures. The study finds that confidence supervision preserves the models’ high‑level reasoning structure rather than merely suppressing specific behaviors.

By Parsa Hosseini, Akasha Tigalappanavara, Sumit Nawathe, Chenrui Fan, Sourya Basu, Genta Indra Winata, Anirban Das, Soheil Feizi, Nima Chitsazan