★ FEATURED ARTICLE 要说近几年大模型本地部署领域最值得反复研究的一篇论文我第一个想到的就是GPTQ。读过它和没读过它对模型量化这件事的理解完全是两个层次。我自己第一次在单卡上压一个几十B量级的模型时也被所谓的后训练量化坑得够呛——模型没压下来多少困惑度倒是涨得飞快。后来把GPTQ的原理彻底吃透才真正搞清楚量化误差是怎么传导的也才明白为什么它能在单张A100上用大概4小时把175B的模型压到3bit/4bit而且困惑度几乎不掉。这篇博文就从GPTQ原本的问题、数学内核、实操参数、效果评估和后续生态五个方面完整拆一遍。1. GPTQ出现之前为什么量化一个175B模型这么难1.1 大模型量化的两条路线训练时量化与后训练量化大模型量化本质上就是在更小的存储和更快的推理与更低的输出质量之间找平衡。很多人以为量化就是把浮点数四舍五入成整数那么简单实际上在大模型这个尺度上问题会被成百倍放大。量化大体分两条路线。一条叫量化感知训练QAT你在训练阶段就把量化误差喂回网络让模型自己学会抵抗精度损失。这条路在BERT、GPT那个体量的时候还行但到了175B这种规模重新训练一遍的成本几乎是不可接受的。你算一下175B的模型用上千张A100训一次动辄几周就算只做部分层的重训资源消耗也极其夸张。另一条路是后训练量化PTQ模型训练完不去动它的参数只在推理前把权重从FP16转成INT8/INT4等低比特表示。这条路的好处很明显不需要重新训练只需要少量校准数据几乎可以即插即用。但它的难点也致命模型越大、比特数越低PTQ的精度损失越难控制。在GPTQ出现之前PTQ在百亿参数以上的模型上基本是能压但不敢用的状态。1.2 直接舍入RTN在低bit下的失败最朴素的PTQ方法叫RTNRound to Nearest就是直接把每个权重四舍五入到最近的量化值。听起来很直觉对吧反正权重的分布整体上还是近似保留了。在INT8精度下RTN确实够用损失微乎其微。但在INT4甚至INT3这个档位每个量化步长变大舍入误差随之增大模型输出的困惑度会肉眼可见地恶化。更麻烦的是大模型权重里往往存在一些绝对值很大的离群点outlier。一个FP16权重值是12.3量化到INT4时理论上限可能只有7或者8这种权重直接被截断clipping掉舍入误差是灾难性的。而且RTN不会做任何补偿——它假设每个权重的量化是独立的、互不干扰的。这个假设在数学上就不成立因为权重矩阵里的列彼此之间是高度耦合的一个权重的量化误差会通过矩阵乘法传导到输出的每一个维度上。一句话总结RTN的失败它以为量化误差可以独立统计实际上误差是在矩阵运算中不断累加放大的。1.3 OBQ虽然准但跑不动复杂度问题在GPTQ之前学术界已经有一个相对可靠的解法叫OBQOptimal Brain Quantization。OBQ的思路很聪明量化每个权重的时候不只是把它四舍五入而是同步调整其他尚未量化的权重用它们的微小偏移去对冲这一次量化造成的误差。这个思路来自上世纪90年代Yann LeCun提出的OBSOptimal Brain Surgeon方法原本是用于网络剪枝的OBQ把它迁移到了量化场景。OBQ的效果确实好在OPT-1.3B这类模型上它的困惑度损失比RTN低一大截。但OBQ有个致命的性能瓶颈它每量化一个权重就要重新计算一次所有剩余权重的补偿量。对于一个175B的模型权重矩阵里有几十亿个参数这种逐参数扫描和更新的复杂度在(O(n^4))级别算到天荒地老也算不完。所以当时业界的真实处境是小模型上OBQ表现不错但一旦模型规模上到十亿甚至百亿参数这个方法就跑不动了。而RTN虽然性能可以接受量化质量又不行。2. 二阶曲率到底怎么用GPTQ的数学内核2.1 量化误差的Taylor展开与Hessian矩阵要理解GPTQ为什么能把精度损失压到几乎为零得先弄明白它对待量化误差的态度不是忽略误差而是让误差在数学上最小化。假设你有一个已经训练好的模型损失函数是(L)权重是(w)。现在你把某个权重(w_q)量化成了(\tilde{w_q})相当于给权重加了一个扰动(\delta \tilde{w_q} - w_q)。损失函数的变化可以用泰勒展开来近似[ \Delta L \approx \nabla L^T \delta \frac{1}{2}\delta^T H \delta ]其中(H)是损失函数对权重的Hessian矩阵二阶导数矩阵。由于模型已经训练收敛梯度(\nabla L)基本为0所以第一项可以忽略。真正决定量化误差影响大小的是第二项——也就是Hessian矩阵(H)。Hammer这个点很重要。Hessian矩阵刻画的是权重和权重之间如何协同影响损失。如果两个权重高度相关那么其中一个有了误差另一个可以通过调整来补偿。这就像双人划船一个人突然使不上劲另一个人只要多出点力船还是能保持直线前进。GPTQ做的事情本质上就是让还没被量化的另一个人主动多出力。2.2 最优脑外科手术用Hessian逆做权重补偿GPTQ使用Hessian矩阵的逆来指导补偿计算。它的核心思想是在量化某一行或某一列的权重时找到一个最优的补偿方向使整体损失增加最小。具体来说对于权重矩阵(W)假设我们按列方式进行量化。在量化某一列(w_\cdot)时量化误差会给模型输出带来一个扰动。通过Hessian矩阵的逆可以求出其他未量化列应该调整的量[ \delta_F -\frac{w_q}{[H^{-1}]{qq}} \cdot H^{-1}{\cdot, q} ]这个公式在OBQ中已经出现但GPTQ的核心贡献是让它能在大模型上跑起来。如果理解不了公式也没关系你只需要抓住一个概念每次量化都会留下一个误差GPTQ知道这个误差会往哪里扩散于是在同一层的其他权重上做一个反向的微调把误差抹掉一部分。这种补偿策略就是它的二阶曲率内核。2.3 从OBS到OBQ再到GPTQ三代方法的演进这其实是一条很清晰的技术演进线。OBS是1990年代针对网络剪枝提出的它告诉你如果要移除一个权重为了保持损失不变其他权重应该怎么调一刀切下去还要缝针。OBQ把OBS迁移到量化上量化问题比剪枝更精细——你不需要移除权重只是降低它的精度但整体思路一致每次量化都做一次最优补偿。问题是OBQ逐参数计算补偿在大模型上不可行。GPTQ对OBQ做了三个实质性的算法改进让它从理论上可行变成工程上可跑。这三个改进就是下一节要说的核心加速技巧。2.4 三个关键加速固定顺序、懒惰批量更新、Cholesky分解GPTQ论文里最精华的部分是它在保留OBQ精度的同时把计算复杂度降到了可以处理175B模型的程度。首先是固定顺序贪心。OBQ在每一步都重新选择量化哪个权重最划算这个动态选择的计算量极高。GPTQ通过分析发现在固定顺序比如按列顺序从左到右下量化精度损失并没有显著增加。于是它放弃了动态挑选改成固定顺序逐列处理省掉了大量重复的最优决策计算。其次是懒惰批量更新。OBQ量化一个权重就立刻更新所有剩余权重这个操作非常频繁。GPTQ发现与其每个权重更新一次不如把权重分成块block在块内先累计所有的量化误差等整块处理完再统一更新一次权重。这就像你写代码时不会每敲一个字符就保存一次文件而是写完一个函数再统一保存。这个改动把频繁的矩阵更新变成了批量矩阵运算GPU的吞吐量瞬间就上来了。最后是Cholesky分解。计算Hessian矩阵的逆在数学上是(O(n^3))的复杂度而且数值上不稳定。GPTQ通过Cholesky分解在分块块内预先分解好Hessian矩阵然后直接利用分解结果来做补偿。这个方法在内存占用上也极其友好每块的处理是独立完成的所以你不需要把整个模型的所有Hessian都塞进显存。这三个改进加在一起直接让OBQ从没法用变成了4小时压完175B。3. 单卡跑通175B算法流程与实操参数3.1 GPTQ的完整算法流程理解了内核我们来看实操时的算法流程。GPTQ对每一层做处理的过程大致可以拆成四步第一步拿到某一层的输入激活值(X)计算这一层的Hessian矩阵(H 2XX^T \lambda I)。这里的(\lambda)是阻尼项用来保证矩阵可逆同时也让数值更稳定。第二步对Hessian矩阵做Cholesky分解得到下三角矩阵。这样后续求逆或者做列操作的时候直接用分解结果替代避免反复求逆。第三步把权重矩阵按列分成多个块block size通常取128。在每个块内部逐列进行三个操作计算该列最优量化尺度、对该列权重做舍入量化、累计量化误差。块内所有列处理完成后用Hessian相关的补偿项统一更新后续未量化列的权重。第四步保存量化后的权重和量化参数scale、zero-point等继续处理下一层。整个流程是逐层的所以内存峰值只和某一层的权重和激活有关而不是整个模型。这就是为什么175B的大模型在单张A100上也能完成量化——只要你的CPU内存够大能容纳模型的原始权重GPU只需要按层处理即可。3.2 硬件与软件环境准备论文里报告的时间是175B模型在单张NVIDIA A10080GB上约需4小时完成3bit和4bit的量化峰值GPU内存占用不到15GB。这个数据我实测过类似的流程基本相符。A100的算力当然很重要但更关键的是CPU内存你需要把整个模型的FP16权重放到内存里。175B的FP16权重约350GB所以量化175B模型时你的CPU内存最好有512GB甚至更多。如果你只是量化和部署7B、13B这些量级的模型门槛就低很多了。一台32GB内存的游戏本配合一张24GB显存的消费级显卡就能跑。我自己在7B模型上做4bit量化用单张RTX 4090大概3到5分钟完事13B模型也就10到15分钟。软件层面现在主流是两条路径一是HuggingFace Transformers直接集成的GPTQConfig二是开源的AutoGPTQ库。我用后者比较多因为它对模型格式的支持更完整量化后的模型可以直接导出给vLLM、ExLlamaV2这类推理引擎使用。3.3 关键超参数group size、damp percent、desc_act实操时有两个参数必须仔细理解group size和desc_act。group size指的是权重矩阵按列分组时多少列共享一套量化参数scale和zero-point。默认值是128我个人的经验是128是质量和体积的最佳折中。如果你追求极致精度可以试group size32量化误差会更小但存储开销会增加权重本身小但scale这些元数据变得相对更多最终实际体积差不多多了10%到15%。反过来group size512会让体积更小但精度会下降一截一般只在显存极其紧张时才用。desc_actact-order是另一个关键开关。开启后GPTQ会按每列激活值的绝对值大小来排序处理顺序先量化对输出影响最大的列。这样精度更高但推理阶段需要一个额外的gather操作来恢复列顺序多少会拖慢推理速度。我的建议是如果你要做推理服务且对延迟敏感可以选择desc_actFalse如果只是离线量化后追求最高精度就保持True。damp_percent这个参数是Hessian对角线上的阻尼比例默认0.01。我见过有人在这个参数上踩坑某些模型如果校准集规模偏小Hessian会接近奇异矩阵这时候适当加大damp_percent到0.1反而能稳定量化过程代价是精度略微下降。3.4 Transformers落地示例现在用Transformers做GPTQ量化已经非常简洁官方GPTQConfig可以直接完成量化。以下代码段是实际可用的from transformers import AutoTokenizer, AutoModelForCausalLM, GPTQConfig model_id meta-llama/Llama-2-7b-hf tokenizer AutoTokenizer.from_pretrained(model_id, use_fastTrue) config GPTQConfig( bits4, group_size128, desc_actTrue, damp_percent0.01, datasetc4, tokenizertokenizer, ) model AutoModelForCausalLM.from_pretrained( model_id, quantization_configconfig, device_mapauto, )这段代码会自动下载模型权重、在C4数据集上取一小部分校准样本、逐层完成4bit量化最终加载的就是量化后的模型。注意一点calibration dataset要用和目标任务分布相近的文本。如果模型是用于代码生成的你用纯新闻语料做校准量化出来的效果会打折扣。4. 困惑度近乎无损的真相量化效果实测4.1 困惑度Perplexity到底是什么讨论量化效果时最常用的指标是困惑度Perplexity缩写PPL。困惑度衡量的是一个语言模型对给定文本的预测能力数值越低说明模型对下一个词的预测越准确也就是语言建模能力越强。直观理解就是模型面对一段文本时有多惊讶。一个训练良好的模型对正常文本几乎不惊讶PPL值就低量化后的模型如果预测能力退化面对同样文本时惊讶程度上升PPL就升高。GPTQ论文的实验在WikiText-2数据集上考察了OPT系列多个尺寸的模型从125M到175B都覆盖到了。这里我直接引用论文中OPT-175B的数据可以非常直观地看到量化带来的损失量级。4.2 OPT-175B在WikiText-2上的实测数据配置困惑度近似值相对FP16变化FP16 原始权重约8.34基准GPTQ 4bit约8.370.03GPTQ 3bit约8.460.12GPTQ 2bit约10.482.14看这个表4bit下困惑度只上升了0.033bit也才上升0.12。这个级别的损失对实际生成质量的影响基本是感知不到的。哪怕到2bit模型输出已经明显变差但依然能生成语法通顺的文本——这在此前的任何方法上都是做不到的。作为对比RTN在同样的3bit设定下困惑度通常会劣化1到2个点生成的文本开始出现明显的语义断裂。GPTQ在3bit上的一骑绝尘靠的就是用Hessian曲率信息做了误差补偿。需要留意的是这些数据是在通用领域的文本上评测的。如果你的场景是垂直领域比如法律文书、医疗问答量化时校准数据要换成对应领域的语料效果才能对齐。4.3 3bit vs 4bit质量与体积的实际取舍175B模型在不同精度下的体积大致这样算FP16约350GB4bit约87.5GB加上量化元数据大概90GB出头3bit约65.6GB加上元数据大概68GB所以如果你只有一张80GB显存的A100/H1003bit几乎是刚好能塞进显存的极限而4bit就必须配合CPU offload才能跑。很多人的直觉是4bit比3bit好那我统一用4bit。但实际部署时3bit带来的40%体积差意味着你能不能用单卡跑这是质变。我个人的经验是在13B这个规模上3bit和4bit的生成质量差距很小到了70B以上优先选择4bit因为模型足够大冗余度高4bit的精度优势更明显。至于3bit更适合那种显存卡得很死、又必须单卡推理的场景。4.4 光看困惑度还不够生成质量与zero-shot任务困惑度是效率很高的指标但它不代表全部。量化方法在评估时还需要看生成任务的下游效果比如常识推理zero-shot任务准确率。GPTQ论文里做了丰富的验证在OPT和BLOOM两个模型系列上测了多个zero-shot基准3bit下的准确率相对FP16平均下降不到1%4bit几乎持平。实践中有个常见现象是困惑度看起来只差0.1但在长文本生成时量化模型可能更容易出现重复循环、逻辑短路。所以如果你要部署一个对话或生成服务上线前除了跑PPL最好再跑一轮长度为512或1024的长文本生成人工抽检几十条输出。我踩过这个坑光看困惑度选了一个2bit模型实际生成时经常逻辑断裂最后只能老老实实换回4bit。5. 为什么是代际奠基神作后GPTQ时代的量化生态5.1 从GPTQ到AWQ、QuIP激活感知量化的崛起GPTQ解决了逐层最优量化误差补偿的工程问题但它有一个明显的盲区只看权重分布不看激活分布。实际推理时激活值经常存在明显的离群点activation outlier这些离群通道对量化结果的影响甚至比权重本身的舍入误差还大。所以GPTQ之后学界的方向分了几支。AWQActivation-aware Weight Quantization的思路是先统计校准数据中每个通道激活值的分布找出那些激活值显著偏大的重要通道然后在量化时给这些通道的权重分配更充裕的精度。SmoothQuant的思路则是主动把激活中的离群值平滑迁移到权重里让激活分布更均匀再去做常规量化。QuIP则用随机正交旋转让权重和激活的分布更接近各向同性量化时误差更小。这些方法虽然各有侧重但基本都建立在GPTQ的框架之上按层处理、校准集统计、低比特量化、误差补偿。GPTQ作为一个底座定义了后训练量化的基本范式。5.2 GPTQ与主流推理框架的结合GPTQ发表后开源生态快速跟进。AI社区很快就出现了GPTQ-for-LLaMA项目让大热的LLaMA系列模型可以一键量化为4bit。再后来AutoGPTQ把整个流程包装成了pip安装即可用的库并且支持导出到多种格式。到今天vLLM、ExLlamaV2、TGIText Generation Inference这些主流推理引擎都直接支持加载GPTQ量化模型。我目前实测下来的一个选型经验是推理框架GPTQ支持情况适合场景vLLM原生支持高并发服务、吞吐优先ExLlamaV2原生支持单卡消费级显卡、追求低延迟Transformers内置支持快速验证、动态加载llama.cppGGUF不兼容GPTQ格式纯CPU/混合架构推理注意一点llama.cpp走的是GGUF格式这是另一个独立的量化流派和GPTQ互不兼容。如果你在HF上看到一堆GGUF文件那需要另一种工具链来加载别和GPTQ混淆。5.3 边界与后续GPTQ解决不了的问题GPTQ并非万能。它在batch推理和并发场景下的表现不够稳定因为量化精度是静态的一旦batch内样本的激活分布和校准集差异过大误差就会显现。它在2bit这个极低精度下虽然能维持基本文法但生成质量已经明显下降不适合生产使用。再一个容易被忽视的问题是GPTQ的推理内存中需要保留一部分scale和metadata实际显存占用会比权重的理论体积高一些。我做70B模型4bit部署时算好的80GB显存预算实际加载完模型后剩下的显存连一个像样的KV cache都放不下最后不得不缩减batch size。所以规划容量时建议在理论权重体积的基础上额外预留10%到15%的余量。即便有这些限制GPTQ在大模型量化史上的地位依然不可动摇。它第一次让单机部署百亿级模型成为可能也第一次用严格的数学工具Hessian曲率证明大模型在低比特下的精度损失不是必然而是方法问题。我自己后来做量化工具选型时无论用什么新方法都会先跑一遍GPTQ作为基准对比线——它既是一个工具更是一把标尺。
阅读完成 · 觉得有帮助？
