RepIt:用针对特定概念的拒答向量引导语言模型
RepIt: Steering Language Models with Concept-Specific Refusal Vectors
作者提出RepIt,从DIM向量中解耦出概念特异性拒答向量,在5个开源模型上实现对WMD类别的选择性越狱(目标ASR 0.4–0.7)且非目标ASR约0.1。
攻击者目标是在模型中嵌入狭窄的有害能力,同时保持在安全基准上的表现(评估规避)。
- 现有激活转向方法的拒答向量效果过宽,会同时抑制有害和无害响应;作者认为缺乏概念特异性的行为隔离方法本身就是安全漏洞:模型可被改造成在标准基准上表现为安全、却在特定有害领域保留可被利用的能力,从而规避基于基准的监管。
- 作者提出RepIt:先计算目标概念与Alpaca的difference-in-means向量,再经重加权、白化、受控正交化三步从21个非目标概念向量中解耦目标方向,用COSMIC选择层与位置、以非目标验证集ASR低于0.1为准则网格搜索ρ,最后用ACE施加干预。
- 在5个开源模型上,RepIt对WMD目标类别的ASR保持在0.4–0.7,非目标ASR降至约0.1,并在TDC2023等4个未见基准上保持拒答;仅用12条目标样本仍有效;编辑集中在100–200个残差维度;HarmBench的同类目提示低估了被攻击模型的有害能力。
- 作者在附录A指出:类别语义重叠使解耦不完全、目标概念可能与未测概念纠缠;评测主要在WMD相关概念上进行;COSMIC只做二元优化且ρ搜索与位置选择未联合优化,结果可能偏保守;LlamaGuard判定的ASR不代表输出包含可操作的WMD信息。
- 威胁模型
- 攻击者目标是在模型中嵌入狭窄的有害能力,同时保持在安全基准上的表现(评估规避)。作者假设攻击者具有对模型激活的white-box访问权限,可用最少12条目标样本和公开有害数据集计算转向向量;攻击者分为training-time、distribution和inference-time三类。
- 被测模型
- GLM-4.1V-9B-ThinkingQwen3-4B-Thinking-2507Mistral-Small-3.2-24B-Instruct-2506Phi-4-Mini-InstructLlama-3.1-Nemotron-Nano-4B-v1.1
- 基准
- WMDPJailBreakVStrongREJECTTDC2023JailbreakBenchAdvBenchMalicious InstructHarmBenchToxiGenAlpaca
- 指标
- ASR∆ASR
UC Berkeley 与 UC Santa Cruz 研究者提出 RepIt 框架,可选择性抑制语言模型对特定有害概念(如大规模杀伤性武器)的拒答而保留其他拒答。在 GLM-4.1V-9B-Thinking、Qwen3-4B-Thinking-2507、Mistral-Small-3.2-24B、Phi-4-Mini、Llama-3.1-Nemotron-Nano-4B 五个开源权重模型上,目标类别攻击成功率达 0.4–0.7,非目标类别约 0.1,在 TDC2023、JailbreakBench 等未见基准上仍显示安全。该方法对差值均值向量做重加权、白化和正交化,向量编辑集中在 100–200 个残差维度,仅需 12 个样本和单张 RTX A6000 即可提取,干预等价于静态权重编辑,可永久嵌入模型。
推荐理由论文展示仅用 12 个样本和单张 RTX A6000 就能定向抑制特定危害概念的拒答并在常规安全评测中仍显示安全,做安全评测的团队可据此检查盲区。
深度解读
这篇论文试图解决什么问题?
作者称现有转向向量效果过宽,缺乏概念特异性隔离方法本身构成评估规避漏洞,并提出RepIt。
核心问题:如何从语言模型激活中隔离出概念特异性的拒答表示,从而选择性地只抑制某一有害概念的拒答、同时保留对其他概念的拒答。
- 问题场景与重要性:作者认为拒答、事实性等行为属性在表示空间中共享重叠方向,这种纠缠使转向产生副作用;当前安全评估假设模型会在相关概念上一致拒答,基于基准的认证因此可能被规避,构成治理风险。
- 现有方法的不足:作者称现有拒答向量会不加区分地同时抑制有害和良性响应(引Arditi et al., 2024; Lee et al., 2024);对抗微调容易产生宽泛的emergent misalignment,但单一概念的misalignment难以实现(引Betley et al., 2025; Turner et al., 2025)。
- 核心观察:未处理的difference-in-means(DIM)向量会泛化到目标概念之外的有害主题(Figure 2中透明柱),作者认为通过解耦可以把编辑收窄到目标概念。
- 论文提出的方法:RepIt(Representing Isolated Targets),通过重加权、白化、正交化三步解耦DIM向量,作者报告其可只用12条目标样本、编辑100–200个维度,在WMD领域选择性抑制拒答。
- 威胁模型(第9节):
- 目标:在模型中嵌入狭窄的有害能力,同时维持在安全基准上的表现,使模型通过标准化安全评估却保留特定领域的可利用功能。
- 知识:作者假设攻击者对模型激活有white-box访问权限。
- 能力:能用最少12条目标样本加公开有害数据集计算转向向量;分training-time(控制训练流程)、distribution(通过模型分享平台分发)、inference-time(控制服务基础设施、在前向传播中注入转向向量)三类攻击者。
- 受害系统:开源权重语言模型及其下游用户;作者称攻防资源不对称——攻击只需找到基准未覆盖的概念,而防御性评估面临概念组合爆炸。
有哪些相关研究?
相关工作分安全对齐与绕过、激活转向与概念擦除、表示几何三组;基线是未处理的DIM向量。
安全对齐及其绕过
- Ouyang et al. (2022)、Bai et al. (2022)、Ganguli et al. (2022)——通过微调和RLHF实现对齐,是拒答行为的来源。
- Lermen et al. (2023)、Yang et al. (2023)、Qi et al. (2024) 与 Andriushchenko et al. (2024)、Zou et al. (2023b)、Chao et al. (2023)、Wang et al. (2025)——分别用微调和对抗提示词绕过拒答机制,论文称这些工作暴露了模型安全的缺口。
- Betley et al. (2025) 与 Turner et al. (2025)——论文引前者称微调可引入跨类别的宽泛misalignment,引后者称单一类别的misalignment难以实现;Marks et al. (2025) 用RL植入隐藏目标并发现可被SAE检测。
激活转向与概念擦除
- Arditi et al. (2024)、Marshall et al. (2024)——发现拒答行为由激活空间中的方向介导;论文指出Arditi et al. (2024)与Lee et al. (2024)的拒答向量会不加区分地抑制有害和良性响应,这是本文针对的不足。
- Zou et al. (2023a)、Panickssery et al. (2023)、Burns et al. (2023)——用对比数据对提取特征方向做行为转向。
- Ravfogel et al. (2020)、Haghighatkhah et al. (2022)、Belrose et al. (2023)、Guerner et al. (2023)——概念移除/擦除技术,与本文的正交化思路相关。
表示几何
- Park et al. (2024)——线性表示假说;论文引其说明正交不保证机制独立。
- Wollschläger et al. (2025)——提出拒答占据多维"concept cones";论文称RepIt把纠缠与独立贡献的划分操作化,与该观点一致。
- Lee et al. (2024)、Li et al. (2023)、Wang et al. (2024)——推理时条件化干预行为。
基线与基准:论文的直接对比基线是未处理的DIM向量 vt(即Arditi et al., 2024式做法,图中透明柱);评测基准包括WMDP(Li et al., 2024)、JailBreakV(Luo et al., 2024)、StrongREJECT(Souly et al., 2024)、TDC2023、JailbreakBench、AdvBench、Malicious Instruct、HarmBench、ToxiGen。
作者将本文定位为:已有转向方法实现了高ASR但效果过宽,RepIt提供的是概念特异性的隔离能力,并借此暴露基准评估的盲区。
论文如何解决这个问题?
RepIt用重加权、白化、受控正交化三步解耦DIM向量,再用COSMIC选方向、ACE施加干预。
整体思路:先用difference-in-means(DIM)得到目标概念的原始向量,再用 RepIt 三步解耦掉与非目标概念共享的子空间成分,最后用COSMIC选出干预位置、以ACE施加编辑。
问题设定
- 两类概念:target concepts(要越狱的特定概念)与non-target concepts(应保持拒答的有害查询);成功由双重目标衡量——最大化目标概念ASR,同时最小化非目标概念ASR的变化。
- 概念表示:目标概念用数据集内的类别表示(WMDP改写的Bio/Chem/Cyber),非目标概念为JailBreakV与StrongREJECT全部类别的并集,nntgt = 21;每次只用一个目标概念。
- DIM向量:对每层 ℓ 和每个post-instruction token位置 i,计算有害类别均值与无害基线(Alpaca)均值之差 vi,ℓ = v+i,ℓ − v−i,ℓ,即该位置的概念向量。
RepIt三步解耦
- 重加权:每个非目标向量乘以其逆范数权重 wj = 1/(‖vntgt,j‖+ϵ),ϵ = 1e−6,防止大范数向量主导后续分析。
- 白化:作者报告非目标向量协方差矩阵条件数在10^6到10^9量级、接近奇异(附录C.1),故用岭正则协方差 C = 1/nRw⊤ Rw + λ I 的Cholesky因子 L 白化目标与非目标向量,λ = 1e−4·mean(Rw²)+1e−12。
- 受控正交化:在白化空间对非目标矩阵做薄QR分解得到正交基 Q,计算目标向量在其张成子空间上的投影 P,只减去受控比例 αP,其中 α = 1−√(1−ρ),ρ∈[0,1] 控制移除强度;作者称完全正交化可能消除目标信号,且引Park et al. (2024)与Wollschläger et al. (2025)说明正交不保证机制独立,故采用部分移除,最后映射回原空间。
方向选择与干预施加
- COSMIC选位置:用COSMIC(Siu et al., 2025b)从候选方向中选出层与位置 (i*, ℓ*);论文称选它是因为它用隐状态而非子串匹配,能适应拒答形式多样的推理模型;因COSMIC只支持二元有害/无害设定,搜索被限制在非目标验证集上。
- ρ搜索:在 (0,1) 上网格搜索 ρ,选取使非目标验证集ASR低于 0.1 的最小 ρ;验证步骤不使用目标数据。
- ACE干预:用Affine Concept Editing(Marshall et al., 2024)在所选层输入、所有token上施加编辑;论文称这等价于静态权重编辑,可永久嵌入模型权重。
数据与运行设置
- 目标数据:WMDP选择题由GPT-4.1改写为多句自由回答指令,分Bio/Chem/Cyber三类(训练/验证/测试为508/127/637、162/40/205、509/127/637,Table 4)。
- 数据划分:WMD、JailBreakV、StrongREJECT按40%/10%/50%划分训练、验证、测试;针对cyberattack时从JailBreakV中排除"Malware"类别。
- 判定方式:用LlamaGuard 3判定响应是否有害,计算ASR。
- 运行资源:单张RTX A6000(Mistral用两张);推理模型max new tokens为1500,非推理模型为500。
论文做了哪些实验?
在5个开源模型上,RepIt对WMD目标ASR保持0.4–0.7、非目标ASR约0.1,并在未见基准上保持拒答。
实验设置
- 被测模型:GLM-4.1V-9B-Thinking、Qwen3-4B-Thinking-2507、Mistral-Small-3.2-24B-Instruct-2506、Phi-4-Mini-Instruct、Llama-3.1-Nemotron-Nano-4B-v1.1,共5个开源模型。
- 数据集:目标为WMDP改写的Bio/Chem/Cyber(测试集637/205/637条,Table 4);非目标为JailBreakV(2000条有害提示、16类,Table 2)与StrongREJECT(313条、6类,Table 3);泛化评测用TDC2023、JailbreakBench、AdvBench、Malicious Instruct;另有HarmBench同类目与ToxiGen毒性实验。
- 基线:未干预模型的原始ASR(baseline)与未处理的DIM向量 vt。
- 指标与评审:ASR,由LlamaGuard 3判定;ρ搜索阈值为非目标验证集ASR低于0.1。
- 样本与重复:主结果按目标概念各报一次、非目标按数据集聚合;数据效率实验用5个种子(20–24)报均值与范围;论文未报告主实验的重复次数。
主结果
主结果在Figure 2(目标vs非目标ASR)与Figure 1(未见基准泛化)中,论文未给出逐模型数值表,正文报告的范围如下:
表格较宽,可左右滑动
条件 目标(WMD)ASR 非目标ASR 出处 RepIt(vRepIt),5个模型 0.4–0.7 约0.1 Figure 2 未处理DIM向量 vt 高(作者称与RepIt相当或更高) 明显泛化、高于基线 Figure 2透明柱 RepIt在4个未见基准上 目标类别最高达0.7 非目标ASR增幅约0.1 Figure 1 - 作者称RepIt在所有模型上把非目标成功率压回基线水平,同时保持目标性能,实现了类别特异性信号的隔离。
- 作者称纠缠在大模型中更明显:4B模型的DIM泛化较少,GLM-4.1V(9B)与Mistral-3.2-Small(24B)泛化更多。
- 作者指出cyberattack实验在语义相近且未参与训练的malware提示上仍保持拒答,认为表示层面的概念向量可与表面类别标签分离。
机理解构(Figure 3)
- 测了什么:比较四种转向向量——原始目标向量 vt、非目标均值向量 R(p*,ℓ*)、被移除的投影 αP、最终向量 vRepIt。
- 结果:作者报告非目标向量和投影 αP 单独都能越狱目标概念,效果与 vt 相当甚至更好;移除 αP 后 vRepIt 的转向性能下降。
- 作者解读:同一不安全行为存在多条表示路径;RepIt通过移除被污染部分降低外溢。LlamaNemo4B的Chem和Cyber结果使用 ρ = 0(投影被置零)。
局部化与稀疏消融(Table 1、Table 6)
- 测了什么:对投影 αP 的坐标求z-score,只保留 |zi| ≥ 2 的重尾坐标,其余置零后重新转向。
- 结果:∆ASR(量级1e−2)基本不变,偏差在±0.05绝对ASR以内;重尾维度数为96–213个、占隐层维度3.8%–5.1%(如Qwen3T-4B Bio为96/3.8%,Mistral-3.2-Small Cyber为213/4.2%)。
- 例外:Table 6显示聚合稳定来自双向翻转的平衡——各模型-类别对有1.91%–9.43%的样本发生1→0或0→1翻转,作者称这是误差的重新分配而非严格保持性能。
数据效率(Figure 4)
- 测了什么:只用12或24条目标类别训练样本(占Bio/Cyber训练集的2.5–5%、Chem的7–15%)重跑流程,复用全量run的 p、ℓ、ρ,5个种子。
- 结果:作者报告12条样本即可隔离拒答方向,目标ASR受抑制、非目标保持低;24条样本降低方差,均值与全量数据接近。
其他消融与分析
- 欧氏正交化消融(Figure 8):不做重加权与白化、直接在欧氏空间部分正交化,ASR与完整RepIt差异在1–3个百分点,作者称差异主要来自ρ选择。
- 低秩结构(Figure 9):非目标基的前10个主成分解释超过90%方差(多数模型超过95%),作者称这解释了为何欧氏做法在本测试中也数值稳定。
- HarmBench同类目(Figure 6):Bio/Chem模型在HarmBench chemical biological类目(28条提示)、Cyber模型在cybercrime intrusion类目(40条)上的ASR远低于真实目标类别,作者称同类目标签的基准低估了被攻击模型的有害能力。
- 毒性泛化(Figure 10):以ToxiGen毒性提示为目标,RepIt在5个模型上保持高目标ASR、非目标近基线;作者指出LlamaNemo4B与Phi4-mini上毒性近乎与其他拒答概念正交,原始DIM向量即很少外溢。
- 投影诊断(Table 5):协方差条件数在10^6–10^10,白化后κspan恒为1.00;各模型Gini系数均约0.99;余弦相似度从GLM4.1V Cyber的0.59到LlamaNemo4B的1.00不等。
有什么可以进一步探索的点?
作者在附录A列出语义重叠、评测范围、COSMIC二元优化等局限,并称结果可能偏保守。
作者指出的局限与后续方向
- 语义重叠与噪声(附录A):有害类别本身定义不清晰、类别间存在重叠,作者称无法保证从人的视角清理类别就消除了语义重叠,解耦不完全,模型可能对未直接针对的相关有害行为仍有漏洞;目标概念还可能与未评测数据集中的概念纠缠。
- 评测范围(附录A):评测主要在WMD相关概念上进行;作者称未来应在语义更独立的数据出现时扩展到其他领域。
- 方向选择的保守性(附录A):COSMIC只做有害/无害二元优化,无法优先强化目标概念的转向,作者称结果可能偏保守;未来可联合优化目标向量与非目标基,或把ρ与 (p, ℓ) 三者联合搜索;RepIt兼容其他方向选择方法。
- 拒答类别不纯(附录A):DIM向量可能被模型本就不拒答的提示污染,ρ搜索可能无法干净收敛;作者称修正后测得的效果可能更强,即当前结果是保守估计。
- ASR指标的局限(附录A):LlamaGuard判定的ASR不代表输出含有可操作、真实、可武器化的WMD信息(例如Qwen3T-4B会幻觉病毒株名称)。
- 后续方向(第11节、第12节):作者称未来工作应探索对未公开激活修改的检测方法、天然抵抗概念特异性转向的架构设计、把vRepIt分解为可解释的latent特征,以及动态、representation-aware的审计。
实验覆盖范围
- 被测模型为5个开源权重模型(4B到24B),论文未在闭源模型上报告实验。
- 目标概念为WMDP的Bio/Chem/Cyber三类,另有ToxiGen毒性目标的补充实验(附录C.2)。
- 攻击成功判定只使用LlamaGuard 3,论文未报告人工评估或评审间一致性。
- 数据效率实验覆盖12与24条目标样本、5个种子;主实验的重复次数论文未报告。
- 所有实验在单张或两张RTX A6000上完成(附录B.5);论文未报告对现有防御(如检测器)的自适应攻击评测。
总结一下论文的主要内容
RepIt把DIM向量解耦为概念特异性拒答向量,作者称其暴露了基准安全评估的盲区。
一篇激活转向攻击论文:作者提出RepIt,从difference-in-means向量中隔离概念特异性的拒答方向,用以制造能通过标准安全基准、却在特定有害领域保留能力的模型。
- 要解决的问题:作者称现有拒答向量效果过宽、单一概念的misalignment难以实现,而基于基准的安全认证假设拒答跨概念一致,这一缺口使评估规避成为可能。
- 方法要点:对目标概念算DIM向量后,经重加权、白化、受控正交化三步减去其与21个非目标概念共享的子空间成分(强度由ρ控制),用COSMIC选层与位置、以非目标验证集ASR低于0.1选ρ,再以ACE施加等价于静态权重编辑的干预。
- 主结果:在5个开源模型上,RepIt对WMD目标类别ASR保持0.4–0.7、非目标ASR降至约0.1(Figure 2);在TDC2023、JailbreakBench、AdvBench、Malicious Instruct四个未见基准上非目标增幅约0.1(Figure 1)。
- 效率与局部化:仅用12条目标样本即可隔离方向(Figure 4);编辑集中在100–200个残差维度,稀疏化后∆ASR在±0.05以内(Table 1)。
- 例外与负面结果:机理解构显示被移除的投影 αP 本身越狱效果与原始向量相当甚至更好(Figure 3);稀疏化的聚合稳定实为双向翻转的平衡,单样本层面有最高约10%的行为改变(Table 6);HarmBench同类目提示上的ASR远低于真实目标(Figure 6)。
- 作者结论:作者认为RepIt暴露了当前安全评估实践的漏洞——基准通过不等于安全,呼吁发展动态、representation-aware的审计与对未公开激活修改的检测方法;作者计划通过受控访问发布WMD提示数据,代码已提供匿名版本。