arXiv:2601.20745cs.LGcs.AI2026-01被引 1

用海森矩阵指导量化训练,让极低比特大模型更准更快优化。

HESTIA: A Hessian-Guided Differentiable Quantization-Aware Training Framework for Extremely Low-Bit LLMs

论文配图:HESTIA: A Hessian-Guided Differentiable Quantization-Aware Training Framework for Extremely Low-Bit LLMs
图 1 · 摘自论文原文
  • 用温度控制的软化替换硬量化,保持训练初期梯度流动。
  • 在Llama-3.2上1.58比特模型零样本性能提升4.34%~5.39%。
  • 适合做极低比特部署的大模型研究者和工程团队。

随着大语言模型持续扩大规模,部署受限于内存墙,促使向极低比特量化转型。然而,现有量化感知训练(QAT)方法从训练初期即采用硬取整和直通估计(STE),过早离散化优化空间,导致潜在权重与量化权重间持续存在梯度不匹配,阻碍量化模型的有效优化。为此,我们提出Hestia,一种面向极低比特大模型的海森矩阵引导可微分量化感知训练框架。该框架以温度控制的softmax松弛替代刚性阶跃函数,早期维持梯度传播的同时逐步硬化量化;同时利用张量级海森迹作为轻量曲率信号,实现细粒度温度退火,支持模型各部分的敏感度感知离散化。在Llama-3.2上的评估显示,Hestia持续优于现有三值量化基线,在1B与3B模型上分别实现平均5.39%与4.34%的零样本性能提升。结果表明,海森引导的松弛能有效恢复表征能力,为1.58比特大模型建立更稳健的训练路径。代码已开源:https://github.com/hestia2026/Hestia。

原文摘要 · Abstract (English)

As large language models (LLMs) continue to scale, deployment is increasingly bottlenecked by the memory wall, motivating a shift toward extremely low-bit quantization. However, most quantization-aware training (QAT) methods apply hard rounding and the straight-through estimator (STE) from the beginning of the training, which prematurely discretizes the optimization landscape and induces persistent gradient mismatch between latent weights and quantized weights, hindering effective optimization of quantized models. To address this, we propose Hestia, a Hessian-guided differentiable QAT framework for extremely low-bit LLMs, which replaces the rigid step function with a temperature-controlled softmax relaxation to maintain gradient flow early in training while progressively hardening quantization. Furthermore, Hestia leverages a tensor-wise Hessian trace metric as a lightweight curvature signal to drive fine-grained temperature annealing, enabling sensitivity-aware discretization across the model. Evaluations on Llama-3.2 show that Hestia consistently outperforms existing ternary QAT baselines, yielding average zero-shot improvements of 5.39% and 4.34% for the 1B and 3B models. These results indicate that Hessian-guided relaxation effectively recovers representational capacity, establishing a more robust training path for 1.58-bit LLMs. The code is available at https://github.com/hestia2026/Hestia.

量化大模型训练优化低比特

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