不依赖微调的低比特量化方法,提升大模型压缩精度与速度。
LoPRo: Enhancing Low-Rank Quantization via Permuted Block-Wise Rotation
- 通过分块置换与沃尔什-哈达玛变换优化残差矩阵量化
- 2/3比特量化下性能超越现有免微调方法,接近微调效果
- 适合追求高效推理且无微调资源的大模型部署场景
训练后量化(PTQ)可在保持较高准确率的前提下实现有效模型压缩。当前仅量化权重的PTQ方法主要聚焦于难度较大的亚3比特领域,通常面临显著精度下降,常需微调才能达到竞争力水平。本文重新审视权重量化的基本特性,分析低秩近似下残差矩阵量化的挑战,提出一种无需微调的新型PTQ方法LoPRo。该方法通过分块置换和沃尔什-哈达玛变换旋转重要性相似的列,同时显式保留最显著列块的量化精度。此外,引入基于秩1草图(R1SVD)的混合精度快速低秩分解,进一步降低量化开销。实验表明,LoPRo在2比特和3比特量化下均优于现有免微调方法,性能接近微调基线。在LLaMA-2和LLaMA-3系列模型上达到当前最佳量化精度,速度提升最高达4倍。在MoE模型Mixtral-8x7B中,量化耗时仅2.5小时,困惑度降低0.4,准确率提升8%。相较其他低秩量化方法,LoPRo以更低秩实现更高精度,兼具高推理效率与极小额外延迟。
原文摘要 · Abstract (English)
Post-training quantization (PTQ) enables effective model compression while preserving relatively high accuracy. Current weight-only PTQ methods primarily focus on the challenging sub-3-bit regime, where approaches often suffer significant accuracy degradation, typically requiring fine-tuning to achieve competitive performance. In this work, we revisit the fundamental characteristics of weight quantization and analyze the challenges in quantizing the residual matrix under low-rank approximation. We propose LoPRo, a novel fine-tuning-free PTQ algorithm that enhances residual matrix quantization by applying block-wise permutation and Walsh-Hadamard transformations to rotate columns of similar importance, while explicitly preserving the quantization accuracy of the most salient column blocks. Furthermore, we introduce a mixed-precision fast low-rank decomposition based on rank-1 sketch (R1SVD) to further minimize quantization costs. Experiments demonstrate that LoPRo outperforms existing fine-tuning-free PTQ methods at both 2-bit and 3-bit quantization, achieving accuracy comparable to fine-tuning baselines. Specifically, LoPRo achieves state-of-the-art quantization accuracy on LLaMA-2 and LLaMA-3 series models while delivering up to a 4$\times$ speedup. In the MoE model Mixtral-8x7B, LoPRo completes quantization within 2.5 hours, simultaneously reducing perplexity by 0.4$\downarrow$ and improving accuracy by 8\%$\uparrow$. Moreover, compared to other low-rank quantization methods, LoPRo achieves superior accuracy with a significantly lower rank, while maintaining high inference efficiency and minimal additional latency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。