All papersLLMs & Language ModelsTop pick on Nov 11University College London

LowRAM: Low-Rank Keys and Amortised Values for Composable KV Cache Compression at Test-Time

/ Paper arXiv 2611.07025

Curator's notes

LowRAM compresses KV caches by predicting values from keys and low-ranklow-rankAn approximation that represents a matrix using fewer dimensions than its full rank. key compression, achieving up to 5× faster decoding at 4× compression.

Curator's notesWritten by AI from the paperEvery number here appears in the abstractReport a mistake

What's new

Introduces per-head value prediction from keys and cross-layer low-rank key compression, a combination not used in prior KV cacheKV cacheA storage of key and value vectors used by transformers during autoregressive decoding. compression methods.

Key result

LowRAM reaches 26.9× compression with 94.5% of baseline performance on long contexts.

Why it matters

Builders can run larger language models with less memory, enabling higher throughput and batch sizes.

Key numbers

94.5%reaching 26.9times compression at 94.5% of baseline performance on long contexts

From the paperCopied from the abstract, not written by AI.

Abstract

From the paperThe authors' own words.

Autoregressive decoding relies on the KV cache, whose memory footprint grows linearly with context length and caps batch size and throughput. Existing post-training methods exploit redundancy across tokens, layers, and feature dimensions, or lower numerical precision. We identify a complementary axis: within each attention head, values are largely predictable from their corresponding keys. We exploit this with value cache amortisationamortisationFitting a lightweight predictor to reconstruct values, storing only residual errors., which fits a lightweight per-head predictor at test time and stores residuals only for the tokens it reconstructs worst, giving 4times near-lossless value compression. The predictor can be initialised by meta-learning for substantially reduced prefill overhead, or fitted from scratch in settings that allow a one-off cost, such as multi-query inference. We pair value amortisationamortisationFitting a lightweight predictor to reconstruct values, storing only residual errors. with cross-layer low-rank key compression to produce LowRAM, an asymmetric compression pipeline that stays close to lossless across three model families. By reconstructing only a selected subset of KV pairs at each decoding step, LowRAM can keep decoding overhead minimal and achieve up to 5times higher peak decode throughput at 4times compression. LowRAM further composes with quantisation and eviction, reaching 26.9times compression at 94.5% of baseline performance on long contexts. These results show that treating keys and values as structurally distinct enables efficient, aggressive, and composable KV cache compression.

From the full paper

From the paperCopied from the paper itself by our code.

Reading the paper on arXiv…

What you could build

Written by AIIntermediateRequires understanding of attention internals and implementing per-head predictors.

  1. Implement a KV cache wrapper that applies LowRAM compression for open-source LLM inference.
  2. Create a benchmarking tool comparing throughput with and without LowRAM on various context lengths.
  3. Develop a plugin for popular inference frameworks to auto‑select KV pairs for reconstruction.

Try it

Code, models and demos that use this paper.

Ask this paper

/ Ask a questionAnswers come from the full paper and name the section

    / Quiz yourself

    What two main techniques does LowRAM combine to compress the KV cache?

    / Your notes

    Select text in the abstract to highlight it or add a note. They stay in this browser.

      / Next pickSplatAnything: Gaussian Spacing Normalization for Feed-forward 3DGS across Arbitrary Scenes
      Read the PDF