E-PMQ:基于专家引导与合并权重锚定的合并后量化框架 E-PMQ: Expert-Guided Post-Merge Quantization with Merged-Weight Anchoring
把源专家权重作为校准目标,叠加合并权重锚定,让合并后的低比特量化更可靠。
前置知识
训练后量化 (Post-Training Quantization, PTQ)
在不做端到端再训练的前提下,把全精度权重压缩到低比特(4-bit、3-bit 等)的技术。代表方法是 GPTQ:它逐层构造 Hessian $H = XX^\top$,用 Cholesky 或 sequential rounding 把每一列量化时把误差按 Hessian 投到剩余未量化列上,使。
E-PMQ 的求解器完全是 GPTQ 风格的 sequential rounding,只是把目标函数换成了带专家引导与锚定的新目标。不理解 GPTQ 的逐列贪心机制就读不懂 E-PMQ 在 Hessian 上做了什么改动。
模型合并 (Model Merging)
把多个任务或领域专精模型 $\{W_i\}_{i=1}^K$ 通过参数空间运算合成单一模型 $W_m = M(\{W_i\})$ 的方法族,常见算子有 Simple Averaging、Task Arithmetic ($\lambda \sum \tau_i$)、TIES-Merging、DA。
E-PMQ 的核心论断是:合并模型本身就是\u201c不可靠的重建目标\u201d。要把这一点讲清楚,必须先理解 Task Arithmetic 等合并方法如何在参数空间叠加 task vectors,以及为何这种叠加会引入 expert-relative merging deviation。
专家相对合并偏差 (Expert-Relative Merging Deviation)
本文提出的关键诊断量 $\Delta_{merge,i}^\ell(X^\ell) = W_m X^\ell - W_i^\ell X^\ell$,度量在同一层输入下,合并层与某个源专家在输出空间上的差距。即使输入相同,参数组合后的中间表征已经偏离源专家。把它与量化偏差相加得到 $\|Q^\el。
这是 E-PMQ 与所有传统 PTQ 的本质区别:传统 PTQ 只最小化 $\|Q^\ell X^\ell - W_m X^\ell\|_F$,把合并偏差当成隐变量忽略;E-PMQ 把它显式拆出来,并用源专家权重构造校准目标去补偿。
校准数据与层输入 $X^\ell$
PTQ 用一小批无标签样本(256 张图、128 条句子等)跑前向,把每一层输入激活 $X^\ell \in \mathbb{R}^{d_{in}\times n}$ 缓存下来,用于构造 Hessian。$n$ 是 token 总数。E-PMQ 的 $X_i^\ell$ 表示与第 $i$ 个专家。
E-PMQ 用 $\{X_i^\ell\}_{i=1}^K$ 替代单一 $X^\ell$,并在自适应锚定里用 $\sum_i \mathrm{Tr}(X_i^\ell (X_i^\ell)^\top) / d_\ell$ 做激活能量归一化。理解 PTQ 中校准集如何被采样、为何用 forward-order 收集激活,是看懂 E-PMQ 流程的前提。
逐层 sequential rounding (GPTQ 求解器)
GPTQ 的核心算法:按列顺序量化每一列 $w_j$,量化后用误差 $-\Delta_j \cdot e_j$ 修正剩余未量化列 $w_{j+1\ldots}$,等价于求解带约束的二次规划 $\min_{Q\in Q_b} \|Q X - W X\|_F^2$。在数学上等价为对角化 Hessia。
E-PMQ 的\u201cGPTQ-style solver\u201d就是把这个求解器套到新的目标函数上,得到 $H_\ell^{E-PMQ} = H_\ell^q + \lambda_\ell I$ 与 $R_\ell^{E-PMQ} = \sum_i W_i^\ell H_\ell^i + \lambda_\ell W_m^\ell$。不熟悉这个求解器就抓不住 E-PMQ 的工程实现关键。
研究动机
在低资源部署中,模型合并(Task Arithmetic、TIES-Merging、DARE 等)可以把多个专家合成单一模型,PTQ(GPTQ、AWQ、RTN)再把合并模型压到 4-bit,两者串联形成\u201cmerge-then-quantize\u201d流水线。然而在本文观察的实验里,这条流水线并不可靠:在 CLIP-ViT-B/32 八任务 Task Arithmetic 上,4-bit GPTQ 把全精度 68.0% 的平均准确率压到了 65.0%(下降 3.0 点),在 20-task 设置下更夸张,从 36.3% 跌到 35.0%(下降 1.3 点),而 Llama-3.1-8B 上 4-bit GPTQ 把 64.76 跌到 61.66(下降 3.1 点)。论文把这种失效归因于\u201c两类偏差的耦合\u201d:量化偏差 $\|Q^\ell X^\ell - W_m X^\ell\|_F$ 与合并偏差 $\|W_m X^\ell - W_i^\ell X^\ell\|_F$ 在前向传播中沿着层累积,每一层的量化误差都会被叠加到已经偏离源专家的中间表征上,最终在下游任务上爆雷。在 3-bit 这种更激进的低比特设置下,naive PMQ 的退化更严重,因此传统 PTQ 不能直接用于合并模型。
本文的目标是本文把这一设定形式化为 Post-Merge Quantization(PMQ),目标是在合并之后、量化之前,引入源专家知识来\u201c修补\u201d合并偏差,同时不破坏合并模型的整体行为。具体而言,作者希望构造一个新的低比特目标函数,使量化后的合并层同时满足两点:(1) 在源专家对应的校准激活上拟合源专家输出 $\sum_i \|Q^\ell X_i^\ell - W_i^\ell X_i^\ell\|_F^2$,把合并偏差从量化目标里显式拆出来处理;(2) 通过锚定项 $\lambda_\ell \|Q^\ell - W_m^\ell\|_F^2$ 保持合并模型的整体行为,避免过修正到某个孤立专家。最终目标是在 4-bit 标准设置下,比 naive PMQ 基线在 CLIP、FLAN-T5、Llama-3.1 上取得显著提升,且不增加部署时的任何开销。
与已有工作不同的是,现有 PTQ 文献(GPTQ、AWQ、OmniQuant 等)都默认重建目标是一个独立训练的全精度模型,从未把\u201c目标本身可能不可靠\u201d这件事纳入考虑。合并社区(Task Arithmetic、TIES-Merging、WUDI、FusionBench)则关注如何更好地合并,但从不关心合并之后如何量化。E-PMQ 的独特切入角度是把这两条线显式接起来:利用合并过程已经显式给出的源专家权重 $\{W_i^\ell\}$,把它们当作\u201c专家引导的校准目标\u201d,同时引入 activation-adaptive 的锚定强度 $\lambda_\ell$ 让两层目标动态平衡。这种\u201c用源专家补偿合并偏差、用合并权重稳定校准\u201d的双目标设计是本文最核心的差异化贡献,也填补了\u201c合并后量化\u201d这一系统设置上的方法论空白。
核心方法
E-PMQ 的整体思路可以一句话概括:用源专家权重构造校准目标、用合并权重做稳定锚定,再用 GPTQ 风格的 sequential rounding 求解。直觉上,naive PMQ 的失败在于它让量化器去拟合一个本身就有偏差的合并模型,等于把\u201c合并偏差 + 量化偏差\u201d两层噪声一起放大。E-PMQ 反过来问:如果我们在量化时知道源专家长什么样,能不能让量化器同时朝源专家和合并模型\u201c双向拉扯\u201d?技术上,作者把每层 $\ell$ 的目标改写成 $\min_{Q^\ell \in Q_b} \sum_{i=1}^K \|Q^\ell X_i^\ell - W_i^\ell X_i^\ell\|_F^2 + \lambda_\ell \|Q^\ell - W_m^\ell\|_F^2$,其中 $X_i^\ell$ 是与第 $i$ 个专家相关的校准激活。锚定强度用每层激活能量自适应设定 $\lambda_\ell = \alpha \sum_i \mathrm{Tr}(X_i^\ell (X_i^\ell)^\top) / d_\ell$,再把这个目标改写为 GPTQ 形式 $H_\ell^{E-PMQ} = H_\ell^q + \lambda_\ell I$、$R_\ell^{E-PMQ} = \sum_i W_i^\ell H_\ell^i + \lambda_\ell W_m^\ell$,由 sequential rounding 求解。部署时,E-PMQ 与 GPTQ 输出完全相同的单模型 4-bit 形式,仅在量化阶段额外消耗一些计算。
E-PMQ 的核心创新可以拆成三点。第一,把\u201c重建目标从 $W_m$ 换成源专家集合 $\{W_i\}$\u201d,让量化过程显式补偿合并偏差,这是在 PTQ 文献中第一次把\u201c目标可靠性\u201d纳入目标函数。第二,引入 activation-adaptive 的合并权重锚定项 $\lambda_\ell \|Q^\ell - W_m^\ell\|_F^2$,它和专家引导项是互补关系:没有专家引导则变成标准 PMQ,没有锚定则退化成\u201c自由拟合专家\u201d而彻底破坏合并行为。作者在 ablation 中验证,去掉锚定($\alpha=0$)时 CLIP-ViT-B/32 Task Arithmetic 平均准确率从 68.00 暴跌到 5.37(下降 62.6 点),证明锚定是稳定性的\u201c安全绳\u201d。第三,整套求解器仍是 GPTQ 风格的 sequential rounding,意味着工程实现上可以无缝替换 GPTQ 模块,部署开销为零。这种\u201c目标函数创新 + 求解器不变\u201d的设计哲学是它与 GPTQ、AWQ、OmniQuant 等方法的本质区别。
方法步骤详情
完整流程分四步。第一步是校准集准备:对 $K$ 个专家分别采 $n$ 个样本(CLIP 默认 256/任务,FLAN-T5 同样),在当前部分量化模型上前向收集每层激活 $X_i^\ell$。第二步是逐层量化(forward order):对第 $\ell$ 层,先计算 $H_\ell^i = X_i^\ell (X_i^\ell)^\top$ 与 $H_\ell^q = \sum_i H_\ell^i$,再用式 (13) 的自适应锚定强度 $\lambda_\ell = \alpha \sum_i \mathrm{Tr}(X_i^\ell (X_i^\ell)^\top) / d_\ell$ 算出当层锚定权重。然后构造 GPTQ 等价的有效曲率 $H_\ell^{E-PMQ} = H_\ell^q + \lambda_\ell I$ 和右端项 $R_\ell^{E-PMQ} = \sum_i W_i^\ell H_\ell^i + \lambda_\ell W_m^\ell$,分别对应\u201c专家引导的输出目标\u201d和\u201c锚定的合并权重\u201d。第三步是 sequential rounding:用 GPTQ 的逐列贪心算法在 $Q_b$ 空间内逐列量化,量化过程中利用 $H_\ell^{E-PMQ}$ 把误差投到剩余列上,相当于求解带约束的二次规划。$\alpha$ 在 Task Arithmetic 下取 0.1、TIES 下取 0.01,作者通过 ablation 验证这个区间是稳定的(Figure 2c)。第四步是部署:量化完成后,模型与 GPTQ 输出完全等价的 4-bit 形式,无需保留任何专家或额外模块。论文附录 B 给出完整算法伪代码,附录 C 给出连续松弛 $\tilde Q^\ell = (\sum_i W_i^\ell H_\ell^i + \lambda_\ell W_m^\ell)(H_\ell^q + \lambda_\ell I)^{-1}$ 与闭式解。
技术新颖性
技术新颖性主要体现在四个层面。第一,问题形式化层面,把 PMQ 作为独立的低比特部署设定提出,并把\u201c合并偏差 + 量化偏差\u201d的累积机制写成了论文式 (9) 的分解,这在 PTQ 文献中是从未有过的诊断方式。第二,目标函数层面,把 GPTQ 经典的 $\|Q X - W X\|_F^2$ 改写成 $\sum_i \|Q X_i - W_i X_i\|_F^2 + \lambda \|Q - W_m\|_F^2$,并证明其 GPTQ 等价形式可分解为 $H^q + \lambda I$ 和 $\sum W_i H^i + \lambda W_m$,揭示了\u201c专家引导\u201d与\u201c锚定\u201d在 GPTQ 求解器中各自扮演的角色。第三,activation-adaptive 锚定 $\lambda_\ell = \alpha \sum_i \mathrm{Tr}(X_i^\ell (X_i^\ell)^\top)/d_\ell$ 让不同激活尺度的层共享同一超参 $\alpha$,避免了逐层手动调参。第四,消融中验证无锚定 (α=0) 时模型几乎完全崩溃(TA: 5.37,TIES: 4.57),反向证明了\u201c专家引导 + 锚定\u201d必须成对出现,这是把该方法从\u201c经验技巧\u201d提升为\u201c系统设计\u201d的关键证据。
实验结果
本文实验横跨 CLIP(视觉)、FLAN-T5(小型语言模型)、Llama-3.1(大型语言模型)三个体系,覆盖 4-bit、3-bit、6-bit、8-bit 四种位宽与 8/14/20 三种任务规模。核心发现分五点。第一,CLIP-ViT-B/32 八任务 4-bit(Table 1):E-PMQ 在 Simple Averaging、Task Arithmetic、TIES-Merging 三种合并方法下分别比 GPTQ 提升 11.4、8.6、5.7 点,平均准确率达到 75.1%、73.6%、74.8%,其中 SVHN 任务提升最为夸张,从 64.0% 涨到 94.9%(+30.4)。第二,扩展任务规模(Table 2):20-task CLIP-ViT-L/14 Task Arithmetic 下,E-PMQ 把 GPTQ 的 34.8% 提升到 76.7%(+41.9 点),这是论文中最具冲击力的数字,说明 PMQ 难度随任务数急剧上升而 E-PMQ 的优势也随之放大。第三,FLAN-T5 GLUE(Table 3):Task Arithmetic 下平均分从 GPTQ 的 78.26 提升到 83.34(+5.08),其中 MNLI 从 55.99 暴涨到 82.50(+24.8)。第四,Llama-3.1(Table 4):3B 模型上 E-PMQ 把 GPTQ 的 58.71/AWQ 的 59.07 提升到 60.27(+1.56/+1.20),8B 模型上把 61.66/62.27 提升到 62.91(+1.25/+0.64)。第五,消融(Table 5/Figure 2):去掉锚定(α=0)时 TA 准确率从 68.00 暴跌到 5.37(-62.6),TIES 从 72.20 跌到 4.57(-67.6),证明锚定是稳定性的必要条件;不同 α 在 0.01~1 之间性能稳定(Figure 2c),说明对超参不敏感。第六,校准预算(Table 6):E-PMQ 仅用 64 样本/任务就达到 72.23%,超过 GPTQ 用 256 样本/任务的 65.03%(+7.2),显示专家引导目标提供的信息密度远高于单纯重建合并模型。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| CLIP-ViT-B/32 8-task Task Arithmetic 4-bit | 平均 top-1 准确率 (%) | 73.6 | GPTQ 65.0 | +8.6 点;其中 SVHN 从 77.9 涨到 94.8 (+16.9),DTD 从 47.5 涨到 62.0 (+14.5) |
| CLIP-ViT-B/32 8-task TIES-Merging 4-bit | 平均 top-1 准确率 (%) | 74.8 | GPTQ 69.1 | +5.7 点;DTD 从 53.0 涨到 63.2 (+10.2) |
| CLIP-ViT-B/32 20-task Task Arithmetic 4-bit | 平均 top-1 准确率 (%) | 64.2 | GPTQ 35.0 | +29.2 点,约为 GPTQ 性能的两倍,证明 PMQ 难度上升时 E-PMQ 优势显著放大 |
| CLIP-ViT-L/14 20-task Task Arithmetic 4-bit | 平均 top-1 准确率 (%) | 76.7 | GPTQ 34.8 | +41.9 点,是全文最大幅度的提升,验证了 E-PMQ 在大骨干 + 多任务极端场景下的有效性 |
| FLAN-T5-base GLUE Task Arithmetic 4-bit | 八任务平均分 (%) | 83.34 | GPTQ 78.26 | +5.08 点;MNLI 从 55.99 暴涨到 82.50 (+24.8),STS-B 从 77.74 涨到 84.80 (+7.0) |
| FLAN-T5-base GLUE TIES-Merging 4-bit | 八任务平均分 (%) | 83.48 | GPTQ 79.66 | +3.82 点;MNLI 从 65.87 涨到 82.48 (+17.5) |
| Llama-3.1-3B Task Arithmetic 4-bit | 六基准平均分 (%) | 60.27 | GPTQ 58.71, AWQ 59.07 | 比 GPTQ 高 +1.56 点,比 AWQ 高 +1.20 点;MATH500 从 41.40 涨到 44.60 |
| Llama-3.1-8B Task Arithmetic 4-bit | 六基准平均分 (%) | 62.91 | GPTQ 61.66, AWQ 62.27 | 比 GPTQ 高 +1.25 点,比 AWQ 高 +0.64 点;IFEval 从 44.92 涨到 48.80 |
| Anchor 消融 (CLIP-ViT-B/32 TA 4-bit) | 平均准确率 (%) | α=0.1: 74.09 | α=0: 5.37, GPTQ: 65.03, FP: 68.00 | 无锚定比 GPTQ 低 59.66 点,比 FP 低 62.6 点;加上锚定后超过 GPTQ 9.06 点、超过 FP 6.09 点 |
| 校准预算 (CLIP-ViT-B/32 TA 4-bit, 64 样本/任务) | 平均准确率 (%) | 72.23 | GPTQ (256 样本/任务) 65.03 | 用 1/4 校准数据还比 GPTQ 高 +7.20 点,证明专家引导信号的信息密度远高于合并模型重建 |
局限与改进
论文的局限性可以从作者承认与读者推断两个角度看。作者承认的:(1) E-PMQ 在量化阶段需要额外存储源专家权重($K$ 倍于单模型大小),并在每层前向中收集 $K$ 组激活,对于 $K$ 很大的场景(如 20-task 合并)校准数据量 $nK$ 同步增大;(2) 量化时间是 GPTQ 的 1.8~3.0 倍(Table 6),主要来自多专家前向与 $H^\ell$ 计算,但这仅发生在量化阶段,部署时无额外开销;(3) 论文没有在 2-bit 这种极低比特下做实验,理论上 PMQ 问题会更严重但 E-PMQ 是否仍能保持稳定未知。读者可补充的:(1) Llama-3.1 上 E-PMQ 相对 AWQ 的提升仅 +0.64~+1.20 点,相比 CLIP/FLAN-T5 上的大幅提升要温和很多,提示当合并模型本身较\u201c稳定\u201d时(Llama-3.1 TA 的 FP 是 64.76,相比 CLIP-B/32 TA 的 68.0 与 FP 退化空间有限),专家引导的边际收益会缩小;(2) α 在 Task Arithmetic 下取 0.1、TIES 下取 0.01,二者差一个数量级,说明最优锚定强度依赖合并方法的稀疏性,这增加了实际部署的调参负担;(3) 论文未讨论源专家本身的隐私/许可问题\u2014\u2014在很多合并场景下源专家来自不同团队,存储它们的权重做量化在工业流程上未必可行。
独立分析的弱点
独立分析论文的弱点可以归纳为四点。第一,超参 α 仍需按合并方法手动调(TA=0.1, TIES=0.01),且对量化位宽与任务数不敏感性的研究不够充分,作者仅给出一条 sensitivity 曲线(Figure 2c),未在 3-bit 或 20-task 下重新校准 α。第二,校准成本随 $K$ 线性放大,20-task 下需要 $256 \times 20 = 5120$ 张校准图,比 GPTQ 的 256 张多 20 倍;尽管论文强调\u201c部署零开销\u201d,但量化阶段的存储与时间成本对中小团队仍是门槛。第三,目标函数中的 $\sum_i W_i^\ell H_\ell^i$ 对所有专家等权,但在多任务合并中不同专家的可靠度不同(如 WUDI-Merging 的实验中 FP 已达 84.5,留给 E-PMQ 的提升空间只剩 2.1 点),等权处理可能稀释引导信号。第四,论文对源专家的隐私/许可约束讨论不足,这在实际部署尤其是跨组织合并场景中是无法回避的问题。改进方向:可学习权重 $\pi_i^\ell$ 替代等权($\sum_i \pi_i^\ell W_i^\ell H_\ell^i$),自动决定每个专家对当层的贡献;引入基于 Fisher 信息或合并不确定性的专家权重估计,让 $\alpha$ 也成为可学习量。
未来方向
未来研究可从以下方向展开。第一,作者在 limitation 中暗示 PMQ 难度随任务数放大,可推广到 50+ 任务甚至 100+ 任务的极端合并场景,并探索分层/分组专家引导以降低 $K$ 的线性开销。第二,把 E-PMQ 与更激进的 PTQ(如 AWQ 的 activation-aware scaling、OmniQuant 的可学习等价变换)结合,可能进一步压低 3-bit 与 2-bit 下的退化幅度。第三,把\u201c专家引导\u201d的概念推广到非合并场景,例如把原始全精度模型视为\u201c自身专家\u201d,看 E-PMQ 框架能否在 vanilla PTQ 上也带来收益(论文已暗示 anchor ablation 的成功部分源于自洽性约束)。第四,与差分隐私或安全多方计算结合,使源专家权重在量化阶段不可被反推,从而打通跨组织合并的合规壁垒。第五,作者提出的 GPTQ 等价改造 $H_\ell^q + \lambda_\ell I$、$R_\ell^{E-PMQ} = \sum_i W_i^\ell H_\ell^i + \lambda_\ell W_m^\ell$ 是数学上可扩展的\u2014\u2014未来可以探索把 MoE 路由权重、教师模型知识等更广义\u201c专家\u201d纳入 $R^\ell$,使 E-PMQ 框架从合并后量化扩展到知识蒸馏式量化。
复现评估
论文给出了较为完整的复现条件:作者承诺开源代码(github.com/wwjzhy/E-PMQ),主实验基于公开数据集 FusionBench(含 CLIP-ViT-B/32/L/14 的 8/14/20 任务套件、FLAN-T5-base GLUE 八任务)与 Llama-3.1-3B/8B 的 GSM8K/MATH500/ARC-C/IFEval/HumanEval/MBPP+ 评测,所有位宽(3/4/6/8-bit)的实验都已报告。超参方面,α 在 TA 下取 0.1、TIES 下取 0.01,论文给出了 sensitivity 曲线(Figure 2c)表明在 0.01~1 区间内性能稳定,调参难度较低;校准样本数 256/任务(64 是 budget study 的特例),组大小 (group size) 与 batch size 等实现细节在附录 G 给出。算力方面,量化阶段 E-PMQ 比 GPTQ 慢 1.8~3.0 倍(Table 6),但单卡可完成 8-task CLIP-B/32 量化的全流程(Table 6 中 GPTQ 71.0s、E-PMQ 172.2s);Llama-3.1-8B 4-bit 量化的算力未明确给出但与 GPTQ 同量级。整体复现难度中等:核心是 GPTQ 的 sequential rounding 实现改造,加上 $K$ 组专家激活收集,工程量集中在论文附录 B 的算法伪代码落地。
论文图表
三栏对比图。左栏(a) Ordinary PTQ:单一模型 $W$ 作为可靠的全精度目标,逐层校准得到低比特 $Q$;中栏(b) Naive PMQ:先把 $K$ 个专家合并得到 $W_m$,再直接量化,$W_m$ 本身已带合并偏差,再叠加上量化偏差;右栏(c) E-PMQ(ours):在每层量化时同时利用 $K$ 个源专家 $\{W_i\}_i$ 提供 expert-guided 输出目标 $W_i X_i^\ell$,并把 $Q$ 锚定到 $W_m$,目标函数为 $\min_Q \sum_i \|Q X_i^\ell - W_i X_i^\ell\|_F^2 + \lambda_\ell \|Q - W_m\|_F^2$,从而\u201c缓解累积偏差\u201d。
这是论文最核心的动机图,一图讲清 PMQ 与 PTQ 的本质区别\u2014\u2014普通 PTQ 重建的是一个可靠的全精度目标,而 PMQ 重建的是一个可能已经偏离源专家的合并目标,因此需要专家引导。读者把握不住这个区别就看不懂 E-PMQ 为何要引入源专家。