Introduction
Modern long-context LLMs face a common pain point: the KV cache's GPU memory footprint. Caching "memory fragments" for every token bloats memory, and memory bandwidth becomes the bottleneck—loading more slows everything down. Naive quantization helps, but pushing to 2 bits causes severe degradation.
Root Cause: Token Norm Imbalance (TNI)
Previous quantization approaches implicitly assumed tokens have similar magnitudes. In reality, token vector norms vary dramatically. Applying a single shared quantization parameter wipes out small-norm tokens and misquantizes large ones, causing heavy distortion.
> Key concept: Token Norm Imbalance (TNI) — the huge spread of token vector norms within a sequence makes a unified quantization standard destructive.
The OScaR Framework
The OScaR framework (Omni-Scaled Canalized Rotation) takes a deliberately simple, two-step approach:
1. Canalized Rotation: rotates the cache to smooth out erratic outliers. 2. Omni-Token Scaling: per-token dynamic scaling so every token is equally well-suited to the quantizer.
The core logic is captured by:
$ \hat{\mathbf{V}} = \text{Quant} \left( \mathbf{S} \cdot (\mathbf{R} \cdot \mathbf{V}) \right) $
That is: rotate the raw cache (\(\mathbf{V}\)) via rotation matrix (\(\mathbf{R}\)), apply the omni-token scaling factor (\(\mathbf{S}\)), then quantize.
Reported Results
| Dimension | Traditional BF16 | OScaR (INT2) | Verdict | | :--- | :--- | :--- | :--- | | Memory usage | Bloated, OOM-prone | 5.3x smaller | Extremely lightweight | | Inference speed | Slow | 3x faster | Instant | | Throughput | A few replies at a time | 4.1x higher | Non-stop | | Quality | Baseline | Near-lossless at INT2 | Remarkable |
Even at an aggressive 2-bit (INT2) setting, models equipped with OScaR remain coherent and robust across long conversations and multimodal benchmarks.
Reference
- Title: OScaR: The Occam's Razor for Extreme KV Cache Quantization in LLMs and Beyond
- Released: May 21, 2026
- arXiv: arXiv:2605.19660
- Problem: Excessive KV cache memory in long-context LLM inference, plus severe accuracy loss from extreme low-bit quantization.
- Contribution: Identifies TNI as the main source of quantization loss and proposes the Omni-Scaled Canalized Rotation (OScaR) framework for near-lossless INT2 compression.