Every Token Leaves a Ripple in the Stream of Thought: Eliciting Model-Internal Token Saliency for Chain-of-Thought Compression
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Chain-of-thought (CoT) reasoning improves multi-step problem solving, but long reasoning traces inflate inference cost. Token-level CoT compression reduces this cost by pruning full reasoning chains into shorter traces for model adaptation, making token selection the central challenge. Existing methods often rely on external scorers or heuristic signals only indirectly tied to the model's internal answer computation. We instead adopt a model-internal perspective: as the model forms an answer, each reasoning token induces a ripple in the residual stream whose effect on the answer reflects the token's contribution to the underlying computation. Building on this view, we propose textscMIST (Model-Internal Saliency for Token-level CoT compression), which defines token importance along two complementary axes: necessity, the drop in answer likelihood when a token's internal contribution is removed, and sufficiency, the gain in answer likelihood when that contribution alone is provided. Combining the two yields a unified importance score for pruning. Across four reasoning benchmarks and four models, textscMIST consistently outperforms baseline methods, suggesting that model-internal saliency provides an effective proxy for reasoning-token importance.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 8 of its 32 references that are in the map. First posted on arXiv 31 Aug 2026.
Builds on
8 of 32 references.
Closest ancestors · 2
Same direction, newest first
- 2026Do LLMs Encode Functional Importance of Reasoning Tokens?6 Jan
- 2025Adaptive GoGI-Skip: Coupling Goal-Gradient Importance with Dynamic Uncertainty for Efficient Reasoning13 May
Other references · 6
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2023Mistral 7Bcited by 3,164
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
- 2025Chain of Draft: Thinking Faster by Writing Lesscited by 297
- 2024How to use and interpret activation patchingcited by 214
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Not All Tokens Are What You Need In Thinking
- 2026SYNLAT: Syntax-Aligned Text-Latent Compression for Chain-of-Thought Reasoning
- 2025How Well do LLMs Compress Their Own Chain-of-Thought? A Token Complexity Approach
- 2026SCA: Segment-Wise CoT Compression with Answer Alignment
- 2026CRISP: Compressing Redundancy in Chain-of-Thought via Intrinsic Saliency Pruning
- 2026Demystifying Entropy-based Selection for Chain-of-Thought Compression in Large Reasoning Models
- 2026Chain Of Thought Compression: A Theoretical Analysis
- 2024Expediting and Elevating Large Language Model Reasoning via Hidden Chain-of-Thought Decoding