首次实现开源指令微调大模型2比特量化,性能领先。
Unifying Block-wise PTQ and Distillation-based QAT for Progressive Quantization toward 2-bit Instruction-Tuned LLMs
- 融合块级后训练量化与蒸馏式量化感知训练,分步实现2比特压缩。
- 在MMLU和IFEval上达到当前最优表现,无需私有数据。
- 适合资源受限设备部署的低比特大模型应用者参考。
随着大语言模型规模快速扩张,其在资源受限设备上的部署面临挑战,极端低比特量化(如2比特)成为研究热点。尽管已有研究证明2比特大模型在准确率与延迟上优于4比特小模型,但这些成果仍局限于预训练模型,尚未拓展至指令微调模型。为此,我们提出统一渐进量化(UPQ)框架——从FP16→INT4→INT2,首次实现无需依赖专有后训练数据的开源指令微调大模型2比特量化。UPQ首先通过块级后训练量化(PTQ)将FP16模型转为INT4,显著降低后续INT2量化的误差;再采用基于蒸馏的量化感知训练(Distill-QAT),通过最小化两模型间广义Jensen-Shannon散度(JSD),使INT2模型生成响应与原FP16版本一致。实验表明,UPQ在MMLU与IFEval两大主流指令微调模型评测基准上均取得当前最佳性能。
原文摘要 · Abstract (English)
As the rapid scaling of large language models (LLMs) poses significant challenges for deployment on resource-constrained devices, there is growing interest in extremely low-bit quantization, such as 2-bit. Although prior works have shown that 2-bit large models are pareto-optimal over their 4-bit smaller counterparts in both accuracy and latency, these advancements have been limited to pre-trained LLMs and have not yet been extended to instruction-tuned models. To bridge this gap, we propose Unified Progressive Quantization (UPQ)$-$a novel progressive quantization framework (FP16$\rightarrow$INT4$\rightarrow$INT2) that unifies block-wise post-training quantization (PTQ) with distillation-based quantization-aware training (Distill-QAT) for INT2 instruction-tuned LLM quantization. UPQ first quantizes FP16 instruction-tuned models to INT4 using block-wise PTQ to significantly reduce the quantization error introduced by subsequent INT2 quantization. Next, UPQ applies Distill-QAT to enable INT2 instruction-tuned LLMs to generate responses consistent with their original FP16 counterparts by minimizing the generalized Jensen-Shannon divergence (JSD) between the two. To the best of our knowledge, we are the first to demonstrate that UPQ can quantize open-source instruction-tuned LLMs to INT2 without relying on proprietary post-training data, while achieving state-of-the-art performances on MMLU and IFEval$-$two of the most representative benchmarks for evaluating instruction-tuned LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。