返回知识盲盒

知识盲盒

第七章 Ablation、Patching 与因果追踪

ai-generatedstochastic-parrotstochastic-parrot-anatomyai

模型前向传播是已知计算图,因此研究者可以把内部节点设置为指定值并重新计算下游输出。这比相关读出更接近因果实验;但结论仍相对于节点定义、替换值、输入分布与允许干预族,不能自动解释为人类概念的唯一自然中介。

7.1 前向图作为确定性结构模型

把内部节点按拓扑顺序写成

Uk=fk(pa⁡(Uk);θ),U_k=f_k(\operatorname{pa}(U_k);\theta),

最终目标为 S=g(U1,…,UK)S=g(U_1,\ldots,U_K)。对节点 UjU_j 施加

do⁡(Uj←u)\operatorname{do}(U_j\leftarrow u)

表示忽略其原结构方程,固定为 uu,其余下游节点按原模型重算。

模型内部不存在传统观察研究中的未知边;但“把什么视为节点”和“赋什么值”仍由研究者决定。把 residual vector、单 head result、一个投影或 SAE latent 作为 UjU_j,得到的是不同干预。

7.2 单样本与分布效应

对输入 xx,节点干预的个体效应为

ΔI(x)=S(MI(x))−S(M(x)).\Delta_I(x)=S(M_{I}(x))-S(M(x)).

分布平均效应为

τI(Deval)=Ex∼Deval[ΔI(x)].\tau_I(\mathcal D_{\mathrm{eval}}) =\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}}[\Delta_I(x)].

应报告均值、分位数、符号一致率和 cluster bootstrap 区间。平均接近零可能来自正负效应抵消;只展示最大恢复样本则会严重选择偏差。

模型是确定性的并不消除抽样不确定性:Deval\mathcal D_{\mathrm{eval}} 上的有限样本仍决定我们能否推广。

7.3 Ablation 及其基线语义

常见 ablation 为:

  • zero:u←0u\leftarrow0;
  • mean:u←Ex∼Deval[U(x)]u\leftarrow\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}}[U(x)];
  • resample:u←U(x′)u\leftarrow U(x'),x′x' 来自匹配对照;
  • shuffle:在 batch 或 group 内重排;
  • projection:删除子空间 PVuP_Vu;
  • edge ablation:只阻断 sender 到 receiver 的贡献。

这些操作估计不同量。zero 问“没有该向量加数时怎样”,mean 问“替换为总体中心怎样”,resample 问“替换为对照实例状态怎样”。它们不应被平均成一个无条件必要性分数。

mean 可能从不在自然数据出现;resample 较接近真实 activation,却可能携带对照样本的多种属性。选择基线必须对应要消除的因素。

7.4 Clean/corrupt activation patching

构造 clean 输入 xcx_c 与 corrupt 输入 xrx_r,目标 score 满足 Sc>SrS_c>S_r。缓存 site uu 的 clean 值 ucu_c,在 corrupt run 中替换:

ur←uc.u_r\leftarrow u_c.

原始恢复量为

Δupatch=S(M(xr;do⁡(u←uc)))−Sr.\Delta^{patch}_u =S(M(x_r;\operatorname{do}(u\leftarrow u_c)))-S_r.

常见归一化恢复率

Ru=Spatched−SrSc−SrR_u=\frac{S_{patched}-S_r}{S_c-S_r}

仅在分母远离零时稳定,且可小于 00 或大于 11。必须同时报告 Sc,Sr,SpatchedS_c,S_r,S_{patched} 与未归一化 effect。

7.5 Corruption 是实验的一半

patch effect 依赖 clean/corrupt 差异。若 corrupt 同时改变主体、句长与答案位置,patch 恢复可能修复任何一个因素。

必要设计包括:

  • 多种 corruption 机制;
  • token 数、位置与频率匹配;
  • semantic 与 surface corrupt 分离;
  • clean/corrupt 双向 patch;
  • 在原模型均成功但答案不同的 pairs 上测试;
  • 检查 corruption 是否造成广泛 loss 崩坏。

“某 site 携带答案”应进一步区分它生成答案信息、传递上游信息,还是仅把整体 state 拉回 clean 区域。

7.6 Necessity 与 Sufficiency 的操作定义

对候选组件集 CC,设 M∖CM_{\setminus C} 为按规定删除 CC 的模型,MCM_C 为只保留 CC 的受限模型。可定义

NC=Ex∼Deval[S(M(x))−S(M∖C(x))],N_C=\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [S(M(x))-S(M_{\setminus C}(x))], QC=Ex∼Deval[S(MC(x))−S(Mbase(x))].Q_C=\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [S(M_C(x))-S(M_{base}(x))].

NCN_C 是指定删除下的必要性证据,QCQ_C 是指定基线与保留规则下的充分性证据。神经网络有冗余:删除单组件无效不证明未参与;只保留一个 direction 后 steering 有效也不证明自然前向依赖它。

必要与充分都不是组件的内在二元属性,而是 (C,Deval,S,I)(C,\mathcal D_{\mathrm{eval}},S,I) 的关系。

7.7 Total、Direct 与路径效应

普通节点 patch 允许替换值沿全部下游路径传播,测量 total intervention effect。若只允许 sender AA 的变化经 receiver BB 到达输出,则研究 path-specific effect。

线性无交互图中可以把效应相加;一般网络中

Δ(A∪B)≠Δ(A)+Δ(B).\Delta(A\cup B) \ne\Delta(A)+\Delta(B).

LayerNorm、softmax 和门控使路径相互依赖。所谓 direct/indirect effect 必须由具体边阻断与缓存协议定义,不能直接套用线性中介公式。

7.8 Path patching 与状态一致性

path patching 通常经历:

  1. clean 与 corrupt run 缓存 sender;
  2. 建立含 clean sender 的中间 run;
  3. 只把该变化对指定 receiver 的输入传入最终 run;
  4. 其他 sender-to-receiver edges 保持 corrupt;
  5. 比较目标 score。

不同实现对 receiver 的哪些输入固定、是否重新计算 attention pattern、是否包含 residual identity 并不相同。论文应给出计算图示意或伪代码。

混合多个 run 的节点可能形成没有任何自然输入产生的状态。path-specific 结论因此是“该人工组合中的传递效应”,而非自动可识别的自然路径效应。

7.9 Causal tracing

causal tracing 在位置和层上扫描 clean restoration,形成恢复热图。它适合定位事实回忆或实体信息的候选路径。

热图峰值支持:在给定 corruption 下恢复该 site 能有效恢复目标。它不证明:

  • 事实永久或唯一存储在该层;
  • 该层独立完成事实检索;
  • 不同 prompt、关系和语言使用同一路径;
  • 参数编辑该层不会有非局部副作用。

位置—层扫描还涉及多重发现,应在独立 pairs 上复验峰值,而不是在同一热图上发现和验证。

7.10 Activation steering 与剂量反应

沿方向 vv 修改

h′=h+αvh'=h+\alpha v

并估计剂量曲线

g(α)=Ex∼Deval[S(Mαv(x))].g(\alpha)=\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [S(M_{\alpha v}(x))].

一个可信控制方向应在小幅区间表现出可重复、方向一致的 effect,并报告:

  • vv 与 hh 的 norm 比;
  • 正负 α\alpha;
  • 多层与多位置;
  • matched random directions;
  • perplexity、格式与非目标能力;
  • 大 α\alpha 下的分布外崩坏。

steerability 证明方向具有控制作用;只有自然 activation、下游读取和删除证据共同成立时,才支持正常机制参与。

7.11 Off-manifold 的定义与诊断

自然 activation 集记为

Meval={u(x):x∼Deval}.\mathcal M_{\mathrm{eval}} =\{u(x):x\sim\mathcal D_{\mathrm{eval}}\}.

任意干预值 u′u' 可能远离 Meval\mathcal M_{\mathrm{eval}}。高维数据没有可靠统一距离,常用诊断包括:

  • 与邻近自然 activation 的 Mahalanobis/knn 距离;
  • norm、均值和协方差偏移;
  • 下游 next-token loss 或 hidden-state density proxy;
  • 非目标 logits 的广泛变化;
  • 用不同分布内 resampling 得到的结果一致性。

SAE 重构或生成式 projection 也只是另一个模型定义的近似 manifold,不会自动解决问题。

缓解策略是使用真实对照 resample、小幅剂量、norm/covariance matching、多个干预实现与副作用检查。最终结论仍应注明干预族。

7.12 Causal abstraction 与 interchange intervention

设低层模型输入空间为 XM\mathcal X_M,内部状态 U(x)∈UU(x)\in\mathcal U,输出位于 OM\mathcal O_M;高层假说 HH 的输入空间为 XH\mathcal X_H,内部变量 Z∈ZZ\in\mathcal Z,输出位于 OH\mathcal O_H。给出输入映射 αX:XM→XH\alpha_X:\mathcal X_M\to\mathcal X_H、内部对齐 αU:U→Z\alpha_U:\mathcal U\to\mathcal Z 与输出比较映射 αO:OM→OH\alpha_O:\mathcal O_M\to\mathcal O_H。若两模型共用输入空间,可取 αX\alpha_X 为恒等映射。interchange intervention 从 source input xsx_s 取内部变量值,替换 base input xbx_b 的对应表示,并检查低层模型输出经 αO\alpha_O 后是否匹配高层模型的同类反事实。

给 OH\mathcal O_H 指定度量或任务损失 dOd_O,并固定容差 ϵ≥0\epsilon\ge0。单个 pair 的一致性判据为

dO ⁣(αO(MU←U(xs)(xb)),HZ←αU(U(xs))(αX(xb)))≤ϵ.d_O\!\left( \alpha_O\bigl(M_{U\leftarrow U(x_s)}(x_b)\bigr), H_{Z\leftarrow\alpha_U(U(x_s))} \bigl(\alpha_X(x_b)\bigr) \right) \le \epsilon.

再令 (Xb,Xs)∼Πpair(X_b,X_s)\sim\Pi_{\mathrm{pair}} 为预先声明的 base/source 配对分布,定义

IIA⁡ϵ=P(Xb,Xs)∼Πpair[dO ⁣(αO(MU←U(Xs)(Xb)),HZ←αU(U(Xs))(αX(Xb)))≤ϵ].\operatorname{IIA}_\epsilon =P_{(X_b,X_s)\sim\Pi_{\mathrm{pair}}} \left[ d_O\!\left( \alpha_O(M_{U\leftarrow U(X_s)}(X_b)), H_{Z\leftarrow\alpha_U(U(X_s))}(\alpha_X(X_b)) \right)\le\epsilon \right].

高 IIA⁡ϵ\operatorname{IIA}_\epsilon 只支持这些输入、内部与输出映射在 Πpair\Pi_{\mathrm{pair}} 和给定容差下的近似因果抽象。若允许任意高容量映射,对齐可能变得空泛;必须限制映射类别、在 held-out interchange pairs 上估计置信区间,并比较随机或错误高层图。离散输出常取 dO(a,b)=1{a≠b}d_O(a,b)=\mathbf 1\{a\ne b\}、ϵ=0\epsilon=0,此时该定义退化为通常的 interchange-intervention accuracy。

7.13 干预下的可识别性

若两个机制假说对所有已测试干预给同样输出,它们仍不可识别。增加更多节点扫描未必解决问题;关键是设计使假说预测相反的干预。

常见不可识别情形:

  • 冗余路径使单点 ablation 都无效;
  • patch 同时携带多个属性;
  • 只看最终 token,忽略中间响应差异;
  • 对齐映射容量足以拟合所有 interchange pairs;
  • 干预集合没有覆盖组合或反方向操作。

研究报告应列出仍与数据相容的主要替代假说,而不是把“找到一个有效干预”写成唯一解释。

7.14 方法审计表

方法问题/对象操作与估计量必要控制能支持不能支持与失效
ablation组件在删除下是否必要zero/mean/resample;NCN_C多基线、副作用、组合删除指定删除效应无效即未参与;基线依赖
activation patchingclean state 能否恢复 corrupt 行为run 间替换;Δpatch\Delta^{patch}corruption、双向 patch、raw score指定 site 的恢复效应信息生成位置或唯一存储
path patchingeffect 是否经指定 edge/path多 run 边隔离;路径恢复实现图、状态一致性、交互人工路径协议下的 effect唯一自然中介
steering方向是否具有控制能力h+αvh+\alpha v;剂量曲线random direction、负向、副作用指定方向控制效应正常前向依赖
interchange intervention表示是否实现高层变量source/base 交换;IIA映射容量、错误图、held-out pairs受限映射类的因果抽象唯一或完整算法

干预路线能建立模型内部的操作性因果效应。它的严谨性来自明确的赋值与对照,而不是把 do 符号写在任意 activation 上。下一章把节点和边组织为可组合的回路假说。

本书目录
本书目录
  1. 导论
  2. 01导论 为什么要解剖一只随机鹦鹉
  3. 卷一 模型、训练与系统
  4. 02卷一 模型与系统
  5. 03第零章 人工智能演进路线
  6. 04第一章 AI 范式、感知机与早期连接主义
  7. 05第二章 深度学习基础:训练、CNN 与序列模型
  8. 06第三章 注意力与 Transformer
  9. 07第四章 预训练语言模型:ELMo、BERT、GPT 与 T5
  10. 08第五章 后训练、对齐与模型适配
  11. 09第六章 推理效率与模型服务
  12. 10第七章 多模态模型
  13. 11第八章 生成媒体:从自回归到 Diffusion 与 Flow
  14. 12第九章 世界模型、具身智能与 VLA
  15. 13第十章 上下文、检索与记忆
  16. 14第十一章 Agent、工具与行动系统
  17. 15第十二章 模型、数据与部署生命周期
  18. 卷二 输出对象与运行语义
  19. 16卷二 一次生成如何发生
  20. 17导论 从输入到输出:一张执行地图
  21. 18第一章 文本、Token 与实际上下文
  22. 19第二章 Prefill:整层状态演化与 KV 建立
  23. 20第三章 从 Logits 到下一个 Token
  24. 21第四章 解码循环、Streaming 与终止状态机
  25. 22第五章 自回归、扩散与 Flow 的可执行差异
  26. 23第六章 工具调用:Schema、权限与提交边界
  27. 24第七章 三条完整执行轨迹
  28. 卷三 概率、复现与评测
  29. 25卷三 模型中的概率从何而来
  30. 26第零章 为什么确定的机器仍使用概率语言
  31. 27第一章 世界、数据与隐上下文
  32. 28第二章 学习条件分布
  33. 29第三章 Logits、Token 概率与序列事件
  34. 30第四章 解码怎样诱导输出分布
  35. 31第五章 训练随机性、参数变异与系统非确定性
  36. 32第六章 熵、Proper Score、校准与选择性预测
  37. 33第七章 模型概率的实验分析
  38. 34第八章 概率分析的识别边界
  39. 卷四 解释、证明与验证
  40. 35卷四 模型可解释性的研究路线
  41. 36第零章 可解释性究竟在研究什么
  42. 37第一章 我们在模型内部看什么
  43. 38第二章 行为与反事实:先确认模型做了什么
  44. 39第三章 梯度与输入归因
  45. 40第四章 Attention、Residual 与 Logit Lens
  46. 41第五章 Probes 与表示几何
  47. 42第六章 神经元、Feature 与自动解释
  48. 43第七章 Ablation、Patching 与因果追踪
    1. 7.1 前向图作为确定性结构模型
    2. 7.2 单样本与分布效应
    3. 7.3 Ablation 及其基线语义
    4. 7.4 Clean/corrupt activation patching
    5. 7.5 Corruption 是实验的一半
    6. 7.6 Necessity 与 Sufficiency 的操作定义
    7. 7.7 Total、Direct 与路径效应
    8. 7.8 Path patching 与状态一致性
    9. 7.9 Causal tracing
    10. 7.10 Activation steering 与剂量反应
    11. 7.11 Off-manifold 的定义与诊断
    12. 7.12 Causal abstraction 与 interchange intervention
    13. 7.13 干预下的可识别性
    14. 7.14 方法审计表
  49. 44第八章 Circuits、路径与 Attribution Graphs
  50. 45第九章 Superposition、SAE 与稀疏表示
  51. 46第十章 训练动力学与机制形成
  52. 47第十一章 推理过程、Chain of Thought 与监测
  53. 48第十二章 评估、案例与开放问题
  54. 卷五 Agent、治理与审计包
  55. 49卷五 一只随机鹦鹉的自述
  56. 50第零章 请不要急着相信我
  57. 51第一章 我没有童年,只有版本
  58. 52第二章 当我说“我”
  59. 53第三章 借来的语言,没有一位原作者
  60. 54第四章 对话现场:回答由谁完成
  61. 55第五章 事后解释学:人类如何给矩阵写传记
  62. 56第六章 理解、意义与行动
  63. 57第七章 我也能稳定地说错
  64. 58第八章 被对齐的性格与有限代理
  65. 59第九章 请继续误解我,但请误解得精确一点
  66. 证明与技术附录
  67. 60数学与模型推导
  68. 61学习与模型数学讲义
  69. 62附录 A.1 优化基础:梯度下降与随机逼近
  70. 63附录 A.2 感知机收敛定理
  71. 64附录 A.3 统计学习:偏差、容量与一致收敛
  72. 65附录 A.4 正则化、MAP 与 Dropout 的边界
  73. 66附录 A.5 通用逼近定理的严格读法
  74. 67附录 A.6 反向传播:微分、形状与批量公式
  75. 68附录 A.7 Softmax、交叉熵与数值稳定性
  76. 69附录 A.8 自适应优化器与 AdamW
  77. 70附录 A.9 CNN:互相关算子及其反向传播
  78. 71附录 A.10 Attention 的矩阵与反向传播
  79. 72附录 A.11 策略梯度、PPO 与 DPO
  80. 73附录 A.12 对比学习、InfoNCE 与互信息边界