Knowing When to Stop: Efficient Context Processing via Latent Sufficiency Signals
On the map: Efficient LLM inference and serving → KV cache optimization
Abstract (arXiv)
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.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in KV cache optimization, one of 12 directions in Efficient LLM inference and serving. Builds on 30 of its 36 references that are in the map. First posted on arXiv 3 Feb 2025.
Builds on
30 of 36 references.
Closest ancestors · 3
Same direction, newest first
- 2024TokenSelect: Efficient Long-Context Inference and Length Extrapolation for LLMs via Dynamic Token-Level KV Cache Selection5 Nov
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttention12 Sep
- 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models24 Jun
Other references · 27
- 2017Attention Is All You Needcited by 59,932
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,049
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
- 2024Qwen2 Technical Reportcited by 2,076
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,270
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
- 2019BERT Rediscovers the Classical NLP Pipelinecited by 1,046
15 more
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2022ZeroQuant: Efficient and Affordable Post-Training Quantization for Large-Scale Transformerscited by 517
- 2023LongLLMLingua: Accelerating and Enhancing LLMs in Long Context Scenarios via Prompt Compressioncited by 443
- 2024MInference 1.0: Accelerating Pre-filling for Long-Context LLMs via Dynamic Sparse Attentioncited by 375
- 2022Self-critiquing models for assisting human evaluatorscited by 372
- 2024∞Bench: Extending Long Context Evaluation Beyond 100K Tokenscited by 360
- 2024LLMLingua-2: Data Distillation for Efficient and Faithful Task-Agnostic Prompt Compressioncited by 343
- 2024HELMET: How to Evaluate Long-Context Language Models Effectively and Thoroughlycited by 119
- 2018Improving Automatic Source Code Summarization via Deep Reinforcement Learningcited by 99
- 2020Language Models are Open Knowledge Graphscited by 95
- 2024LLM Internal States Reveal Hallucination Risk Faced With a Querycited by 80
- 2024500xCompressor: Generalized Prompt Compression for Large Language Modelscited by 45
- 2024Insights into LLM Long-Context Failures: When Transformers Know but Don't Tellcited by 36
- 2024In-Context Former: Lightning-fast Compressing Context for Large Language Modelcited by 12
- 2024Language Models are Symbolic Learners in Arithmeticcited by 11
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Retrieval--Reasoning Processes for Multi-hop Question Answering: A Four-Axis Design Framework and Empirical Trendscited by 8
- 2026LensVLM: Selective Context Expansion for Compressed Visual Representation of Textcited by 2
- 2026On-Demand Attention: Language Models Know When to Recallcited by 0
- 2026The Model Knows When to Stop: Training-Free Early Stopping for Long-Context Readingcited by 0
- 2026The Model Knows When to Stop: Training-Free Early Stopping for Long-Context Reading28 Sep 2026
- 2026On-Demand Attention: Language Models Know When to Recall17 Sep 2026
- 2026LensVLM: Selective Context Expansion for Compressed Visual Representation of Text7 May 2026
- 2026Retrieval--Reasoning Processes for Multi-hop Question Answering: A Four-Axis Design Framework and Empirical Trends2 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2026Language Models Can Control Their Own Attention
- 2025Training-free Context-adaptive Attention for Efficient Long Context Modeling
- 2026Inference Time Context Sparsity: Illusion or Opportunity?
- 2026Entropic Context Shaping: Information-Theoretic Filtering for Context-Aware LLM Agents
- 2026CL-bench: A Benchmark for Context Learning
- 2025AttentionInfluence: Adopting Attention Head Influence for Weak-to-Strong Pretraining Data Selection
- 2025Hidden in the Haystack: Smaller Needles are More Difficult for LLMs to Find
- 2025Adaptive Distraction: Probing LLM Contextual Robustness with Automated Tree Search