arXiv:2503.01901cs.LGcs.AI2025-03被引 11

提出新方法精准识别量化敏感权重,显著提升大模型压缩效果。

Identifying Sensitive Weights via Post-quantization Integral

  • 用后量化积分法精确计算权重敏感度,克服传统方法低估问题。
  • 在Llama 3.2 1B上实现2.66的困惑度提升,优于现有量化方法。
  • 适合关注大模型高效部署与精度优化的研究者与工程师。

服务大语言模型成本高昂,而训练后权重量化可通过压缩模型大小和节省带宽来缓解。由于并非所有权重维度同等重要,现有方法依赖敏感度度量来预处理权重以获得更好量化效果。本文通过实证研究发现,基于梯度和海森矩阵的敏感度度量极不准确:它们对损失函数的影响估计严重偏低,主要源于局部二阶近似收敛半径过小(即泰勒展开中梯度与海森项的局限性)。为此,我们提出后量化积分(PQI)方法,可精细地估计后验敏感度。为进一步利用该度量,我们设计了ReQuant框架,包含两个核心组件:自适应异常值选择与分步显著权重解耦。实验表明,ReQuant显著提升了现有最优的训练后量化方法,在使用QTIP的Llama 3.2 1B模型上实现了2.66的困惑度下降。

原文摘要 · Abstract (English)

Serving Large Language Models (LLMs) is costly. However, post-training weight quantization can address this problem by both compressing their sizes for limited memory and saving bandwidth for acceleration. As not all weight dimensions are equally important, those methods typically rely on a sensitivity metric, which indicates the element-wise influence of weights on loss function and is used to preprocess original weights for better quantization. In this work, we conduct an empirical study on the accuracy of the sensitivity metric, and find that existing gradient and Hessian based metrics are very inaccurate: they underestimate quantization's impact on the loss function by orders of magnitude, mainly due to the small convergence radius of local 2nd order approximation, \ie, gradient and Hessian term in Taylor's formula. To tackle this problem, we propose Post-quantization Integral (PQI), an accurate metric to estimate posterior sensitivity in a fine-grained manner. To leverage this accurate metric, we further propose ReQuant, a simple yet powerful framework that mainly consists of two Dense-and-Sparse detach components: self-adaptive outlier selection and step-wise significant weights detach. Results show that ReQuant boosts state-of-the-art post-training quantization methods, with a pronounced improvement of 2.66 perplexity gain on Llama 3.2 1B with QTIP.

量化大模型敏感度分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。