PulseAugur
实时 11:59:36
English(EN) SoftWater: Class-Aware Rate Allocation for Softmax Quantization

SoftWater 方法优化 LLM 量化以减小内存占用

研究人员开发了一种名为 SoftWater 的新方法,用于量化大型语言模型的 softmax 输出层。该技术将量化视为一个率失真问题,以原始分布和量化分布之间的 KL 散度进行优化。SoftWater 根据类感知分配量化比特,为频繁的低方差类别分配更多精度,为稀有类别分配较少精度,这对于具有 Zipfian 令牌分布的模型尤其有效。该方法优于现有量化器,并在困惑度增加极小的情况下显著减小了模型的内存占用,使得头量化更加实用。 AI

影响 通过在性能下降极小的情况下显著降低内存需求,从而能够更有效地部署 LLM。

排序理由 关于 LLM 量化新方法的学术论文。[lever_c_demoted from research: ic=1 ai=1.0]

在 arXiv cs.LG 阅读 →

AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →

SoftWater 方法优化 LLM 量化以减小内存占用

报道来源 [1]

  1. arXiv cs.LG TIER_1 English(EN) · Joao V. Cavalcanti, Ashia C. Wilson ·

    SoftWater:用于 Softmax 量化的类别感知率分配

    arXiv:2608.12026v1 Announce Type: new Abstract: Post-training quantization pipelines routinely leave the softmax output layer in high precision. Yet in small LLMs with modern vocabularies, the head holds 15--30\% of all parameters, so a nominal ``2-bit'' model with an fp16 head c…