mlsanityData to 7 Oct 2026

Thin Keys, Full Values: Reducing KV Cache via Low-Dimensional Attention Selection

arXiv 2603.04427 ↗First posted 16 Feb 2026cs.LG cs.AIcited by 2

On the map: Efficient LLM inference and serving → KV cache optimization

Abstract (arXiv)

Standard Transformer attention uses identical dimensionality for queries, keys, and values, yet these components serve different roles: queries and keys produce scalar attention weights (selection), while values carry rich representations (value transfer). We show that selection requires only O(log N) dimensions to distinguish among N relevant token categories (e.g., syntactic roles, semantic clusters, positional patterns) -- far fewer than value transfer needs. We introduce factored keys, which exploit this asymmetry to physically shrink the KV cache of any pretrained model without retraining from scratch -- unlike Grouped-Query Attention (GQA) and Multi-Head Latent Attention (MLA), which must be designed into the architecture before pretraining. We factorize each key projection WK ≈ Ad × r Br × d via truncated singular value decomposition (SVD) (where r is the chosen compression dimension), set WK' = A as the new key projection producing compact r-dimensional keys for the cache, and absorb B^top into the query projection (WQ' = WQ B^top) at zero cost -- since queries are never cached. At the 7B scale, training from scratch with r = d/4 (where d is the model dimension) matches full-attention perplexity (9.24 vs 9.25 PPL after 20B tokens, mean over two seeds) while using 12% fewer parameters and training 8% faster. For existing models, SVD followed by QK fine-tuning (3 epochs, less than 1% of pretraining data) achieves 75% key cache savings at roughly 2% quality cost on both GPT-2 and Mistral-7B. The approach composes with GQA and quantization for up to 16× combined key cache compression. For a 7B model serving a 128K context, factored keys save 25 GB of KV cache per user, enabling roughly 60% more concurrent users on identical hardware.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in KV cache optimization, one of 12 directions in Efficient LLM inference and serving. Builds on 22 of its 29 references that are in the map. First posted on arXiv 16 Feb 2026.

Builds on

22 of 29 references.

Closest ancestors · 1

Same direction, newest first

  1. 2025Compression Barriers for Autoregressive Transformers21 Feb

Other references · 21

  1. 2017Attention Is All You Needcited by 59,932
  2. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  3. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  4. 2020Longformer: The Long-Document Transformercited by 3,362
  5. 2023Mistral 7Bcited by 3,065
  6. 2016Pointer Sentinel Mixture Modelscited by 2,992
  7. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  8. 2020Reformer: The Efficient Transformercited by 1,658
  9. 2019Generating Long Sequences with Sparse Transformerscited by 1,523
  10. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
  11. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
  12. 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
9 more
  1. 2024KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cachecited by 604
  2. 2024KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationcited by 571
  3. 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
  4. 2024SVD-LLM: Truncation-aware Singular Value Decomposition for Large Language Model Compressioncited by 231
  5. 2024Loki: Low-rank Keys for Efficient Sparse Attentioncited by 87
  6. 2024AsymKV: Enabling 1-Bit Quantization of KV Cache with Layer-Wise Asymmetric Quantization Configurationscited by 19
  7. 2025Inference-time sparse attention with asymmetric indexingcited by 11
  8. 2025Cost-Optimal Grouped-Query Attention for Long-Context Modelingcited by 4
  9. 2025Efficient Low Rank Attention for Long-Context Inference in Large Language Modelscited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025KQ-SVD: Compressing the KV Cache with Provable Guarantees on Attention Fidelity
  2. 2024Reducing Transformer Key-Value Cache Size with Cross-Layer Attention
  3. 2025Unlocking the Address Book: Dissecting the Sparse Semantic Structure of LLM Key-Value Caches via Sparse Autoencoders
  4. 2026Fast KV Compaction via Attention Matching
  5. 2024A Simple and Effective L2 Norm-Based Strategy for KV Cache Compression
  6. 2026Learning how to Forget: Fine-tuning for Long-Context Sparse Attention
  7. 2025Expected Attention: KV Cache Compression by Estimating Attention from Future Queries Distribution
  8. 2025Homogeneous Keys, Heterogeneous Values: Exploiting Local KV Cache Asymmetry for Long-Context LLMs