首次实现大模型1.61比特量化,突破极低比特压缩瓶颈。
PTQ1.61: Push the Real Limit of Extremely Low-Bit Post-Training Quantization Methods for Large Language Models
- 用一维结构化掩码将关键权重分配至4比特,仅增0.0002比特/权重
- 通过块级缩放优化,使非关键通道实现高效二值化,保持性能
- 提出量化预处理新范式,重塑权重分布以降低极低比特压缩难度
大语言模型在极端低比特(低于2比特)量化时性能严重下降。现有子2比特后训练量化(PTQ)方法采用混合精度策略,通过非结构化细粒度掩码区分重要权重,但每权重额外增加1比特或更多。为探索PTQ的真实极限,本文提出极低比特量化方法PTQ1.61,首次实现1.61比特量化。核心思路:从输入激活角度出发,引入一维结构化掩码,将关键权重通道分配至4比特,仅增加0.0002比特/权重;对非关键通道,设计高效的块级缩放因子优化框架,隐式考虑行间相关性和角度偏差。不同于以往聚焦量化方法调整的工作,本文提出“量化预处理”新范式,认为量化前重构预训练模型权重分布可缓解极低比特逐通道量化的困难。大量实验表明,PTQ1.61在极端低比特量化中达到当前最优性能。代码已开源:https://github.com/zjq0455/PTQ1.61。
原文摘要 · Abstract (English)
Large Language Models (LLMs) suffer severe performance degradation when facing extremely low-bit (sub 2-bit) quantization. Several existing sub 2-bit post-training quantization (PTQ) methods utilize a mix-precision scheme by leveraging an unstructured fine-grained mask to explicitly distinguish salient weights, while which introduces an extra 1-bit or more per weight. To explore the real limit of PTQ, we propose an extremely low-bit PTQ method called PTQ1.61, which enables weight quantization to 1.61-bit for the first time. Specifically, we first introduce a one-dimensional structured mask with negligibly additional 0.0002-bit per weight based on input activations from the perspective of reducing the upper bound of quantization error to allocate corresponding salient weight channels to 4-bit. For non-salient channels binarization, an efficient block-wise scaling factors optimization framework is then presented to take implicit row-wise correlations and angular biases into account. Different from prior works that concentrate on adjusting quantization methodologies, we further propose a novel paradigm called quantization preprocessing, where we argue that transforming the weight distribution of the pretrained model before quantization can alleviate the difficulty in per-channel extremely low-bit PTQ. Extensive experiments indicate our PTQ1.61 achieves state-of-the-art performance in extremely low-bit quantization. Codes are available at https://github.com/zjq0455/PTQ1.61.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。