mlsanityData to 14 Oct 2026

CASK: Core-Aware Selective KV Compression for Reasoning Traces

arXiv 2604.10900 ↗First posted 13 Apr 2026cs.AI cs.LGcited by 0

On the map: LLM reasoning → Test-time compute and efficient thinking

also in 1 more
Abstract (arXiv)

In large language models performing long-form reasoning, the KV cache grows rapidly with decode length, creating bottlenecks in memory and inference stability. Existing reasoning-oriented KV compression has mostly followed an eviction-centered view: estimate token importance more accurately, then discard lower-ranked entries. Our analysis suggests that scorer refinement alone often fails to substantially reorganize the actual keep-set and may therefore not be the main lever for preserving reasoning behavior. We instead frame reasoning KV compression as a behavior-preserving structured consolidation problem. CASK partitions the decode-time reasoning trace into a protected core that anchors answer formation and intermediate state, and mergeable scratch with high redundancy. The core is preserved, while selective consolidation is applied only to the scratch. To address prompt-heavy regimes where the prefix can exhaust the budget before decode-stage compression becomes active, CASK further uses a two-stage design: prefix eviction followed by decode-stage consolidation. On the H100 reasoning gate, CASK shows higher full-KV continuation fidelity than TriAttention at matched budgets on both AIME24 and AIME25, with recurring cask@384 > triattention@512 crossings. In prompt-heavy replay, multi_news and vcsum act as decode-active witnesses, while qmsum and gov_report expose the prefix_budget_exhausted boundary. The overall evidence supports a simple conclusion: effective reasoning KV compression depends less on more elaborate scorer engineering than on combining core preservation with selective scratch consolidation to lower the usable budget frontier.

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 6 of its 9 references that are in the map. First posted on arXiv 13 Apr 2026.

Builds on

6 of 9 references.

Closest ancestors · 3

Same direction, newest first

  1. 2026TriAttention: Efficient Long Reasoning with Trigonometric KV Compression6 Apr
  2. 2026Crystal-KV: Efficient KV Cache Management for Chain-of-Thought LLMs via Answer-First Principle5 Jan
  3. 2025R-KV: Redundancy-aware KV Cache Compression for Reasoning Models30 May

Other references · 3

  1. 2024Model Tells You Where to Merge: Adaptive KV Cache Merging for LLMs on Long-Context Taskscited by 96
  2. 2024D2O: Dynamic Discriminative Operations for Efficient Long-Context Inference of Large Language Modelscited by 31
  3. 2025KeepKV: Achieving Periodic Lossless KV Cache Compression for Efficient LLM Inferencecited by 9

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026KARA: Efficient Reasoning LLM Serving via Sliding-Window KV Cache Compression
  2. 2025SkipKV: Selective Skipping of KV Generation and Storage for Efficient Inference with Large Reasoning Models
  3. 2025Hold Onto That Thought: Assessing KV Cache Compression On Reasoning
  4. 2025Semantic Integrity Matters: Benchmarking and Preserving High-Density Reasoning in KV Cache Compression
  5. 2026KV-Rescue: Recovering Reasoning Language Model KV Eviction Loss via Stepwise Interleaving
  6. 2025Breadcrumbs Reasoning: Memory-Efficient Reasoning with Compression Beacons
  7. 2026BeaconKV: Key-Value Cache Compression Guided by Beacon Queries for Efficient Large Reasoning Model Inference
  8. 2026Random Attention: Rethinking KV Cache Eviction for Efficient Reasoning