Loop-similar KV states leave low-outlier residuals, enabling cheap KV cache quantization

sehoonkim418 · x · 2026-10-12

Looping lets a model do more computation with the same weights, but each loop produces its own K/V states, so the KV cache grows with the number of loops — longer contexts and larger batches amplify the cost, limiting concurrency and increasing memory traffic during decoding.

The obvious fix, KV cache quantization, loses too much information at low bit widths. The opportunity: KV states are highly similar across loops. Reconstructing one loop's states from another leaves a residual with far fewer outliers, making it representable in just a few bits. The shared chart shows |K₁| and |K₂| nearly identical while the residual |K₂ − α₂K₁| collapses toward zero, supporting the approach.

Related event: KAIST's ResidualQuant: 2-bit Residual Quantization Cuts Looped Transformer KV Cache by 80%(10 posts)→

Original post →

More from Infra

Infra channel →