通过保留重要权重结构,提升大模型量化精度
Preserve-Then-Quantize: Balancing Rank Budgets for Quantization Error Reconstruction in LLMs
- 先保留权重主要方向,再对残差做量化与误差重建
- 2比特量化下GLUE得分平均提升5.9个百分点
- 适合需要高效微调的大模型部署场景
量化误差重构(QER)通过将权重近似为 $\mathbf{W} \approx \mathbf{Q} + \mathbf{L}\mathbf{R}$,利用秩-$r$修正项恢复量化误差,从而降低后训练量化(PTQ)的精度损失。现有方法将全部秩预算用于误差重构,但在权重本身具有低秩结构且量化破坏主方向时表现不佳。我们提出结构化残差重构(SRR),在量化前保留激活缩放后的权重的前-$k$个奇异子空间,仅对残差进行量化,并用剩余秩 $r-k$ 进行误差重构。我们基于理论推导出选择 $k$ 的准则,平衡量化暴露能量与不可恢复误差。进一步证明,所得 $\mathbf{Q} + \mathbf{L}\mathbf{R}$ 参数化天然支持量化参数高效微调(QPEFT),并通过沿保留方向的梯度缩放稳定微调过程。实验表明,在多种模型与量化设置下,该方法持续降低困惑度;在2比特QPEFT下,GLUE平均得分提升5.9个百分点。
原文摘要 · Abstract (English)
Quantization Error Reconstruction (QER) reduces accuracy loss in Post-Training Quantization (PTQ) by approximating weights as $\mathbf{W} \approx \mathbf{Q} + \mathbf{L}\mathbf{R}$, using a rank-$r$ correction to reconstruct quantization error. Prior methods devote the full rank budget to error reconstruction, which is suboptimal when $\mathbf{W}$ has intrinsic low-rank structure and quantization corrupts dominant directions. We propose Structured Residual Reconstruction (SRR), a rank-allocation framework that preserves the top-$k$ singular subspace of the activation-scaled weight before quantization, quantizes only the residual, and uses the remaining rank $r-k$ for error reconstruction. We derive a theory-guided criterion for selecting $k$ by balancing quantization-exposed energy and unrecoverable error under rank constraints. We further show that resulting $\mathbf{Q} + \mathbf{L}\mathbf{R}$ parameterization naturally supports Quantized Parameter-Efficient Fine-Tuning (QPEFT), and stabilizes fine-tuning via gradient scaling along preserved directions. Experiments demonstrate consistent perplexity reductions across diverse models and quantization settings in PTQ, along with a 5.9 percentage-point average gain on GLUE under 2-bit QPEFT. The project page is available at https://ai-isl.github.io/srr.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。