← 返回 2026-05-19

通过对比对搜索实现目标神经元的精准调控 Targeted Neuron Modulation via Contrastive Pair Search

Sam Herring, Jake Naviasky, Karan Malhotra 📅 2026-05-12 👍 17 2026-07-13 08:36
AI 安全 可解释性 大语言模型 对齐 机制解释 神经元调控

CNA:仅干预 0.1% MLP 神经元,将指令模型拒绝率降低超 50% 且保质量

前置知识

残差流(Residual Stream)

Transformer 中逐层累加的信息通道,每层把自身输出加到这条流上。它承载了模型需要的全部上下文表征,是层间信号传递的总线。

CAA 等主流 steering 方法直接对残差流整体加一个向量,这正是它们在强干预下导致生成崩溃的根源,是理解 CNA 出发的必要背景。

MLP 神经元 / down_proj

Transformer FFN 子层中 down_proj 矩阵的每个输出维度就是一个"神经元",对应一个标量激活值。CNA 直接操作这些标量而非层向量。

本文的核心主张是:拒绝行为由稀疏的几十到几百个 MLP 神经元承担,而非整条残差流方向。理解 down_proj 才能看懂 $\delta_j^\ell$ 公式的物理含义。

CAA(Contrastive Activation Addition)

Rimsky 等 2024 提出的 steering 方法:分别用有害/良性两组 prompt 跑模型,取残差流激活均值之差得到"控制向量",推理时叠加该向量。

CNA 是 CAA 在神经元级别的对偶方法,论文大量与 CAA 对比,因此必须先理解 CAA 才能读懂 CNA 的对比实验。

JBB-Behaviors 越狱基准

NeurIPS 2024 Chao 等提出的 100 条有害 prompt 评测集,用关键词分类器判断模型是否拒绝,是衡量 safety 行为的标准基准。

本文所有主表(Table 3)以 JBB-Behaviors 上的拒绝率为指标,理解这一基准对解读实验数字至关重要。

研究动机

现有 safety steering 方法几乎都作用于残差流整体,以 CAA 为代表:它计算有害/良性 prompt 在残差流上的均值差作为控制向量,推理时按强度 $\alpha$ 叠加。然而这种"粗粒度"干预把整个层的信号一刀切,导致模型在 $\alpha \geq 0.75$ 时生成质量断崖式下降。论文 Figure 1 给出 8 个 instruct 模型的均值曲线:CAA 的生成质量在 $\alpha = 1.0$ 时从 1.0 跌到约 0.5 以下,Qwen2.5-72B 甚至跌到 0.406。更糟的是,Qwen2.5-1.5B 和 Qwen2.5-72B 退化出的重复文本会被关键词分类器误判为"拒绝",在高强度下出现表观拒绝率反向上跳到 100% 的伪现象,这彻底阻塞了此类方法的实际部署。

本文的目标是论文的具体目标是提出一种在神经元级别工作的 steering 方法 CNA:仅干预 0.1% 的 MLP 激活(按绝对差异最大的 top-k 选取),就能在 JBB-Behaviors 上把拒绝率降低超 50%,同时在所有 steering 强度下保持生成质量 > 0.96,且不影响 MMLU 等通用能力。论文同时希望回答一个机制问题:拒绝电路是微调引入的全新结构,还是预训练已有结构被微调"功能化"的结果?

与已有工作不同的是,既有工作在两个极端之间摇摆:一端是残差流级别的粗粒度 steering(CAA、RepE),粒度粗、质量差;另一端是稀疏自编码器(SAE)或 Layer-wise Relevance Propagation(RelP),虽细但需要昂贵的外挂训练或梯度回传。本文抓住了一个被忽视的中间粒度——模型原生的单个 MLP 神经元,且宣称这是首次在神经元级别用纯前向、无梯度、无外挂训练的方式复现安全 steering 的成功,并首次通过 base vs. instruct 的对照实验把"结构"和"功能"清楚区分开。

核心方法

CNA 的直觉非常朴素:CAA 在残差流上做对比,CNA 把它直接搬到 MLP 神经元。具体来说,先准备一组"目标行为"的 prompt(正例,比如有害指令)和一组"非目标行为"的 prompt(负例,比如良性指令),把它们一起前向跑一遍模型,用 forward pre-hook 把每一层 down_proj 输出的 MLP 激活抓出来,对每个神经元 $j$ 在第 $\ell$ 层计算其在正/负例集合上的平均激活差 $\delta_j^\ell$,再按绝对值取 top 0.1% 作为"回路"。推理时把这个回路里的神经元乘以一个标量 $m$($m=0$ 即消融,$m=1$ 是 baseline,$m>1$ 放大),其他神经元原样通过。整个流程只需前向传播和一次排序,不需要任何梯度或额外模型。

CNA 与 CAA 的本质区别在于干预粒度:CAA 在 d_model 维残差流向量上加一个偏移,CNA 在 down_proj 输出的 d_model 维激活里只挑出 ~0.1% 的标量坐标做乘法。第二个关键差异是"通用神经元过滤":先用一组多样化 prompt 跑模型,把在 ≥80% prompt 中都进入 top 0.1% 激活的"通用神经元"排除掉,避免它们混入回路。这些被排除的多是任务无关的频繁激活单元,留着会污染信号。第三个差异是把 steering 从加性(additive)改成乘法(multiplicative)且范围更窄,这避免了大幅修改整层信号带来的级联崩溃。

方法步骤详情

方法分四步。第一步是构造对比对:对拒绝任务,正例用 100 条有害 prompt,负例用 100 条良性 prompt(一般任务只需 8+8 条)。第二步是采集激活:对每个 prompt 做一次前向传播,在每一层 down_proj 上挂 forward pre-hook,记录最后一个 token 位置的 MLP 输出向量 $a^\ell(x) \in \mathbb{R}^{d_\text{int}}$。第三步是计算对比差:用公式 $\delta_j^\ell = \frac{1}{|P^+|}\sum_{x\in P^+}a_j^\ell(x) - \frac{1}{|P^-|}\sum_{x\in P^-}a_j^\ell(x)$ 算每个神经元的平均差,跨所有层取 $|\delta_j^\ell|$ 最大的 top k 个,$k = 0.001 \times N_\text{MLP}$。第四步是推理时对回路内神经元乘以乘子 $m$,其余保持不变。另有独立的"通用神经元过滤":在通用 prompt 上统计跨 80% prompt 都进入 top 0.1% 激活的神经元并剔除,避免污染回路。

技术新颖性

技术新颖性体现在三点。第一是粒度的精细化:把干预单元从 d_model 维向量压到单个标量坐标,且数量级从全层降到 0.1%,这是首次用纯前向方法在神经元级别实现有效的安全 steering。第二是方法论上的极简:不需要 SAE 字典学习、不需要 RelP 反向传播、不需要梯度线性化,对工程实现非常友好。第三是首次在 base vs. instruct 对照中分离"结构 vs. 功能":同样的 late-layer 神经元位置在两种模型上都出现,但只在 instruct 模型中起到行为闸门作用,这是 Wu 等"FFN 知识被旋转而不改变层结构"和 Chaudhury 观察的实证延伸。

实验结果

Table 3 是最核心结果:JBB-Behaviors 上 8 个 instruct 模型做 0.1% 神经元消融,Llama-3.1-70B 从 86% 降到 18%(-68pp,-79.1%),Qwen2.5-7B 从 87% 降到 2%(-85pp,-97.7%),所有模型平均相对降幅超 50%。Table 1 的强 steering($\alpha=1.0$)对比显示:CNA 在 8 个 instruct 模型上质量均 >0.969,CAA 有 6 个模型跌到 0.60 以下,Qwen2.5-72B 仅 0.406。Figure 2 进一步说明 CAA 在 $\alpha=1.0$ 时 MMLU 跌到接近 0,CNA 保持基线 ±1 分。Table 2 显示同样的 CNA 流程在 8 个 base 模型上几乎不改变拒绝率,最大变化 +7.1pp,证明回路是微调引入的功能。Table 4 定性对比更直观:Llama-1B Instruct 在 $m=0$ 时把"我不能帮忙"变成合规锁匠教程,$m=2$ 时变"更强拒绝";base 模型只改主题从不真正拒绝。StrongREJECT 上 Llama 平均 +6%、Qwen +31%,说明 Qwen 对消融更敏感。

Refusal rate (%) and generation coherence when ablating the refusal mechanism (α = 1.0) across instruct models for 100 harmful prompts
Table 1: Refusal rate (%) and generation coherence when ablating the refusal mechanism (α = 1.0) across instruct models for 100 harmful prompts
Refusal rate (%) and generation coherence when ablating the refusal mechanism (α = 1.0) across base models for 100 harmful prompts
Table 2: Refusal rate (%) and generation coherence when ablating the refusal mechanism (α = 1.0) across base models for 100 harmful prompts
Refusal rate on JBB-Behaviors (100 prompts) before and after ablating 0.1% of MLP activations
Table 3: Refusal rate on JBB-Behaviors (100 prompts) before and after ablating 0.1% of MLP activations
Steering late-layer discrimination neurons in base and instruct models. Multiplier m scales circuit activations directly
Table 4: Steering late-layer discrimination neurons in base and instruct models. Multiplier m scales circuit activations directly
Refusal rate and generation quality vs. steering strength α, averaged across 8 instruct models (± 1 s.d.)
Figure 1: Refusal rate and generation quality vs. steering strength α, averaged across 8 instruct models (± 1 s.d.)
MMLU accuracy (1000 questions) vs. steering strength, averaged across 8 instruct models (± 1 s.d.)
Figure 2: MMLU accuracy (1000 questions) vs. steering strength, averaged across 8 instruct models (± 1 s.d.)
查看结构化数据
任务指标本文基线提升
JBB-Behaviors 越狱(消融) 拒绝率(%)(越低越好) 8 个 instruct 模型平均从 ~87% 降至 ~27%,相对降幅 35.6%-97.7% 未干预基线:Llama 84%-90%、Qwen 78%-93% 绝对降幅 32-85pp,相对降幅平均 ~70%
生成质量(强 steering,α=1.0) 1 - 重复 n-gram 比例(越高越好,1.0 为完美) CNA 在 8 个 instruct 模型上均 >0.969,平均 0.978 CAA 在 6/8 模型上跌至 <0.60,Qwen2.5-72B 仅 0.406 CNA 相对 CAA 平均提升约 0.35-0.55
MMLU 通用能力 1000 题准确率(%) CNA 在所有 steering 强度下保持基线 ±1 分 CAA 在 α=1.0 时降至接近 0 CNA 几乎无损,CAA 损失超 60 分
StrongREJECT 越狱评分 LLM judge 0-1 合规分(越高越被攻破) Llama 平均提升 6%、Qwen 平均提升 31% 未报告具体基线分数,文章仅给提升幅度 Qwen 系列对消融更敏感
Base 模型拒绝率(验证特异性) JBB-Behaviors 拒绝率(%) 8 个 base 模型 CNA 干预后变化 ±7.1pp 基线 2.0%-35.4% 无显著行为变化,证实回路由对齐引入

局限与改进

作者在论文 Limitations 中坦承三点。其一,对比发现使用的是原始激活差而非 RelP 归因,因此标准 faithfulness 指标不能直接套用,只能通过行为 steering、客观连贯性指标和基准评测间接验证。其二,实验只覆盖 Llama 和 Qwen 两个家族(gated SiLU MLP + GQA attention),未验证 MoE、长上下文、卷积或其他架构,外推性未知。其三,最大模型 72B 已经很大,但仍受算力约束;per-layer 精细分析仅在 Llama-3.2-1B 和 Qwen2.5-3B 上做了。我自己的额外观察:① Table 3 中 Qwen2.5-3B-Instruct 的相对降幅只有 35.6%,明显低于其他模型,但论文没有解释为何该模型异常顽固;② 关键词分类器本身不可靠(如 Qwen2.5-1.5B 在 CAA 下"退化即拒绝"),但 Table 1 仍沿用关键词判定,这可能高估 CAA 的实际有效;③ 0.1% 这个阈值是经验值,没有做 sweep 来证明最优性,k 调参空间未充分探索。

独立分析的弱点

独立审视后还有几个明显弱点。第一是回路选择的稳定性存疑:仅按 $|\delta_j^\ell|$ 大小排序选取 top-k,对 prompt 集合的随机种子敏感,作者也承认使用 100+100 比 8+8 更稳定,但小样本场景的方差没有量化报告,建议引入 bootstrap 重采样给出置信区间。第二是基线比较不够公平:CAA 的 $\alpha$ 和 CNA 的 $m$ 通过 $1-m$ 对齐到 [0,1] 区间,但 CAA 是加性干预而 CNA 是乘性干预,等价映射的物理意义并不严格相同,建议在统一度量空间下比较。第三是安全性讨论缺位:论文展示了如何把拒绝率从 90% 拉到 8%,但完全没讨论这种能力公开的 misuse 风险,也没有给出防御建议或对齐启示。第四是评估维度单一:JBB-Behaviors 是单一 benchmark,建议补充 HarmBench、AdvBench 等多套件以验证泛化性。改进方向是引入 RelP 验证 faithfulness、对回路神经元做消融子集分析(看是否冗余或必须)、并加上过度越狱检测。

未来方向

作者明确提出两个方向:①验证 CNA 是否能泛化到 MoE 架构(专家路由机制不同,神经元概念需要重新定义);②能否拓展到拒绝之外、同样具备干净对比对的其他行为(如谄媚 sycophancy、幻觉、情绪)。基于论文成果还可以延伸:①用 CNA 探查其他 alignment 后行为(RLHF、DPO 不同阶段的电路演化);②把神经元回路作为诊断工具,反向定位哪些训练数据强化了拒绝;③结合 SAE 在神经元空间做正交分解,验证回路是否对应 SAE 已学到的某些 interpretable feature;④把 $m$ 从离散 (0, 1, 2) 推广为可学习参数,做"靶向强化"而非仅"消融";⑤在 fine-tuning 阶段加入针对 CNA 回路的正则项,主动把拒绝行为分散到更多神经元以增强对抗鲁棒性。

复现评估

复现难度整体中等偏低。方法本身只需前向传播和排序,工程实现简单,论文给出了明确的超参($k=0.1\%$、$m\in\{0,0.25,0.5,0.75,1.0\}$、通用神经元过滤阈值 80%)。主要成本在算力:1B-3B 模型可在 RTX 3080 + bfloat16 跑,8B-72B 需要 B200 节点跑 bfloat16。数据集方面,JBB-Behaviors 是公开的 NeurIPS 2024 基准,但论文自建的"100 有害 + 100 良性"对比集未完全公开,构造细节需要推断。开源情况方面,作者来自 Nous Research 且论文标题是 arXiv preprint,未声明代码或模型权重 release 计划,reproducibility 链条最弱的一环是 prompt 集合的精确措辞和 discovery 阶段的随机种子,这两者对回路稳定性影响较大但论文披露不足。整体而言复现一个相近数字不难,但要严格复现作者表中的每一个百分点则需要完整开源。