arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02446cs.LGcs.CL
收藏

LQER:面向LLM的低秩量化误差重建

LQER: Low-Rank Quantization Error Reconstruction for LLMs

Imperial College London(帝国理工学院);University of Cambridge(剑桥大学)

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

AI总结:

本文提出LQER方法,结合量化与低秩近似,利用激活诱导缩放矩阵驱动量化误差的奇异值分布,在无需知识蒸馏等复杂优化的情况下实现近乎无损的W4A8量化,并在下游任务中减少硬件资源消耗。

摘要:

大型语言模型(LLMs)的训练后量化具有挑战性。在本研究中,我们提出了低秩量化误差降低(LQER),它结合量化和低秩近似来恢复模型能力。LQER利用激活诱导的缩放矩阵,将量化误差的奇异值分布驱动至理想分布,从而在无需知识蒸馏、网格搜索或基于梯度的迭代优化的情况下,在各种LLM和下游任务上实现近乎无损的W4A8量化。与现有方法不同,LQER的计算模式无需专门的Scatter和Gather过程即可从不规则内存位置收集高精度权重。我们的W4A8 LLM在六个流行下游任务上实现近无损性能,同时使用的硬件资源比领先的最先进方法少1.36倍。我们在https://github.com/ChengZhang-98/lqer开源了该框架。

英文摘要:

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

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑