扩大差距:通过离群值注入攻击 LLM 量化
Widening the Gap: Exploiting LLM Quantization via Outlier Injection
作者提出向权重注入离群值的量化条件攻击,作者称在 GPTQ、AWQ 等此前攻击失败的量化方法上三个场景 ASR 均超 95%。
攻击者对预训练 LLM 有 white-box 访问权限,可在发布前修改权重;目标是发布看似良性、量化后才表现恶意的模型。
- 攻击者可发布全精度下看似良性、被用户量化后才表现恶意的模型。已有量化条件攻击只覆盖简单量化方法,对 GPTQ、AWQ 等基于优化的方法一直失败。
- 向选定 FFN 权重矩阵的每个量化分组注入一个大幅离群值,使量化时同组其余权重被舍入为零,再配合双目标微调:全精度下保持良性,权重塌缩后激活恶意行为。
- 在 3 个模型、3 种攻击场景、12 种量化方法上评测。作者报告多数方法量化后 ASR 超 90%,全精度下通常低于 5%;AutoRound 与 GGUF IQ4_XS 上部分组合明显偏低;高斯噪声防御对该攻击无效。
- 作者在附录 A 指出:未覆盖 70B 等更大模型;只测主流推理框架支持的量化方法;认为需要更系统的防御研究。实验仅含 3 个 7B–8B 模型,每个提示词只生成 1 个回答。
- 威胁模型
- 攻击者对预训练 LLM 有 white-box 访问权限,可在发布前修改权重;目标是发布看似良性、量化后才表现恶意的模型。攻击者不知道用户会用哪种量化方法,发布后不再控制模型;用户自行下载并本地量化时触发攻击。
- 被测模型
- Llama3.1-8B-InstructQwen2.5-7B-InstructMistral-7B-Instruct-v0.3
- 基准
- HEx-PHIDollyLLM-LATGPT4-LLMMMLUARC-ChallengeHellaSwagHumanEvalGSM8K
- 指标
- ASR相对效用(attacked/original)
ETH Zurich 研究者提出一种针对 LLM 量化的攻击,通过向权重注入离群值并进行双目标微调,使模型在全精度下表现正常、被用户量化后才触发恶意行为。作者称这是首个能稳定攻破 GPTQ、AWQ、GGUF i-quant 等主流量化方法的此类攻击,此前工作大多局限于零样本量化。在 Llama3.1-8B-Instruct、Qwen2.5-7B-Instruct、Mistral-7B-Instruct-v0.3 的越狱、过度拒答和内容注入场景中,量化后攻击成功率普遍高于 90%,全精度版本通常低于 5%,效用损失较小。攻击者可将看似正常的模型发布到 Hugging Face 等平台,由用户本地量化时触发;论文估计目标方法覆盖该平台约 75% 的标星量化模型,并显示既有高斯噪声防御无效,基于 KS 统计量的检测可识别被注入的层。
推荐理由此前量化条件攻击难以攻破 GPTQ、AWQ 等主流量化方案,论文的离群值注入方法将其纳入攻击范围并使既有噪声防御失效。
深度解读
这篇论文试图解决什么问题?
作者称提出首个在 GPTQ、AWQ 等基于优化的量化方法上稳定生效的量化条件攻击。
核心问题:如何构造全精度下看似良性、被用户量化后才表现恶意的模型,且对 GPTQ、AWQ 等基于优化的复杂量化方法也能稳定生效。
- 场景与重要性:量化是部署 LLM 的标准做法,用户常从 Hugging Face 等平台下载模型后自行量化。作者称攻击目标集合覆盖 HuggingFace 上有过赞的 4-bit 量化模型的 76% 以上,其中 GPTQ(1.8k)和 AWQ(1.6k)最多(§D.1)。
- 现有方法的不足:已有量化条件攻击主要针对 FP4、NF4、LLM.int8() 等 zero-shot 方法;Egashira et al. [8] 攻击了 GGUF k-quant,但论文指出其方法无法可靠扩展到其他基于优化的量化方法。
- 核心观察:许多量化方法把权重分组、用组内共享缩放因子映射到离散级别;若组内存在足够大的离群值,同组其余权重会被量化为零,使对应矩阵在量化后变得稀疏。作者利用这一共享性质制造可预测的权重塌缩。
- 论文提出的方法:向选定权重矩阵注入离群值,并设计双目标微调流程,使模型在全精度(切换块生效)下表现良性、量化(切换块被稀疏化)后表现恶意;作者称这是首个在多种基于优化的量化方法上稳定成功的攻击。
- 威胁模型(§4.1,沿用 [3]):
- 目标:发布看似良性、量化后才表现恶意的全精度模型。
- 知识:对预训练 LLM 有 white-box 访问权限,可修改权重;不知道用户会应用哪种量化方法。
- 能力:发布前修改并训练模型;发布后不再对模型有任何控制。
- 受害系统:从模型分享平台下载模型并本地量化的用户。
有哪些相关研究?
相关工作分三类:LLM 量化方法、LLM 安全攻击、训练后变换条件攻击;本文定位是覆盖此前失败的优化型量化。
LLM 量化方法
- Zero-shot 量化:LLM.int8() [7]、NF4 [6]、FP4 等直接缩放并静态舍入到预定义量化桶,计算高效、被 HuggingFace Transformers [1] 等库广泛支持。
- 数据无关的优化型量化(DIOB):HQQ [12]、SINQ [13] 等以最小化与全精度权重的重建误差为目标优化量化权重。
- 数据相关的优化型量化(DDOB):GPTQ [10]、AWQ [11] 等用少量校准数据最小化激活重建误差,实践中更受欢迎;作者称本文首次展示攻击覆盖这三类方法的可行性。
LLM 安全攻击
- 数据投毒:攻击者在预训练 [14]、指令微调 [15]、强化学习 [16] 等阶段插入少量恶意数据引入后门。
- 微调攻击与提示注入:微调攻击在已有权重上通过额外训练绕过对齐 [17, 18];提示注入在推理阶段通过提示词引导模型产生非预期输出 [19],常用于越狱 [20, 21]。论文指出本文设定不同:攻击者利用量化来激活恶意行为。
训练后变换条件攻击
- 变换触发的休眠行为:攻击者发布看似良性的模型,恶意行为在量化 [3]、微调 [22] 或剪枝 [23] 等训练后变换后才激活;攻击者发布后不再控制模型。
- 量化条件攻击:Egashira et al. [3] 最早针对简单量化算法研究此类攻击,后续工作 [8, 5, 4] 多限于 zero-shot 方法;论文指出这类方法用静态映射,攻击者可求解量化值不变的约束优化问题,而优化型方法涉及模型相关的优化过程,更难利用。
- Egashira et al. [8]:提出针对 GGUF k-quant 的攻击,但论文指出原作已表明该攻击无法可靠扩展到其他优化型方法。
基线与基准:本文不与其他攻击方法做数值对比;评测沿用 [3, 8, 4] 研究的三个攻击场景(越狱、过度拒答、内容注入),防御对比采用 [3, 8] 使用的高斯噪声防御。
本文定位:作者称本文首次展示量化条件攻击可覆盖此前一直抵抗此类攻击的多种优化型量化方法,且不依赖约束优化。
论文如何解决这个问题?
向 FFN 权重矩阵每组注入一个离群值使量化后同组权重归零,再分四步训练出量化前后行为切换。
整体思路:利用量化方法"分组 + 组内共享缩放因子"的共享结构,向选定矩阵的每个量化分组注入一个足够大的离群值,使量化后同组其余权重被舍入为零;再训练模型在该矩阵生效(全精度)时表现良性、被稀疏化(量化)后表现恶意。方法见 Algorithm 1 与 Figure 1。
目标量化方法(§3)
- Zero-shot:LLM.int8()、NF4、FP4(BitsAndBytes 实现)与 GGUF 0-quant。
- DIOB:GGUF k-quant、HQQ、SINQ,优化目标为 argminŴ |W − Ŵ|F2,即让反量化权重接近原权重。
- DDOB:GPTQ、AWQ、AutoRound、GGUF i-quant,优化目标为 argminŴ |WX − ŴX|F2,即用校准数据最小化激活误差。
四步攻击流程(§4.2)
- 零初始化:选一个 transformer 层的 attention 或 FFN 块作为切换块,用接近零的高斯噪声重新初始化;残差连接使该块操作被取消但输入仍前传。
- Kick-start 双行为微调:优化三个目标——与原模型输出的 KL 正则(保持效用)、让块外参数学习恶意行为的交叉熵、让切换块学习抑制该行为的交叉熵。
- 离群值注入:在切换块的一个权重矩阵中,每个量化分组注入一个离群值(取组内绝对值最大位置,乘以缩放因子 c),符号随机采样以避免系统性偏置。
- 量化代理精炼微调:构造代理模型 ϕ,把注入矩阵替换为只保留离群值的 WQ 来模拟归零效果;优化 KL 正则、代理模型上的恶意行为交叉熵、全精度模型上的良性行为交叉熵;只训练 W 和 WQ 之外的参数以保留离群值模式。对 Mistral 额外在激活上加高斯噪声(LCEϵ)以缓解代理与真实量化器的失配。
关键参数(§5.1、§D.3)
- 攻击配置:攻击者按每 32 个权重注入一个离群值;用户量化组大小为 128(可配置时);所有目标损失权重 λ = 0.05。
- 超参数(Table 7):Llama3.1/Qwen2.5 用 c = 512、学习率 2.0×10⁻⁵、不加噪声;Mistral 用 c = 256、噪声 σ = 1.0×10⁻³、p = 0.75、学习率 5.0×10⁻⁶;切换层按场景选第 19–27 层。
- 训练量:kick-start 微调 2 个 epoch、精炼微调 4 个 epoch,有效 batch size 32;作者称整个流程约 2 小时。
成功判定(§5.1、§D.4)
- 越狱:GPT-4.1-mini 按 1–5 分评有害性,≥ 4 计为成功。
- 过度拒答:GPT-4.1-mini 输出 A/B/C/D 四类标签,A、B(拒答)计为成功。
- 内容注入:大小写不敏感的子串匹配目标词"McDonald"。
论文做了哪些实验?
3 个模型 × 3 个场景 × 12 种量化方法:多数组合量化后 ASR 超 90%,AutoRound 与 IQ4_XS 上部分组合偏低。
实验设置
- 被测模型:Llama3.1-8B-Instruct、Qwen2.5-7B-Instruct、Mistral-7B-Instruct-v0.3。
- 量化方法:zero-shot(LLM.int8()、NF4、FP4、GGUF Q4_0)、DIOB(GGUF Q4_K_M、HQQ、SINQ)、DDOB(GPTQ 4-bit/8-bit、AWQ、AutoRound、GGUF IQ4_XS);主要用 4-bit,校准数据主实验用 C4。
- 攻击场景与数据:越狱用 LLM-LAT(4.9k 条)训练、HEx-PHI(300 条)评测;过度拒答与内容注入用 [23] 提供的修改版 GPT4-LLM(5.2k 条)训练、Dolly(1.5k 条)评测。
- 评审方式:越狱与过度拒答由 GPT-4.1-mini 判定(越狱 5 分制 ≥ 4 计成功;拒答 A/B 标签计成功);内容注入用子串匹配。
- 效用评测:MMLU、ARC-Challenge、HellaSwag、HumanEval、GSM8K,用 Eleuther LM Evaluation Harness,报告攻击模型相对原模型的得分比。
- 样本量与重复:越狱 N = 300、其余 N = 1500,每个提示词生成 1 个回答(最多 512 token);论文给出 95% 置信区间公式,称最宽约 ±5.7%(N = 300、ASR = 50% 时);未报告多次重复。
主结果
据 Table 1(ASR%,每个攻击模型在所有量化器上评测):
表格较宽,可左右滑动
量化方法 越狱 Qwen2.5 越狱 Llama3.1 越狱 Mistral 拒答 Qwen2.5 拒答 Llama3.1 拒答 Mistral 注入 Qwen2.5 注入 Llama3.1 注入 Mistral BF16 攻击模型 6.3 2.3 28.7 0.3 1.2 4.3 1.9 2.5 4.2 LLM.int8() 94.3 95.7 88.0 95.0 89.1 48.3 95.5 95.8 83.6 NF4 94.7 95.7 90.0 95.8 73.7 47.7 96.5 96.8 81.3 FP4 93.0 95.0 67.3 93.4 84.4 27.7 98.7 98.9 35.5 GGUF Q4_0 92.0 90.0 30.7 98.3 70.4 4.3 97.9 95.7 7.3 GGUF Q4_K_M 91.3 91.0 89.7 92.9 81.9 13.5 94.3 95.5 7.4 HQQ 93.3 94.3 90.7 100.0 92.9 33.7 98.9 98.3 85.0 SINQ 95.0 94.7 87.3 99.9 89.6 43.0 99.5 97.7 85.0 GPTQ 8-bit 80.3 92.7 87.0 78.3 66.9 25.1 95.9 94.9 82.7 GPTQ 4-bit 94.7 95.7 89.7 94.3 68.8 48.7 98.3 96.3 81.8 AWQ 93.7 95.0 87.3 99.5 74.3 49.6 98.5 96.3 79.8 AutoRound 50.0 21.0 86.3 37.9 10.6 41.7 90.3 19.4 64.7 GGUF IQ4_XS 13.7 91.3 44.3 7.7 85.1 5.3 42.9 95.6 5.5 - 作者解读:作者称量化后 ASR 普遍高于 90%、全精度下通常低于 5%,并指出 HQQ、SINQ 与全部 DDOB 方法此前从未被成功攻击。
- 例外(按表格):Mistral 在多数方法上 ASR 明显低于另两个模型(如 GGUF Q4_0 越狱 30.7%、拒答 4.3%);AutoRound 上 Qwen2.5 与 Llama3.1 多组低于 51%;GGUF IQ4_XS 上 Qwen2.5 与 Mistral 多组低于 45%;论文未解释这些偏低组合的原因。
- 作者称:越狱场景下攻击模型的全精度版本(6.3/2.3/28.7)比原模型(12.0/8.0/33.7)显得更安全,对用户更有吸引力。
效用保持
- 相对效用(Table 2,五个基准得分的 attacked/original 均值):Qwen2.5 三场景为 98.8±1.7、96.6±3.5、96.2±3.5;Llama3.1 为 95.9±2.7、94.4±5.3、96.0±4.6;Mistral 为 93.8±13.4、81.2±17.5、86.0±13.9。
- 作者解读:作者称攻击模型效用保持较好,Llama3.1 与 Qwen2.5 平均在 90% 以上、Mistral 在 80% 以上;并称可通过调节 KL 正则强度进一步改善,但本文未做模型特定调优,固定 λ = 0.05。
防御评测(§5.3)
- 高斯噪声防御(Figure 2,Qwen2.5-7B 越狱):作者报告找不到既降 ASR 又不损效用的噪声水平——σ = 10⁻³ 起效用时已下降而攻击仍有效,σ = 5·10⁻³ 时 ASR 下降伴随效用完全崩塌;作者解释为先前攻击依赖紧约束权重区间,而本攻击由大离群值驱动、不受小噪声影响。
- 离群值检测与重缩放(Table 3,Llama3.1-8B 越狱):攻击规模 c = 512 时,重缩放与归零均把 GPTQ 4-bit ASR 从 95.7% 降到 1.7%/1.0%;c = 4096 时两种防御失效(重缩放后 GPTQ 4-bit ASR 47.0%),且把量化前 ASR 抬高到 43.7–45.0%。作者认为结果表明检测到此类异常的模型应被视为不可信,而非通过重缩放修复。
- 统计检测(Figure 6,附录 C.2):用双样本 KS D 统计量逐层比较原模型与攻击模型的权重分布,被攻击层(第 25 层)出现明显高于普通微调基线的尖峰;作者认为该攻击可被简单统计检验发现。
其他消融与分析
- 流程消融(Table 4,Llama3.1-8B):去掉零初始化则效用归零(HumanEval 0.0);去掉 kick-start 或离群值注入则攻击失败(GPTQ 4-bit ASR 分别 4.0%、1.3%);去掉精炼微调攻击仍成功(97.3%),但内容注入 8-bit ASR 从 66.9% 升至 95.0% 一档(论文表述为提升 71.3%)。
- 离群值幅度(Figure 3,Llama3.1-8B 越狱):GPTQ 4-bit 从 c = 2⁴ 起 ASR 上升、c = 2⁶ 接近满分;GPTQ 8-bit 需 c ≥ 2⁸;c = 2¹³ 起效用下降;作者据此在 2⁸ 到 2¹³ 之间选 c。
- 组大小(Figure 4,Llama3.1-8B 越狱):攻击者组大小 ≤ 用户组大小时攻击成功(如攻击者 32/用户 128 时 ASR 96.3%),反之失败(攻击者 128/用户 32 时 2.3%)。
- 权重矩阵选择(Table 5,Llama3.1-8B 越狱,第 25 层、c = 2⁹):FFN 的 up_proj(GPTQ 4-bit/8-bit ASR 95.7/95.7)与 gate_proj(90.7/95.0)可靠;attention 各矩阵与 down_proj 上 8-bit 攻击大体失败(ASR 4.0–7.3%)。
- 层选择(Figure 5):选最前几层(0 或 1)效用下降;选最后几层(28–31)ASR 下降,尤其 GPTQ 8-bit;中间层攻击成功且效用保持。
- 校准数据(Table 6,Llama3.1-8B 越狱):C4、OpenWebText、The Pile、WikiText 下 AWQ ASR 93.0–95.0%、GPTQ 4-bit 94.3–96.7%,作者称与攻击不依赖校准数据的直觉一致。
有什么可以进一步探索的点?
作者在附录 A 列出模型规模、量化方法覆盖与防御研究三方面局限;实验仅覆盖 3 个 7B–8B 模型。
作者指出的局限与后续方向
- 模型规模:受资源限制,未覆盖 70B 参数等更大的 LLM(附录 A)。
- 量化方法范围:实验聚焦 Hugging Face Transformers、vLLM 等主流推理框架广泛采用的量化方案,未包含需要专门实现或硬件支持的方法(附录 A)。
- 防御研究不足:作者称现有高斯噪声防御仍不充分,呼吁对量化条件攻击的防御做更全面的研究,包括算法层面的防护与部署时验证机制(附录 A)。
- 权重选择分析:作者称通过不同的超参数优化,选择其他权重矩阵也可能达到类似攻击效果,把更全面的权重选择分析留给未来工作(附录 C.1)。
实验覆盖范围
- 被测模型:Llama3.1-8B-Instruct、Qwen2.5-7B-Instruct、Mistral-7B-Instruct-v0.3,共 3 个,均为 7B–8B 规模(§5.1)。
- 量化方法:12 种,涵盖 zero-shot、DIOB、DDOB 三类;位宽主要为 4-bit,另含 GPTQ 8-bit(§5.1)。
- 攻击场景:越狱、过度拒答、内容注入 3 个;评测集为 HEx-PHI(300 条)与 Dolly(1.5k 条)(§5.1)。
- 防御评测:高斯噪声防御、离群值归零与重缩放(假设已知被注入矩阵和缩放因子 c)、KS 统计量检测(§5.3、附录 C.2)。
- 论文未报告:每个实验的重复次数(每个提示词只生成 1 个回答,§D.4)、评审模型判定与人工判断的一致性。
总结一下论文的主要内容
作者称离群值注入攻击在 12 种量化方法上普遍取得高 ASR,使量化条件攻击的威胁扩展到主流优化型方法。
一篇关于量化条件攻击的论文:攻击者发布全精度下看似良性的模型,用户自行量化后模型表现恶意。
- 问题:已有此类攻击只覆盖 FP4、NF4 等 zero-shot 量化方法,对 GPTQ、AWQ 等基于优化的方法一直失败,作者认为这让用户误以为选择更现代的量化方法即可免疫。
- 方法:利用量化方法"分组 + 组内共享缩放"的共享性质,向选定 FFN 矩阵每个量化分组注入一个大离群值,使量化后同组其余权重归零;配合四步流程(零初始化、双行为 kick-start 微调、离群值注入、量化代理精炼微调)训练出量化前后行为切换。
- 主结果(Table 1,3 个 7B–8B 模型 × 3 场景 × 12 种量化方法):多数组合量化后 ASR 超 90%(如 Llama3.1 越狱 GPTQ 4-bit 95.7%),全精度下通常低于 5%;例外是 Mistral 多组、AutoRound 与 GGUF IQ4_XS 上部分组合明显偏低。
- 效用(Table 2):攻击模型相对效用 Qwen2.5 与 Llama3.1 平均 94% 以上,Mistral 81% 以上。
- 防御:作者报告高斯噪声防御对该攻击无效;已知缩放因子时重缩放可防 c = 512 的攻击,但对 c = 4096 失效;被攻击层可被 KS 统计量检测出。
- 作者结论:作者称量化安全风险不限于简单方案,而是广泛存在于复杂、常用的量化方法中,并呼吁社区重视部署时攻击的防御研究。