Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

Cheng Zhang, Jianyi Cheng, George A. Constantinides, Yiren Zhao• 2024

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText2
Perplexity5.12
3785
Language ModelingWikiText-2 (test)
PPL4.94
2333
Language ModelingC4
Perplexity7.79
1688
Question AnsweringARC Challenge
Accuracy48.67
906
Question AnsweringARC Easy--
597
Question AnsweringPIQA
Accuracy77.67
505
Sentence CompletionHellaSwag
Accuracy67
364
Question AnsweringBoolQ--
317
Word PredictionLAMBADA
Accuracy74
192
Language ModelingPerplexity
Perplexity (PPL)3.55
149
Showing 10 of 22 rows

Other info

Follow for update