PN-Mix Achieves Comparable Accuracy to Higher-Bit Baselines at 2.26-Bit Precision for LLM Inference
·2026.10.23 15:26
Key point
The method uses pair-norm analysis to allocate precision to salient channel pairs, maintaining accuracy down to 1.5 bits where baselines degrade sharply.
Details
- PN-Mix is a mixed-precision key-cache quantization method for RoPE-based decoder-only LLMs that addresses KV cache memory and bandwidth bottlenecks in long-output inference.
- By analyzing attention logit perturbation at the RoPE channel-pair level, the method identifies that quantization errors in channel pairs with large query-key pair-norm products significantly impact logit variation.
- PN-Mix allocates higher precision to these salient channel pairs identified once during the prefill stage and reuses them throughout decoding.
- At 2.26-bit average precision, PN-Mix achieves comparable or better average accuracy than prior mixed-precision methods using 2.32–2.39 bits on standard reasoning, code-generation, and AIME benchmarks.
- In Pareto sweeps, PN-Mix degrades gracefully down to 1.5 bits, whereas compared baselines exhibit sharper performance drops below 2.25 bits.
This summary was generated automatically by AI. Check the original for the author's claims and context. Copyright belongs to the original author.
Our guide explains how the AI works. Report summary errors, attribution issues, or removal requests via Contact.