跳到正文
Reddit · r/LocalLLaMA· /u/pmttyji·· 4 天前AI 评分49

PN-Mix:面向长输出 LLM 推理的 Pair-Norm 引导混合精度 Key Cache 量化论文

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

来源与原文

Reddit · r/LocalLLaMA

作者:/u/pmttyji

发布时间:(北京时间)

阅读原始来源 · reddit.com ↗
完整链接与内容说明

https://www.reddit.com/r/LocalLLaMA/comments/1xd6dfr/paper_pnmix_pairnorm_guided_mixedprecision_key/

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

论文提出 PN-Mix,一种面向 RoPE decoder-only LLM 的混合精度 key cache 量化方法,通过 RoPE 通道对的 query-key pair-norm 乘积识别显著通道对,在 prefill 阶段分配一次高精度并在解码全程复用。2.26-bit 平均精度下,PN-Mix 在推理、代码生成和 AIME 基准上达到与 2.32–2.39 bit 先前方法相当或更好的准确率,且可平滑降至 1.5 bit。

来源:Reddit · r/LocalLLaMA · reddit.com