PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
October 5, 20250 citationsOpen Access

ThinKV: Thought-Adaptive KV Cache Compression for Efficient Reasoning Models

View Full Paper
ARAkshat RamachandranMNMarina NeseemCSCharbel Sakr

Key Points

  • ThinKV achieves near-lossless accuracy with under 5% of the original kv cache, maintaining essential data.
  • The hybrid quantization-eviction strategy allows for token precision adjustments according to thought importance.
  • Extensive tests on multiple benchmarks highlight significant gains in efficiency with up to 5.8x performance improvement.
  • Such innovations in kv cache management may redefine resource utilization in large-scale reasoning models.

Abstract

The long-output context generation of large reasoning models enables extended chain of thought (CoT) but also drives rapid growth of the key-value (KV) cache, quickly overwhelming GPU memory. To address this challenge, we propose ThinKV, a thought-adaptive KV cache compression framework. ThinKV is based on the observation that attention sparsity reveals distinct thought types with varying importance within the CoT. It applies a hybrid quantization-eviction strategy, assigning token precision by thought importance and progressively evicting tokens from less critical thoughts as reasoning trajectories evolve. Furthermore, to implement ThinKV, we design a kernel that extends PagedAttention to enable efficient reuse of evicted tokens' memory slots, eliminating compaction overheads. Extensive experiments on DeepSeek-R1-Distill, GPT-OSS, and NVIDIA AceReason across mathematics and coding benchmarks show that ThinKV achieves near-lossless accuracy with less than 5% of the original KV cache, while improving performance with up to 5.8x higher inference throughput over state-of-the-art baselines.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Ramachandran et al. (2025) studied this question.

synapsesocial.com/papers/68e25382d6d66a53c2474955https://doi.org/10.48550/arxiv.2510.01290
Ask AI
Helpful
Bookmark
Share
View Full Paper