arXiv AI By Egor Maximov, Aleksei Merkelov, Vasiliy Alekseev, Ivan Krivorotov, Valentin Beliaev, Nikita Arsenin, Dmitry Bocharov, Aleksei Goncharov

DCKV: Learned Transform Coding of the Prompt KV Cache in Hybrid Language Models

Read the original on arXiv AI →

The paper introduces DCKV, a differentiable codec designed to compress the key-value (KV) cache used in hybrid language models. DCKV jointly optimizes its transform, quantizer, and entropy models—conditioned on prompt tokens and neighboring positions—directly against a frozen language model, balancing compression rate with minimal impact on output quality. Experiments on Qwen3.5-2B demonstrate that DCKV can reduce the storage requirement for a compressed token by over 55× compared to bf16, with a loss of no more than 1.8 points on long‑context benchmarks, and by over 100× with a loss of no more than 3.6 points.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv Computation and Language
2d ago

EntropyCache: Decoded Token Entropy Guided KV Caching for Diffusion Language Models

EntropyCache is a training‑free KV caching technique for diffusion‑based large language models that uses the maximum entropy of newly decoded token distributions to decide when to recompute cached states. By leveraging two empirical observations—entropy correlating with cache drift and feature volatility persisting across multiple steps—EntropyCache makes a skip‑or‑recompute decision with only O(V) computation per step, independent of context length or model size. Experiments on LLaDA‑8B‑Instruct and Dream‑7B‑Instruct demonstrate speedups of 15.2×–26.4× on standard benchmarks and 22.4×–24.1× on chain‑of‑thought benchmarks while maintaining competitive accuracy and keeping decision overhead below 0.5% of inference time.

By Minsoo Cheong, Donghyun Son, Woosang Lim, Sungjoo Yoo
arXiv Machine Learning
Oct 16

AbridgeKV: Compressing Long Reasoning into the KV Cache of Learned Pause Tokens

AbridgeKV compresses long chain‑of‑thought outputs by encoding completed text blocks into the KV cache of specially trained pause tokens, freeing the original cache space. The method uses pause tokens positioned within the block, allocates cache budget per layer based on entry value, and compresses large blocks. Experiments on 446 competition problems show that with only a quarter of the original KV cache, AbridgeKV matches the accuracy and output length of a full‑KV model, while enabling faster inference and a 3.5× smaller cache per sequence.

By Jack Cai, Lijie Yang, Vasilis Kontonis, Dimitris Papailiopoulos, Shivam Garg