OScaR: The Occam's Razor for Extreme KV Cache Quantization in LLMs and Beyond
cs.LG, cs.CL
Submitted: 2026-05-19
Updated: 2026-10-14
Code: https://github.com/ZunhaiSu/OScaR-KV-Quant
License: http://creativecommons.org/licenses/by/4.0/
The gist: The rapid advancement toward long-context reasoning and multi-modal intelligence has made the memory footprint of the Key-Value (KV) cache a dominant memory bottleneck for efficient deployment.
Terminology
Abstract
The rapid advancement toward long-context reasoning and multi-modal intelligence has made the memory footprint of the Key-Value (KV) cache a dominant memory bottleneck for efficient deployment. While the established per-channel quantization effectively accommodates intrinsic channel-wise outliers in Key tensors, its efficacy diminishes under extreme compression. In this work, we revisit the inherent limitations of the per-channel quantization paradigm from both empirical and theoretical perspectives. Our analysis identifies Token Norm Imbalance (TNI) as the primary bottleneck to quantization fidelity. We demonstrate that TNI systematically amplifies errors when shared quantization parameters are required to span token groups exhibiting substantial norm disparities. Instead of relying on intricate quantization pipelines (e.g., TurboQuant), we propose OScaR (Omni-Scaled Canalized Rotation), an accurate and lightweight KV cache compression framework for X-LLMs (i.e., text-only, multi-modal, and omni-modal LLMs). Advancing the per-channel paradigm, OScaR employs Canalized Rotation followed by Omni-Token Scaling to mitigate TNI-induced sequence-dimensional variance both effectively and efficiently, further supported by our optimized system design and CUDA kernels. Extensive evaluations across X-LLMs show that OScaR consistently outperforms existing methods and achieves near-lossless performance under INT2 quantization, establishing it as a robust, low-complexity, and universal framework that defines a new Pareto front. Compared with the BF16 FlashDecoding-v2 baseline, our OScaR implementation achieves a notable up to 3.0x speedup in decoding, reduces memory footprint by 5.3x, and increases throughput by 4.1x. The code for OScaR is publicly available at https://github.com/ZunhaiSu/OScaR-KV-Quant.
Sources
- HadaCore: Tensor Core Accelerated Hadamard Transform Kernel
- gpt-oss-120b & gpt-oss-20b Model Card
- Systematic Outliers in Large Language Models
- Qwen3-VL Technical Report
- PyramidKV: Dynamic KV Cache Compression based on Pyramidal Information Funneling
- Qwen3-Coder-Next Technical Report
- SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models
- GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformers
- Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs
- Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs
- PolarQuant: Quantizing KV Caches with Polar Transformation
- WorldSense: Evaluating Real-world Omnimodal Understanding for Multimodal LLMs
- IsoQuant: Hardware-Aligned SO(4) Isoclinic Rotations for LLM KV Cache Compression
- Massive Values in Self-Attention Modules are the Key to Contextual Knowledge Understanding
- LLaVA-OneVision: Easy Visual Task Transfer
- A Survey on Large Language Model Acceleration based on KV Cache Management
- Memory-Efficient Visual Autoregressive Modeling with Scale-Aware KV Cache Compression
- TWEO: Transformers Without Extreme Outliers Enables FP8 Training And Quantization For Dummies
- KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache
- A White Paper on Neural Network Quantization
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks