English

Preserve-Then-Quantize: Balancing Rank Budgets for Quantization Error Reconstruction in LLMs

Machine Learning 2026-05-14 v2 Artificial Intelligence

Abstract

Quantization Error Reconstruction (QER) reduces accuracy loss in Post-Training Quantization (PTQ) by approximating weights as WQ+LR\mathbf{W} \approx \mathbf{Q} + \mathbf{L}\mathbf{R}, using a rank-rr correction to reconstruct quantization error. Prior methods devote the full rank budget to error reconstruction, which is suboptimal when W\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-kk singular subspace of the activation-scaled weight before quantization, quantizes only the residual, and uses the remaining rank rkr-k for error reconstruction. We derive a theory-guided criterion for selecting kk by balancing quantization-exposed energy and unrecoverable error under rank constraints. We further show that resulting Q+LR\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.

Keywords

Cite

@article{arxiv.2602.02001,
  title  = {Preserve-Then-Quantize: Balancing Rank Budgets for Quantization Error Reconstruction in LLMs},
  author = {Yoonjun Cho and Dongjae Jeon and Soeun Kim and Moongyu Jeon and Albert No},
  journal= {arXiv preprint arXiv:2602.02001},
  year   = {2026}
}

Comments

Accepted at ICML 2026. Project page: https://ai-isl.github.io/srr

R2 v1 2026-07-01T09:31:39.168Z