LQER: Low-Rank Quantization Error Reconstruction for LLMs
About
Post-training quantization of Large Language Models (LLMs) is challenging. In this work, we introduce Low-rank Quantization Error Reduction (LQER), which combines quantization and low-rank approximation to recover the model capability. LQER leverages an activation-induced scale matrix to drive the singular value distribution of quantization error towards a desirable distribution, which enables nearly-lossless W4A8 quantization on various LLMs and downstream tasks without the need for knowledge distillation, grid search, or gradient-base iterative optimization. Unlike existing methods, the computation pattern of LQER eliminates the need for specialized Scatter and Gather processes to collect high-precision weights from irregular memory locations. Our W4A8 LLMs achieve near-lossless performance on six popular downstream tasks, while using 1.36$\times$ fewer hardware resources than the leading state-of-the-art method. We open-source our framework at https://github.com/ChengZhang-98/lqer
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText2 | Perplexity5.12 | 2839 | |
| Language Modeling | WikiText-2 (test) | PPL4.94 | 1949 | |
| Language Modeling | C4 | Perplexity7.79 | 1071 | |
| Question Answering | ARC Challenge | Accuracy48.67 | 906 | |
| Question Answering | ARC Easy | -- | 597 | |
| Question Answering | PIQA | Accuracy77.67 | 374 | |
| Question Answering | BoolQ | -- | 317 | |
| Sentence Completion | HellaSwag | Accuracy67 | 276 | |
| Language Modeling | Perplexity | Perplexity (PPL)3.55 | 149 | |
| Word Prediction | LAMBADA | Accuracy74 | 148 |