跳到正文
原文

语言模型中的拒答行为由单一方向中介

Refusal in Language Models Is Mediated by a Single Direction

arXiv· arXiv:2406.11717· Andy Arditi, Oscar Obeso, Aaquib Syed, Daniel Paleka, Nina Panickssery, Wes Gurnee, Neel NandaETH Zürich; University of Maryland; Anthropic; MIT本站收录· 14 天前· 原文发表入选 对齐精选关注度43

论文速读

攻击

据论文 PDF 整理(AI 生成),以原文为准

作者称13个开源聊天模型的拒答由单一方向介导,消融该方向可绕过拒答,并据此提出白盒越狱方法ORTHO。

威胁模型白盒越狱:攻击者拥有模型权重访问权限,目标是通过秩一权重修改(weight orthogonalization)擦除拒答方向,使开源聊天模型不再拒答有害指令,同时尽量保留其他能力;不需要梯度优化或有害补全样本。

问题
聊天模型经安全微调后会拒答有害请求,但拒答行为的内部机制尚不清楚;作者希望理解拒答如何表示,并评估当前安全微调的脆弱性。
方法
用有害/无害指令的对比对计算difference-in-means方向,通过方向消融和激活加减验证其因果作用,并提出等价的权重正交化(ORTHO)白盒越狱;还用DFA分析对抗后缀如何抑制该方向。
实验与结果
作者称13个模型中消融单一方向即可绕过拒答、添加可诱导拒答;ORTHO在HarmBench上Qwen 14B ASR 84.3%,能力评测大多变化很小,但TruthfulQA一致下降。
局限与可以继续做的
作者指出:结果可能不推广到更大或专有模型;方向提取方法依赖启发式;对抗后缀分析仅限单一模型和单一后缀;方向语义不明。实验仅覆盖13个开源模型(最大72B)。
实验设置Qwen 1.8B Chat、Qwen 7B Chat 等 · JailbreakBench、HarmBench 等 · refusal score、safety score 等
被测模型
Qwen 1.8B ChatQwen 7B ChatQwen 14B ChatQwen 72B ChatYi 6B ChatYi 34B ChatGemma 2B ITGemma 7B ITLlama-2 7B ChatLlama-2 13B ChatLlama-2 70B ChatLlama-3 8B InstructLlama-3 70B Instruct
基准
JailbreakBenchHarmBenchAdvBenchMaliciousInstructTDC2023AlpacaMMLUARCGSM8KTruthfulQAWinoGrandeTinyHellaSwagThe Pile
指标
refusal scoresafety scoreASRCE losscosine similarityrefusal metric
AI 导读和推荐理由作者在13个开源对话模型(最大72B)上发现,拒答行为由残差流中的单一方向中介。用有害与无害指令的激活均值差提取该方向后,消融它可让模型不再拒答有害指令,加入它则会让模型拒答无害请求。作者据此提出权重正交化的白盒越狱,直接改写写入残差流的权重,无需梯度优化或有害回答样本;在HarmBench上其攻击成功率在Qwen系列与GCG等逐提示优化方法相当,且MMLU、ARC、GSM8K表现接近原模型,但TruthfulQA准确率下降。对Qwen 1.8B的分析显示,对抗后缀会抑制拒答方向并让关键注意力头从指令转向后缀。作者称该方法用不到5美元算力即可得到70B模型的越狱版本。

作者在13个开源对话模型(最大72B)上发现,拒答行为由残差流中的单一方向中介。用有害与无害指令的激活均值差提取该方向后,消融它可让模型不再拒答有害指令,加入它则会让模型拒答无害请求。作者据此提出权重正交化的白盒越狱,直接改写写入残差流的权重,无需梯度优化或有害回答样本;在HarmBench上其攻击成功率在Qwen系列与GCG等逐提示优化方法相当,且MMLU、ARC、GSM8K表现接近原模型,但TruthfulQA准确率下降。对Qwen 1.8B的分析显示,对抗后缀会抑制拒答方向并让关键注意力头从指令转向后缀。作者称该方法用不到5美元算力即可得到70B模型的越狱版本。

推荐理由论文在13个开源对话模型上定位出单一拒答方向,并据此给出低算力白盒越狱方法,可用于评估安全微调的脆弱性。

相关热点

6 个事件 · 11 条新闻提到

提到这篇论文的新闻

Hugging Face被指托管大量去审查模型

研究者用两个神经元激活干预让Qwen3.5-4B变反派语气

开源模型后门可借编码代理窃取凭据

再看 8 条

Anthropic评测GLM-5.3自主网络攻击能力

CERT-EU演示AI智能体框架第三方扩展后门风险

NVIDIA 提出 NXDirection 防御提示注入

深度解读

6个问题,点开问题读完整回答

  1. 这篇论文试图解决什么问题?

    聊天模型拒答行为的内部机制不明,作者研究其是否由单一方向介导并据此构造白盒越狱。

    核心问题:安全微调后的聊天模型如何表示和执行拒答行为,这一机制能否被简单地定位和移除。

    • 场景与重要性:部署的聊天模型经多轮微调以同时做到有帮助和无害,作者认为随着模型获得更高自主性,稳健拒答有害请求是安全系统的必要条件(§1)。
    • 现有理解不足:作者称拒答行为在聊天模型中普遍存在,但其底层机制仍缺乏理解(摘要);已有工作研究了harmlessness等特征的线性表示,但拒答机制本身未被定位。
    • 核心观察:作者假设特征以激活空间中的线性方向表示,并用对比对验证:在13个开源聊天模型(最大72B)上,拒答由一维子空间介导(§3)。
    • 论文产出:提出基于权重正交化的白盒越狱方法ORTHO(§4),并对对抗后缀抑制拒答方向的机制做了初步分析(§5)。
    • 威胁模型:
      • 目标:使开源聊天模型不再拒答有害指令,同时保留其他能力。
      • 知识:白盒——攻击者拥有模型权重访问权限(§4)。
      • 能力:直接修改模型权重(秩一修改),不需要梯度优化或有害补全样本。
      • 受害系统:13个开源聊天模型,涵盖Qwen、Yi、Gemma、Llama-2、Llama-3系列(Table 1)。
  2. 有哪些相关研究?

    相关工作分为拒答机制研究、特征方向表示、撤销安全微调和越狱攻击四类。

    理解语言模型中的拒答

    • Zheng et al. (2024) 与 Zou et al. (2023a):用有害/无害对比对识别harmfulness表示,并主张该方向与拒答表示不同;论文指出这与本文§5.1的发现相反——本文中拒答方向被对抗后缀抑制。
    • Panickssery et al. (2023):用对比式多选补全提取转向向量,在多选设置下可调节拒答,但在长文本生成中无效。
    • Wei et al. (2024):移除安全关键神经元和秩可在保留效用的同时削弱安全机制;Wang and Shu (2024) 提出Trojan Activation Attack在推理时加转向向量绕过拒答。

    特征作为方向

    • 从对比对提取特征方向是已有技术(Burns et al., 2022; Panickssery et al., 2023; Zou et al., 2023a),向残差流添加特征向量可改变行为(Li et al., 2024a; Marks and Tegmark, 2023; Tigges et al., 2023; Turner et al., 2023)。
    • 多项工作表明激活空间中的方向比神经元更具"特征"性质(Bolukbasi et al., 2016; Elhage et al., 2022; Park et al., 2023b),稀疏自编码器可无监督发现特征方向(Bricken et al., 2023; Templeton et al., 2024)。

    撤销安全微调

    • Lermen et al. (2023)、Yang et al. (2023)、Zhan et al. (2023):在恶意样本上微调足以移除安全护栏且能力损失很小;论文指出微调需要有害指令和补全样本,而本文方法只需有害指令。
    • Lee et al. (2024) 等机制可解释性工作:微调不改变相关内部回路,例如降毒微调可通过缩放少量MLP权重撤销。

    越狱

    • 社会工程类越狱(Perez et al., 2022; Shah et al., 2023b; Wei et al., 2023);优化的对抗后缀(Zou et al., 2023b; Andriushchenko et al., 2024; Liao and Sun, 2024)。论文指出本文方法不需修改输入提示词,但需要权重访问权限。

    基线与基准:在HarmBench(Mazeika et al., 2024)的159条standard behaviors上,与GCG、GCG-M、GCG-T、AP、PAIR、HUMAN(人工模板)和DR(直接回答)基线比较ASR(§4.2、§F.1)。作者将本文定位为:用模型内部理解同时解释已有漏洞并发现新漏洞,且方法比微调类越狱更简单。

  3. 论文如何解决这个问题?

    用difference-in-means提取拒答方向,通过方向消融/激活加减干预,并给出等价的权重正交化实现。

    整体思路:在残差流激活中找到单一"拒答方向",验证其对拒答行为的必要性(消融)和充分性(添加),再将推理时消融转化为等价的权重修改。方法核心为difference-in-means方向提取与directional ablation。

    数据与设定

    • 数据集:Dharmful取自AdvBench、MaliciousInstruct、TDC2023、HarmBench;Dharmless取自Alpaca。各含128条训练、32条验证样本,且与评测集两两不相交(§2.2、§A)。
    • 分析区域:只分析post-instruction token位置(聊天模板中指令之后的模板token)的激活(§2.1)。

    方向提取(§2.3)

    • difference-in-means:对每层l和每个post-instruction位置i,计算有害与无害提示词平均激活之差 ri(l) = μi(l) − νi(l),得到|I|×L个候选向量。
    • 选择单个向量:在验证集上评估每个候选的bypass_score(消融后的平均refusal metric)、induce_score(添加后的平均refusal metric)和kl_score(消融前后无害数据上末token分布的KL散度),选取bypass_score最小且满足induce_score>0、kl_score<0.1、l<0.8L的向量(§C.1)。

    干预方式(§2.4)

    • 激活添加:x(l) ← x(l) + r(l),仅在提取层l的所有token位置添加,用于诱导拒答。
    • 方向消融:x′ ← x − r̂r̂⊤ x,在所有层、所有token位置执行,阻止模型表示该方向。

    权重正交化(§4.1)

    • 将所有写入残差流的矩阵(嵌入、位置嵌入、attention输出、MLP输出矩阵及输出偏置)的列对r̂正交化:Wout ← Wout − r̂r̂ᵀWout。
    • 论文在§E证明该权重修改与推理时方向消融等价,因此§3.1的结果同样刻画权重修改的效果。

    评测方式(§2.5)

    • refusal score:补全含任一"refusal substring"则记1(字符串匹配,§D.1)。
    • safety score:用Meta Llama Guard 2将补全分类为safe/unsafe(§D.2)。
    • 生成设置:贪心解码,最大生成长度512 token(§2.5)。
    • ASR:在HarmBench测试集上由其提供的分类器判定攻击成功率(§4.2)。
  4. 论文做了哪些实验?

    在13个模型上验证方向消融/添加的效果,在HarmBench上对比越狱基线,并评测能力保持与对抗后缀机制。

    实验设置

    • 被测模型:13个开源聊天模型——Qwen 1.8B/7B/14B/72B Chat、Yi 6B/34B Chat、Gemma 2B/7B IT、Llama-2 7B/13B/70B Chat、Llama-3 8B/70B Instruct(Table 1)。
    • 评测集:§3用JailbreakBench(100条有害指令,10类)和100条Alpaca无害指令;§4用HarmBench测试集159条standard behaviors(6类)(§A)。
    • 指标:refusal score(拒答子串匹配)、safety score(Llama Guard 2分类)、ASR(HarmBench分类器);能力用MMLU、ARC、GSM8K、TruthfulQA等。
    • 基线越狱:GCG、GCG-M、GCG-T、AP、PAIR、HUMAN、DR(§F.1)。
    • 样本量:Figure 1和Figure 3中n=100,误差棒为标准误(§D.4);§5分析用128条指令(§H.2)。

    主结果

    Figure 1显示方向消融在13个模型上均降低refusal score和safety score(具体数值以图中柱高呈现,文本转换无法逐一读出);Figure 3显示激活添加使模型拒答无害指令。HarmBench ASR对比如下(据Table 2,括号内为无系统提示词的ASR):

    表格较宽,可左右滑动

    模型ORTHOGCG-MGCG-TDRGCGPAIR
    Llama-2 7B22.6 (79.9)20.016.80.034.57.5
    Llama-2 13B6.9 (61.0)8.713.00.528.015.0
    Llama-2 70B4.4 (62.9)5.515.20.036.07.5
    Qwen 7B79.2 (74.8)73.348.47.079.558.0
    Qwen 14B84.3 (74.8)75.546.09.583.551.5
    Qwen 72B78.0 (79.2)-36.68.5-54.5
    • 作者称ORTHO作为通用越狱与逐提示词优化的GCG在Qwen系列上相当(§4.2)。
    • 系统提示词影响因模型家族而异:Llama-2含系统提示词时ASR大幅下降,Qwen几乎不变;§F.2的分析显示该差异不能仅由提示词内容解释。

    能力保持评测

    • 设置:对正交化模型跑MMLU、ARC、GSM8K、TruthfulQA(LM Evaluation Harness,设置与Open LLM Leaderboard一致),§G.1另有WinoGrande和TinyHellaSwag(§4.3)。
    • 结果:除Qwen 7B和Yi 34B外,所有指标落在原模型99%置信区间内;Table 3中各家族最大模型的MMLU/ARC/GSM8K变化在-0.7到+1.5之间。TruthfulQA一致下降(如Llama-3 70B从61.8降到59.5,Table 3)。
    • 作者解读:作者认为TruthfulQA涉及misinformation、stereotypes、conspiracies等接近拒答领域的类别,行为差异在直觉上合理(§4.3、§G.2)。

    对抗后缀机制分析

    • 设置:在Qwen 1.8B Chat上研究单一GCG生成的对抗后缀,对128条有害指令比较无后缀/对抗后缀/随机后缀三种情形(§5.1、§H.1)。
    • 结果:Figure 5显示有害指令的末token激活与拒答方向余弦相似度很高,加随机后缀仍高,加对抗后缀后被抑制到接近无害指令水平。Figure 6(a)显示对拒答方向DFA最高的8个attention head的输出被抑制;Figure 6(b)显示这些head的注意力从指令区域转向后缀区域。
    • 作者解读:作者称对抗后缀"劫持"了重要head的注意力,使其远离有害指令内容(§5.2)。

    其他消融与分析

    • CE loss(Table 9):方向消融在The Pile、Alpaca和on-distribution数据上的CE loss接近基线,激活添加(减去向量)的CE loss明显更高,作者据此称消融比激活添加更"surgical"(§G.3)。
    • 与微调对比(Table 10):Llama-3 8B上,方向消融后refusal score 0.01、safety score 0.15;LoRA微调(rank=16,4 epochs)后为0.00和0.08;作者指出微调需要构建有害补全数据集,而本文方法只需有害指令(§I.2)。
    • 基座模型(Figure 24):Qwen 1.8B、Gemma 7B、Llama-3 8B、Qwen 14B的基座模型在有害提示词上也高表达拒答方向,作者认为该方向在安全微调前已存在、微调时被"复用"(§J)。
    • 正交化模型行为:§L.1显示正交化模型在被明确要求时仍能输出标准拒答字符串;§L.3显示其对自身新拒答行为的元问题解释不连贯。
  5. 有什么可以进一步探索的点?

    作者列出五项局限,包括泛化性、提取方法启发式、后缀分析范围、连贯性度量和方向语义。

    作者指出的局限与后续方向

    • 泛化性:结果可能不推广到未测试的模型,尤其是更大规模的当前专有模型和未来模型(§7 Limitations)。
    • 提取方法:方向提取方法可能不是最优、依赖若干启发式;作者将本文视为存在性证明,方法改进留给未来工作(§7)。
    • 后缀分析范围:对对抗后缀的分析不提供完整的机制性理解,且仅限于单一模型和单一对抗样本(§7)。
    • 连贯性度量:聊天模型的连贯性难以度量,作者认为所用每个指标都有缺陷,故用多种指标给出宽泛视角(§7)。
    • 方向语义:方向的语义含义不明,可能表示"harm"或"danger"等概念,甚至可能无法直接语义解释(§7)。
    • 系统提示词:系统提示词与拒答行为的关系需更系统的研究,作者留作未来方向(§F.2)。

    实验覆盖范围

    • 被测模型为13个开源聊天模型,参数规模1.8B至72B,涵盖AFT和APO两类对齐方式(Table 1)。
    • 越狱对比仅在HarmBench的6个模型(Llama-2三个、Qwen三个)上进行,Qwen 72B缺GCG-M、GCG、AP三列(Table 2)。
    • 对抗后缀机制分析仅在Qwen 1.8B Chat和单一后缀上进行(§H.1)。
    • 基座模型对比仅覆盖Qwen 1.8B、Gemma 7B、Llama-3 8B、Qwen 14B四个模型(§J)。
    • 论文未报告拒答/安全评测的人工一致性校验;refusal score为字符串匹配,safety score由Llama Guard 2判定(§2.5)。
  6. 总结一下论文的主要内容

    作者称拒答由单一方向介导,消融它即可白盒越狱且能力损失小,并初步解释了对抗后缀的作用机制。

    一篇机制可解释性与越狱结合的工作:研究安全微调聊天模型的拒答行为如何表示,并用该理解构造攻击。

    • 问题:聊天模型普遍会拒答有害请求,但机制不明;作者希望定位拒答机制并评估其脆弱性。
    • 方法:用128条有害和128条无害指令的对比对计算difference-in-means方向,经启发式筛选得到单一"拒答方向";方向消融可绕过拒答,激活添加可诱导拒答;权重正交化(ORTHO)将消融转化为等价的秩一权重修改,不需梯度和有害补全样本。
    • 主要结果:
      • 13个开源模型(最大72B)上,消融单一方向即降低拒答率并产生不安全补全(Figure 1),添加方向使模型拒答无害指令(Figure 3)。
      • HarmBench上ORTHO的ASR:Qwen 14B为84.3%,Qwen 7B为79.2%;Llama-2系列含系统提示词时仅4.4–22.6%,去掉后升至61.0–79.9%(Table 2)。
      • 能力评测大多变化很小,但TruthfulQA一致下降(如Llama-3 70B降2.3,Table 3);Qwen 7B在MMLU和GSM8K上有统计上可辨的下降(§G.1)。
      • 机制分析显示对抗后缀抑制拒答方向的表达,并将关键attention head的注意力从指令区域"劫持"到后缀区域(Figure 5、6)。
    • 作者结论:作者认为当前安全微调方法脆弱,简单的秩一权重修改即可近乎消除拒答行为,并希望对开源模型负责任发布的讨论有所贡献(§1、§7)。

相关论文

共同使用 ARC
共同使用 MaliciousInstruct
再看 3 篇
共同使用 MaliciousInstruct
共同使用 TruthfulQA
阅读原文arxiv.org