Search papers, labs, and topics across Lattice.
This paper revisits residual error handling in compensation-based LLM quantization methods like GPTQ and GPTAQ, identifying a suboptimal calibration objective that aligns quantized outputs with compensated weights instead of the original full-precision model output. They redefine the calibration objective to directly align with the original full-precision output and introduce the concept of "compensation-aware error" to account for discrepancies between compensated and original weights. Experiments across various LLMs and quantization settings demonstrate significant performance improvements when integrating these enhancements into GPTQ and GPTAQ.
GPTQ-style quantization can be significantly improved by directly aligning quantized layer outputs with the original full-precision model's output, rather than the compensated weights, and accounting for "compensation-aware error."
Methods based on weight compensation, which iteratively apply quantization and weight compensation to minimize the output error, have recently demonstrated remarkable success in quantizing Large Language Models (LLMs). The representative work, GPTQ, introduces several key techniques that make such iterative methods practical for LLMs with billions of parameters. GPTAQ extends this approach by introducing an asymmetric calibration process that aligns the output of each quantized layer with its full-precision counterpart, incorporating a residual error into the weight compensation framework. In this work, we revisit the formulation of the residual error. We identify a sub-optimal calibration objective in existing methods: during the intra-layer calibration process, they align the quantized output with the output from compensated weights, rather than the true output from the original full-precision model. Therefore, we redefine the objective to precisely align the quantized model's output with the original output of the full-precision model at each step. We then reveal that the residual error originates not only from the output difference of the preceding layer but also from the discrepancy between the compensated and original weights within each layer, which we name the'compensation-aware error'. By inheriting the neuron decomposition technique from GPTAQ, we can efficiently incorporate this compensation-aware error into the weight update process. Extensive experiments on various LLMs and quantization settings demonstrate that our proposed enhancements integrate seamlessly with both GPTQ and GPTAQ, significantly improving their quantization performance. Our code is publicly available at https://github.com/list0830/ResComp.
