Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

R-KV: Redundancy-aware KV Cache Compression for Reasoning Models

About

Reasoning models have demonstrated impressive performance in self-reflection and chain-of-thought reasoning. However, they often produce excessively long outputs, leading to prohibitively large key-value (KV) caches during inference. While chain-of-thought inference significantly improves performance on complex reasoning tasks, it can also lead to reasoning failures when deployed with existing KV cache compression approaches. To address this, we propose Redundancy-aware KV Cache Compression for Reasoning models (R-KV), a novel method specifically targeting redundant tokens in reasoning models. Our method preserves nearly 100% of the full KV cache performance using only 10% of the KV cache, substantially outperforming existing KV cache baselines, which reach only 60% of the performance. Remarkably, R-KV even achieves 105% of full KV cache performance with 16% of the KV cache. This KV-cache reduction also leads to a 90% memory saving and a 6.6X throughput over standard chain-of-thought reasoning inference. Experimental results show that R-KV consistently outperforms existing KV cache compression baselines across two mathematical reasoning datasets.

Zefan Cai, Wen Xiao, Hanshi Sun, Cheng Luo, Yikai Zhang, Ke Wan, Yucheng Li, Yeyang Zhou, Li-Wen Chang, Jiuxiang Gu, Zhen Dong, Anima Anandkumar, Abedelkadir Asi, Junjie Hu• 2025

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K
Accuracy96.36
1362
Mathematical ReasoningMATH
Accuracy91.6
882
Mathematical ReasoningMATH500 (test)--
514
Commonsense ReasoningCSQA
Accuracy77
366
Mathematical ReasoningGSM8K
Accuracy (GSM8K)50
358
Question AnsweringOBQA
Accuracy84
300
Mathematical ReasoningAMC
Accuracy (ACC)85
203
Mathematical ReasoningAIME 2024 (test)
Accuracy43.3
159
Mathematical ReasoningOlympiad
Accuracy55.56
137
Mathematical ReasoningMATH500
Accuracy (ACC)20
133
Showing 10 of 31 rows

Other info

Follow for update