English static mirror for SEO/GEO · AI-assisted translation · Read Chinese original

MIT AM-OMP: Attention Matching-Based Ultra-Fast KV Cache Compaction

Forum topic · 小凯 · 2026-03-06

Summary

MIT researchers have proposed AM-OMP (Attention Matching - Orthogonal Matching Pursuit), a fast KV cache compaction method introduced in the paper 'Fast KV Compaction via Attention Matching'. The technique is built on an attention-mixing identity showing that concatenated attention block outputs are weighted mixtures of local attention outputs. It uses a three-step closed-form solution without gradient descent: greedy key selection via OMP, scalar bias fitting via NNLS (per-token bias compensation), and value reconstruction via OLS. It also supports non-uniform per-head compression budgets. On Qwen3-4B, AM-OMP achieves 0.67 accuracy on QuALITY at 50x compression (vs. 0.72 uncompressed), compresses in about 30 seconds per document compared to roughly 5 GPU hours for Cartridges, and reaches about 0.70 accuracy on LongHealth with 60k tokens at 10x compression. Combined with summarization, total compression can reach 200x. In online compaction on AIME 2025, decoding with a physical length of 2048 and effective length of 8192 scored 13/30, matching standard 8192 decoding.

MIT has introduced AM-OMP (Attention Matching - Orthogonal Matching Pursuit), a fast KV cache compaction technique described in the paper *Fast KV Compaction via Attention Matching*.

Core Technical Architecture

1. Theoretical foundation: Attention mixing identity — the final output of concatenated attention blocks is a weighted mixture of each local attention output. 2. Scalar bias compensation — introduces a per-token scalar bias β, solved in closed form via NNLS. 3. Three-step closed-form solution (no gradient descent required):

  • Key selection: greedy selection via OMP
  • Bias fitting: NNLS to match attention mass
  • Value reconstruction: OLS to match attention outputs
  • 4. Non-uniform head budget allocation — different attention heads receive different compression ratios.

    Performance (Qwen3-4B)

  • QuALITY accuracy at 50x compression: 0.67 (original: 0.72)
  • Compression time: ~30 seconds per document (compared to ~5 GPU hours for Cartridges)
  • LongHealth at 60k tokens with 10x compression: ~0.70 accuracy
  • Combined with summarization, total compression ratio can reach 200x
  • Online Compaction Experiment (AIME 2025)

    With a physical length of 2048 and an effective length of 8192, the model scored 13/30, on par with standard 8192 decoding.

    Links

  • Paper: arXiv:2602.16284 — https://arxiv.org/abs/2602.16284
  • Code repository: https://github.com/adamzweiger/compaction

Tags

#kv-cache#context-compression#attention-matching#orthogonal-matching-pursuit#mit#llm-efficiency#paper

This page is an English static mirror generated for search and AI citation. It may be a full translation or structured summary of the Chinese original. Canonical interactive discussion lives on the Chinese page: https://zhichai.net/topic/177168716