返回知识盲盒

知识盲盒

第八章 Circuits、路径与 Attribution Graphs

ai-generatedstochastic-parrotstochastic-parrot-anatomyai

机制解释的核心单位往往不是单个 neuron,而是一组按计算顺序连接的组件。circuit 研究试图找到一个比原模型更小、能预测目标行为及干预响应的计算子图。它是一项有损建模任务,不是从模型中读取唯一线路图。

8.1 相对于任务定义 Circuit

令原计算图为 G=(V,E)G=(V,E),目标行为为 (Deval,S)(\mathcal D_{\mathrm{eval}},S)。候选 circuit

C=(VC,EC),C⊆GC=(V_C,E_C),\qquad C\subseteq G

还必须附带补图如何处理的规则 RR:置零、均值、resample、冻结或由 replacement model 近似。没有 RR,仅给节点集合不能执行。

一个 circuit 可追求:

  • faithfulness:保留 CC 后近似原模型目标行为与干预响应;
  • completeness:CC 覆盖目标相关计算,删除它显著损害行为;
  • minimality:移除内部元素会显著降低前两项;
  • interpretability:节点和边有可检验的功能规则;
  • stability:对样本、阈值、seed 与合理基线稳定。

这些目标冲突,必须报告 Pareto 曲线而非单一“最佳图”。

8.2 Keep 与 Remove 两种评估

设 SM,SC,SbaseS_M,S_C,S_{base} 分别为原模型、只保留 circuit 和基线模型 score。可定义归一化 keep fidelity

Fkeep=Ex∼Deval[SC(x)−Sbase(x)]Ex∼Deval[SM(x)−Sbase(x)],F_{keep} =\frac{\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [S_C(x)-S_{base}(x)]} {\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [S_M(x)-S_{base}(x)]},

该比值只在分母远离零且 score 方向一致时有稳定含义,应同时报告原始差与逐样本分布。删除 circuit 的损伤为

Fremove=Ex∼Deval[SM(x)−SM∖C(x)].F_{remove} =\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [S_M(x)-S_{M\setminus C}(x)].

keep 高说明 CC 在所用补图规则下较充分;remove 大支持必要性。二者可能同时不高,因为冗余路径、非线性交互或补图替换不自然。

“completeness”在文献中命名并不统一,论文应直接给操作定义而不是只报词。

8.3 QK/OV Composition

attention head 的 QK 决定选择,OV 决定读写。跨层 composition 可分为:

  • Q-composition:前一组件写入后层 query 使用的方向;
  • K-composition:写入后层 key 标记;
  • V-composition:写入被后层 value 读取;
  • residual/logit composition:多个模块沿共同输出方向相加或抵消。

矩阵乘积可识别潜在通路,activation 与 path intervention 才说明具体输入上启用。权重层的高 composition score 不应单独命名为行为 circuit。

8.4 Induction head 作为方法范例

在 ... A B ... A 中,一类 induction 机制可分解为:

  1. 上游组件把 previous-token 信息写到当前位置;
  2. induction head 在当前 A 查询过去“前一个 token 是 A”的位置;
  3. OV 路径复制过去配对中的 B 信息;
  4. residual 写入提高 B 的 logit。

这个案例的价值在于四类证据汇合:参数组合、行为签名、activation pattern 与 ablation。它不推出自然语言模型的全部 in-context learning 都由同一回路实现。

训练与规模研究中即使同类算法复现,具体 head 编号也可能改变;应比较功能角色而非坐标身份。

8.5 Edge 与 path patching

节点 patch 改变 sender 的全部下游影响。edge patch 只把 sender uu 对 receiver vv 的输入从 clean 替换进 corrupt run。对 path P=(u0,…,uk)P=(u_0,\ldots,u_k),连续隔离其传递。

路径 effect 不是天然可加 credit。若两条路径在 receiver 非线性交互,分别 patch 的和不等于联合 patch。至少报告:

  • 单边、联合边与完整节点 effects;
  • patch 顺序;
  • residual identity 是否算 edge;
  • attention pattern 是否冻结;
  • 其他 sender 输入来自哪一个 run。

8.6 Causal scrubbing 与高层假说

给定高层计算图 HH 以及模型节点到高层变量的映射,causal scrubbing 对 HH 认为无关的输入差异重采样。如果 scrub 后目标行为保持,说明模型对这些变化的不变性与 HH 相容。

该方法检验的是假说充分性,而不是发现唯一图。结果依:

  • 高层变量粒度;
  • 节点—变量映射;
  • resampling distribution;
  • 保留行为的 metric;
  • 对齐映射容量。

应加入“错误但复杂度相同”的高层图作负对照,并在 held-out intervention pairs 上比较。

8.7 自动 Circuit Discovery

自动方法大致分为:

  1. 逐节点/边 patch:精确但前向次数高;
  2. gradient attribution:用一阶近似筛选 edge;
  3. mask optimization:学习稀疏门控使目标行为保留;
  4. 贪心搜索:从输出向上游加入高 effect 边;
  5. 替代模型图:在 SAE/transcoder features 上构图。

所有方法都包含搜索超参数。阈值、稀疏正则和候选单位改变图;必须在独立验证集运行真实 ablation/patch,而不是用同一 attribution score 既发现又证明。

梯度筛选受饱和和局部线性化影响,mask optimization 可能找到对训练 prompts 特化的子图,贪心法会遗漏只有联合加入才有作用的边。

8.8 Faithfulness、Completeness 与 Minimality

对目标距离 dd,行为保真误差为

Lbeh(C)=Ex∼Deval[d(FC(x),FM(x))].L_{beh}(C)=\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [d(F_C(x),F_M(x))].

令 Π(⋅∣x)\Pi(\cdot\mid x) 为给定输入后的干预核,更强的响应误差为

Lint(C)=Ex∼DevalI∼Π(⋅∣x)[d(FC(IC,x),FM(IM,x))].L_{int}(C) =\mathbb E_{\substack{x\sim\mathcal D_{\mathrm{eval}}\\ I\sim\Pi(\cdot\mid x)}} [d(F_C(I_C,x),F_M(I_M,x))].

若干预与输入独立,Π(⋅∣x)\Pi(\cdot\mid x) 退化为常数核。该联合分布是 circuit faithfulness 主张的一部分,不能只列干预名称而不列抽样规则。

minimality 可用删除每条边后的边际损失分布评估:

me=Lbeh(C∖{e})−Lbeh(C).m_e=L_{beh}(C\setminus\{e\})-L_{beh}(C).

若大量 me≈0m_e\approx0,图可能不是最小;但冗余边会使单边 mem_e 小而联合删除重要。还需组合或 group ablation。

8.9 Replacement model 与误差节点

原 MLP 或 residual 可由稀疏替代模型近似。设原模块输出 yy,替代输出

y^=∑kfkdk,ε=y−y^.\hat y=\sum_k f_kd_k, \qquad \varepsilon=y-\hat y.

attribution graph 若忽略 ε\varepsilon,就把未解释计算静默删除。更诚实的图把 reconstruction error 作为显式节点,并测它对目标与干预的 effect。

替代模型质量至少包含:

  • activation reconstruction;
  • 原模型 loss 增量与 argmax agreement;
  • 目标行为保留;
  • 原/替代模型的 intervention response;
  • error nodes 的正负贡献。

自然输出匹配不保证替代机制匹配。

8.10 Attribution graphs

对固定 prompt,可把 input features、内部 sparse features、attention interactions 与 logits 作为节点,用局部 attribution 建边并按目标 effect 剪枝。这类图擅长展示一个运行中的候选信息流。

需要声明近似:

  • attention pattern 是否冻结;
  • feature interactions 是否只取一阶项;
  • cross-layer transcoder 重构哪些模块;
  • 剪枝按绝对 effect、正 effect 还是路径累积;
  • error term 怎样进入;
  • 图只对一个 token 还是整段生成有效。

2025 年公开的 Circuit Tracing 方法报告展示了大型模型局部 attribution graph 的可扩展构造;其后续 attention QK 分析也仍以替代模型和局部图为对象。replacement error、动态 attention 与局部化限制没有因此消失,不能将这些结果写成全模型已经完成解剖。

8.11 Local 到 Global

局部图固定大量非线性状态,较容易稀疏。全局 circuit 要解释 feature 何时触发、QK 路由怎样随输入变化以及备份路径何时接管。

合并局部图不能简单取并集。可定义 edge 的条件出现率

pe(c)=Pr⁡x∼Deval[e∈C(x)∣c(x)=c]p_e(c)=\Pr_{x\sim\mathcal D_{\mathrm{eval}}} [e\in C(x)\mid c(x)=c]

和 effect 分布,进一步学习可检验的 gating rule。若边在不同 context 下改变符号,平均 edge weight 没有明确机制含义。

全局化需要新样本上的图预测、条件规则和失败簇,不是更多可视化。

8.12 阈值、搜索与不确定性

一个 circuit 应随稀疏阈值形成路径:

(∣EC(λ)∣,Lbeh(Cλ),Lint(Cλ)).(|E_C(\lambda)|,L_{beh}(C_\lambda),L_{int}(C_\lambda)).

报告整条曲线可以显示结论是否依一个任意阈值。还应对数据 bootstrap、patch baseline、attribution method 和随机 seed 重复搜索。

边稳定性可定义为被选择频率,但高频不等于真实;它只说明在所采样分析扰动下可重复。选择频率低的边可能属于多组等价冗余路径。

8.13 回路非唯一与可识别性

不同节点基、补图规则和误差容忍可得到多个行为等价 circuit。若 C1,C2C_1,C_2 在测试分布与干预族上满足

Lbeh(C1)≈Lbeh(C2),Lint(C1)≈Lint(C2),L_{beh}(C_1)\approx L_{beh}(C_2), \qquad L_{int}(C_1)\approx L_{int}(C_2),

就没有证据称其中一个是唯一真实图。更合理的产物是:

  • 共享的稳定 core;
  • 可互换的冗余 modules;
  • 输入条件决定的 branches;
  • 尚由 error nodes 承担的未解释部分。

机制解释可由等价类构成,而不必强求坐标级唯一性。

8.14 方法审计表

方法问题/对象操作与估计量必要控制能支持不能支持与失效
hand-built circuit已知角色能否组合实现行为keep/remove;Fkeep,FremoveF_{keep},F_{remove}补图基线、失败样本、组合删除目标分布上的组合机制唯一全局图
path patchingsender 是否经 receiver 起效edge 隔离;路径 effect状态一致性、交互、双向 patch指定协议的路径效应自然中介唯一性
causal scrubbing高层图的不变性是否匹配模型条件 resample;行为保留错误图、映射容量、held-out假说相容性排除所有替代图
automated discovery能否高效定位稀疏子图attribution/mask/search独立真实 patch、阈值曲线候选 circuit 与效率搜索 score 即忠实性
attribution graph局部 feature 图怎样影响 logitreplacement + edge attributionerror nodes、attention 近似局部替代机制原模型全局完整解剖

circuit 的合格结论不是“发现模型真正线路图”,而是“在声明的行为分布、节点基和干预协议下,这个较小计算模型以量化误差复现了原模型的部分行为与响应”。

本书目录
本书目录
  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 与因果追踪
  49. 44第八章 Circuits、路径与 Attribution Graphs
    1. 8.1 相对于任务定义 Circuit
    2. 8.2 Keep 与 Remove 两种评估
    3. 8.3 QK/OV Composition
    4. 8.4 Induction head 作为方法范例
    5. 8.5 Edge 与 path patching
    6. 8.6 Causal scrubbing 与高层假说
    7. 8.7 自动 Circuit Discovery
    8. 8.8 Faithfulness、Completeness 与 Minimality
    9. 8.9 Replacement model 与误差节点
    10. 8.10 Attribution graphs
    11. 8.11 Local 到 Global
    12. 8.12 阈值、搜索与不确定性
    13. 8.13 回路非唯一与可识别性
    14. 8.14 方法审计表
  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 与互信息边界