Update README.md

README.md

@@ -15,34 +15,15 @@ tags:
 
 # TrimKV: Token Retention for Memory-Bounded Key-Value Eviction
 
-This repository contains the weights for **TRIM-KV-Qwen3-1.7B-Math**, presented in the paper [Make Each Token Count: Towards Improving Long-Context Performance with KV Cache Eviction](https://huggingface.co/papers/2605.09649).
+This repository contains the weights for **TRIM-KV-Qwen3-1.7B-Math**, as presented in the paper [Cache What Lasts: Token Retention for Memory-Bounded KV Cache in LLMs](https://huggingface.co/papers/2512.03324).
 
-The official implementation and training code can be found at [https://github.com/ngocbh/trimkv](https://github.com/ngocbh/trimkv).
+The core idea behind TRIM-KV is to learn the intrinsic importance of each key–value pair at creation time, which we call *token retention*, and then decay this importance exponentially over time to mimic standard inference running with eviction. 
 
-> TRIM-KV is an efficient and learnable key–value eviction strategy designed to improve the efficiency of large language models (LLMs) in long-horizon inference.
+The retention score is query-agnostic and captures the long-term utility of tokens. This is different from attention scores, which are query-dependent: they capture the short-term utility for predicting the next token and are recomputed at every step.
 
-The core idea behind TRIM-KV is to learn the intrinsic importance of each key–value pair at creation time, which we call *token retention*, and then decay this importance exponentially over time to mimic the standard inference running with eviction. 
-
-The retention score is query-agnostic and captures the long-term utility of tokens. This is different from attention scores, which are query-dependent: they capture the short-term utility for predicting the next token and are recomputed at every step, making them local, myopic, and highly dependent on the transient decoding state.
-
-### Why TRIM-KV?
-
-It's fast
-<div align="center">
-    <img width="1000" alt="teaser" src="https://github.com/ngocbh/trimkv/blob/main/assets/speed.png?raw=true"/>
-</div>
-
-It's smart
-<div align="center">
-    <img width="1000" alt="teaser" src="https://github.com/ngocbh/trimkv/blob/main/assets/performance.png?raw=true"/>
-</div>
-
-And it's interpretable
-<div align="center">
-    <img width="1000" alt="teaser" src="https://github.com/ngocbh/trimkv/blob/main/assets/eviction.png?raw=true"/>
-</div>
-
----
+- **Paper:** [Cache What Lasts: Token Retention for Memory-Bounded KV Cache in LLMs](https://huggingface.co/papers/2605.09649)
+- **Code:** [GitHub - ngocbh/trimkv](https://github.com/ngocbh/trimkv)
+- **Arxiv:** [2512.03324](https://arxiv.org/abs/2512.03324)
 
 ## Quick Start
 
@@ -94,6 +75,12 @@ past_key_values = PagedTrimKVCache(
 ## Citation
 
 ```bibtex
+@article{bui2025cache,
+  title={Cache what lasts: Token retention for memory-bounded kv cache in llms},
+  author={Bui, Ngoc and Sharma, Shubham and Lamba, Simran and Mishra, Saumitra and Ying, Rex},
+  journal={arXiv preprint arXiv:2512.03324},
+  year={2025}
+}
 @article{bui2025make,
   title={Make Each Token Count: Towards Improving Long-Context Performance with KV Cache Eviction},
   author={Bui, Ngoc and Nguyen, Hieu Trung and Cohan, Arman and Ying, Rex},