Hugging Face Trending Papers
Oct 13

MultiAlloc: Budget-Aware Precision Allocation for KV Cache Quantization

MultiAlloc is a profiling‑guided framework that allocates limited quantization resources between layer‑wise precision enhancement and token‑wise preservation for KV cache quantization in large language models. By using lightweight offline profiling to assess layer sensitivity and token importance, it automatically derives an effective precision allocation strategy that satisfies a fixed memory budget without online search. Experiments on reasoning and long‑context benchmarks show that MultiAlloc consistently outperforms existing layer‑wise, token‑wise, and multi‑granularity quantization methods across models such as Mistral, Llama, DeepSeek‑Qwen, and QwQ.

arXiv Machine Learning
Oct 15

PN-Mix: Pair-Norm Guided Mixed-Precision Key Cache Quantization for Efficient Long-Output LLM Inference

PN-Mix is a mixed‑precision key‑cache quantization technique for RoPE‑based decoder‑only large language models that allocates higher precision to salient channel pairs identified during the prefill stage. By focusing on channel pairs with large query–key pair‑norm products, PN‑Mix reduces logit variation caused by quantization errors. Experiments on reasoning, code‑generation, and AIME benchmarks show that PN‑Mix achieves comparable or better accuracy than prior methods while using an average precision of 2.26 bits and degrades gracefully down to 1.5 bits.

By Gil Jae Song, Sungjoo Yoo