GlowQ通过分组共享低秩近似,提升4比特量化大模型的精度与效率。
GlowQ: Group-Shared LOw-Rank Approximation for Quantized LLMs
- 按输入共享分组缓存统一右因子,仅恢复收益最高的组或层。
- 相比基线降低5.6%首字延迟,提升9.6%吞吐,困惑度降0.17%。
- 适合部署低比特量化大模型,兼顾速度与精度,尤其适合资源受限场景。
如BitsAndBytes、AWQ和GPTQ等量化技术广泛用于大语言模型部署,但使用低比特表示(如4比特)时常导致精度下降。低秩修正方法(如LQER、QERA、ASER)被提出以缓解此问题,但需对所有层进行恢复,并在每个解码器块插入纠错模块,造成延迟和内存开销增加。为此,我们提出GlowQ,一种针对量化大模型的分组共享低秩近似方法:为每个输入共享组缓存单一共享右因子,仅恢复带来最高精度提升的组或层。GlowQ对每个输入共享组仅计算一次高精度投影并跨模块复用,降低参数与内存开销,同时保持层特定修正的表达能力。我们还提出选择性版本GlowQ-S,仅在收益最大处应用缓存模块。相比强基线,我们的方法平均降低5.6%首字响应时间(TTFB),提升9.6%吞吐率,WikiText-2上困惑度降低0.17%,下游任务准确率提升0.42个百分点。选择性模型GlowQ-S进一步降低延迟,将TTFB减少23.4%,吞吐率提升37.4%,平均精度损失低于0.2个百分点。代码已开源:https://github.com/ahnselim/GlowQ。
原文摘要 · Abstract (English)
Quantization techniques such as BitsAndBytes, AWQ, and GPTQ are widely used as a standard method in deploying large language models but often degrades accuracy when using low-bit representations, e.g., 4 bits. Low-rank correction methods (e.g., LQER, QERA, ASER) has been proposed to mitigate this issue, however, they restore all layers and insert error-correction modules into every decoder block, which increases latency and memory overhead. To address this limitation, we propose GlowQ, a group-shared low-rank approximation for quantized LLMs that caches a single shared right factor per input-sharing group and restores only the groups or layers that yield the highest accuracy benefit. GlowQ computes the high-precision projection once per input-sharing group and reuses it across its modules, reducing parameter and memory overhead, and retaining the expressivity of layer-specific corrections. We also propose a selective variant, GlowQ-S, that applies the cached shared module only where it provides the largest benefit. Compared with strong baselines, our approach reduces TTFB by (5.6%) and increases throughput by (9.6%) on average, while reducing perplexity on WikiText-2 by (0.17%) and increasing downstream accuracy by 0.42 percentage points. The selective model GlowQ-S further reduces latency, cutting TTFB by (23.4%) and increasing throughput by (37.4%), while maintaining accuracy within 0.2 percentage points on average. Code is available at https://github.com/ahnselim/GlowQ.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。