Knowing When to Stop: Efficient Context Processing via Latent Sufficiency Signals

Roy Xie, Junlin Wang, Paul Rosu, Chunyuan Deng, Bolun Sun +2 more
2/3/2025
cs.CL

No official code found. Best implementation: ruoyuxie/when-to-stop (7 stars, Apache-2.0, last commit 2026-02-04).

Abstract

Large language models (LLMs) process entire input contexts indiscriminately, which is inefficient when the information required to answer a query is localized within the context. We present dynamic context cutoff, a novel method enabling LLMs to self-terminate processing upon acquiring sufficient task-relevant information. Through analysis of model internals, we discover that specific attention heads inherently encode "sufficiency signals" -- detectable through lightweight classifiers -- that predict when critical information has been processed. This reveals a new efficiency paradigm: models' internal understanding naturally dictates processing needs rather than external compression heuristics. Comprehensive experiments across six QA datasets (up to 40K tokens) with three model families (LLaMA/Qwen/Mistral, 1B-70B) demonstrate 3.4% accuracy improvement while achieving 1.33x token reduction on average. Furthermore, our method demonstrates superior performance compared to other context efficiency methods at equivalent token reduction rates. Additionally, we observe an emergent scaling phenomenon: while smaller models require probing for sufficiency detection, larger models exhibit intrinsic self-assessment capabilities through prompting.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

[NeurIPS 25] Knowing When to Stop: Efficient Context Processing via Latent Sufficiency Signals

70PythonFeb 2, 20258 months agoApache-2.0

Similar Papers

Cite this paper

@article{xie2025knowing,
  title  = {Knowing When to Stop: Efficient Context Processing via Latent Sufficiency Signals},
  author = {Roy Xie and Junlin Wang and Paul Rosu and Chunyuan Deng and Bolun Sun and Zihao Lin and Bhuwan Dhingra},
  year   = {2025},
  eprint = {2502.01025},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2502.01025v3},
  journal = {NEURIPS 2025 2025}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2502.01025.svg)](https://codeofpaper.com/paper/2502.01025)

Discussion