VarRate slashes LLM memory 80% without retraining
New KV cache codec keeps every token but allocates rank by query salience, training-free.
The key-value (KV) cache is the main memory bottleneck during long-context LLM inference. Existing training-free approaches fall into two flawed families: token-selection methods (SnapKV, Ada-KV) irreversibly evict low-scoring tokens, causing accuracy collapses of 11-15 points when importance signals degrade; uniform low-rank coding wastes budget by spending equal rank on every token. VarRate solves both by allocating rank rather than evicting tokens. It is a training-free KV codec that assigns each token a variable low-rank budget proportional to its query salience, keeping every token at a nonzero rank.
Tested on the 16-task LongBench at a 20% compression budget, VarRate degraded by only 0.8 points from the uncompressed model on both Llama-3.1-8B and Qwen2.5-7B. It significantly outperformed its uniform-rank ablation and was accuracy-equivalent to KVzip (a query-agnostic reuse method) in three of four settings, while requiring roughly one-eighth the prefill overhead. This makes VarRate the strongest matched-memory compressor among training-free methods, enabling longer context processing without costly retraining.
- VarRate is a training-free variable-rate KV cache codec that allocates rank per token based on query salience.
- At 20% memory budget on LongBench, it stays within 0.8 points of uncompressed accuracy on Llama-3.1-8B and Qwen2.5-7B.
- Compared to token-eviction methods (11-15 point drops), VarRate degrades only 3.5-5.5 points, matching KVzip at 1/8th prefill cost.
Why It Matters
Enables massive memory savings for long-context LLMs on existing hardware without expensive retraining or accuracy loss.