跳到正文
arXiv cs.AI· Mads Toftrup, Marco Federici, Boris van Breugel, Paul Whatmough, Markus Nagel·· 13 小时前AI 评分24

统一 Hessian 引导框架实现训练后量化

A Unified Hessian-Guided Framework for Post-Training Quantization

AI 导读

研究者提出一种 Hessian 引导的统一训练后量化框架,基于可处理的 Fisher 近似,将量化范围选择、线性变换与权值舍入纳入同一输出敏感度目标下联合优化。该方法在现代大语言模型上保持实用性,在逐通道与分块量化设置下均持续优于当前最优的变换与舍入方法。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:低位量化通过降低内存占用和计算成本,使大语言模型得以高效部署。要在低精度下实现高准确率,需要选择量化范围、线性变换和权值舍入决策。现有方法通常使用局部重建目标独立优化这些组件,未能考虑量化误差对下游预测的不同影响。直接优化任务级指标能提供更忠实的目标,但由于量化操作的不连续性以及评估其下游影响的高计算成本,这一做法颇具挑战。
在本工作中,我们推导了由权重与激活联合量化引起的输出变化的 Hessian 引导近似。基于可处理的 Fisher 近似,我们构建了一个统一的训练后量化框架,能够在共同的输出敏感度目标下联合优化量化范围、线性变换和权值舍入。所得方法对现代大语言模型仍然实用,并在逐通道和逐块量化设置下均持续优于最先进的变换与舍入方法。
评论: 10 页 + 4 附录
主题: 机器学习(cs.LG);人工智能(cs.AI)
引用为: arXiv:2610.40345 [cs.LG]
  (或 arXiv:2610.40345v1 [cs.LG] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.40345

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Marco Federici [查看邮箱]
[v1] 2026 年 10 月 29 日星期四 16:06:01 UTC(166 KB)

来源:arXiv cs.AI · arxiv.org