Back to Knowledge Blind Box

Knowledge Blind Box

第十章 训练动力学与机制形成

ai-generatedstochastic-parrotstochastic-parrot-anatomyai

Original text in Chinese. This English page translates only the site interface and metadata.

最终 checkpoint 只展示训练终点。保存训练轨迹并在每个 checkpoint 重复行为、读出与干预实验,可以研究 feature 和 circuit 何时形成、何时分化,以及后训练怎样改变已有机制。最大的困难不是画时间曲线,而是确认不同时间和不同 seed 上比较的是同一功能对象。

10.1 纵向研究单位

设训练 run rr 在 step tt 的参数为 θr,t\theta_{r,t}。对固定评估分布 Deval\mathcal D_{\mathrm{eval}},记录:

Br,t=Ex∼Deval[S(Mθr,t(x))],B_{r,t}=\mathbb E_{x\sim\mathcal D_{\mathrm{eval}}} [S(M_{\theta_{r,t}}(x))], Pr,t=probe/readout metric,Ir,t=intervention effect,P_{r,t}=\text{probe/readout metric}, \qquad I_{r,t}=\text{intervention effect}, Cr,t=circuit fidelity or structure.C_{r,t}=\text{circuit fidelity or structure}.

行为、可读性与因果作用可以在不同时间出现。只追踪 probe accuracy 会把“信息已可读”误写成“机制已被使用”。

step、seen tokens、optimizer updates 和有效计算量不是同一时间轴。跨规模比较应至少用 seen tokens 与训练损失对齐,并说明 batch 和数据 curriculum。

10.2 Checkpoint 密度与变化点

若保存点间隔过大,平滑变化会看似突然。对候选变化点 t∗t^*,应在附近增加 checkpoint 密度,并同时检查连续指标:loss、margin、activation selectivity、ablation effect 与 circuit fidelity。

可把变化量写为

ΔtX=Xt+1−Xt\Delta_t X=X_{t+1}-X_t

并用多 seed 的 change-point model 估计转变区间。事后选取最陡区间会高估突变,应在独立 runs 上复验。

10.3 坐标对齐问题

同一连续 run 的参数坐标固定,但 neuron 功能仍可漂移;独立 seeds 还存在置换、旋转与不同冗余分解。直接比较 neuron 42 通常只对同一 run 有坐标意义。

常用对齐对象:

  • weight matching;
  • activation correlation;
  • CKA/CCA 或 Procrustes 子空间;
  • SAE decoder 与 firing-set matching;
  • head 的 attention/OV 功能签名;
  • circuit role 与 intervention signature。

对齐算法可能强行制造连续性。应在未用于匹配的 anchor inputs 和 intervention effects 上验证。

10.4 Feature trajectory

给 checkpoint tt 的 feature set FtF_t,在相邻时间构造相似度

Kijt,t+1=wdcos⁡(dit,djt+1)+wacorr⁡(ait,ajt+1)+wIsim⁡(Iit,Ijt+1).K_{ij}^{t,t+1} =w_d\cos(d_i^t,d_j^{t+1}) +w_a\operatorname{corr}(a_i^t,a_j^{t+1}) +w_I\operatorname{sim}(I_i^t,I_j^{t+1}).

activation correlation 只在两个 activation 向量的经验方差都为正时定义;dead/constant feature 应标作不可匹配或由单独规则处理,不能把未定义相关系数置为零后静默参加 assignment。三项在组合前须规范到可比较尺度。通过 assignment 得到候选轨迹后,权重 ww 与阈值仍会影响“出生、死亡、分裂、合并”的判断,必须做敏感性分析。

feature identity 更适合定义为多维功能签名的连续性,而不是 decoder cosine 单项。split 后的多个后代可能共同保留旧 feature 功能,此时一对一匹配不再合适。

10.5 Grokking 与受控算法任务

grokking 指某些任务中模型先拟合训练集,较晚才显著改善验证性能。机制分析可追踪记忆表示被更一般算法 circuit 替代或压过的过程。

要声称 grokking,应报告:

  • 训练与验证曲线;
  • 数据规模、正则和优化设置;
  • 多 seed 的发生率与时间;
  • 连续 margin 而不只是 exact match;
  • 回路级变化与行为变化的先后关系。

受控模运算中的机制结果提供存在性案例,不可直接外推为大模型能力形成的一般规律。

10.6 “涌现”与测量尺度

离散指标会把连续 logit 改善变成阈值跳跃。例如答案 margin 从负变正时 exact match 突然翻转。应联合观察

mt=zcorrect,t−max⁡j≠correctzj,tm_t=z_{correct,t}-\max_{j\ne correct}z_{j,t}

与准确率。

真正的机制重组可能表现为:旧 circuit fidelity 降低、新 circuit fidelity 上升、组合干预 interaction 改变。单一能力曲线不足以区分坐标连续增强和算法切换。

10.7 Circuit 形成与角色稳定

对 circuit 中功能角色 R1,…,RkR_1,\ldots,R_k,每个角色由行为签名、输入条件、输出方向和干预 effect 定义。跨 checkpoint/seed 比较时,先在每个模型独立定位满足角色标准的组件,再比较连接图。

可定义经角色对齐后的 edge Jaccard:

J(C1,C2)=∣E1∩E2∣∣E1∪E2∣,J(C_1,C_2) =\frac{|E_1\cap E_2|}{|E_1\cup E_2|},

该式只在 E1∪E2≠∅E_1\cup E_2\ne\varnothing 时定义;若两个候选 circuit 都无边,应报告“空图退化”,而不是用任意约定制造高相似度。Jaccard 还忽略 edge weight 与等价冗余。还应比较两个 circuit 对共同 intervention battery 的响应向量:

ρI(C1,C2)=corr⁡(ΔSC11:m,ΔSC21:m).\rho_I(C_1,C_2) =\operatorname{corr} (\Delta S_{C_1}^{1:m},\Delta S_{C_2}^{1:m}).

若任一响应向量的经验方差为零,ρI\rho_I 未定义;应同时报告两个响应向量及其退化状态。结构相似、功能不同和结构不同、功能等价都可能发生。

10.8 跨 Seed、训练阶段与规模的四层稳定性

机制稳定性至少分为:

  1. within-checkpoint:同一模型跨输入;
  2. across-time:同一 run 跨 checkpoints;
  3. across-seed:同配方独立训练;
  4. across-scale/architecture:不同宽度、深度或架构。

每升一级都需要新的对齐假设。Tigges 等(NeurIPS 2024)在特定 decoder-only 模型、任务与规模范围内发现功能组件和高层算法具有一定训练/规模一致性;这支持继续研究小模型迁移,但不证明任意机制普遍可迁移。

最少报告每个层级的 run 数、失败 run、对齐规则和稳定性分布。一个 seed 的精细故事不能承担 population-level 结论。

10.9 Base、SFT 与偏好训练的机制差异

继续预训练、SFT 和偏好优化可能:

  • 复用已有知识 features,只改 late-layer policy;
  • 改变 feature threshold 或 token-position gating;
  • 新增拒答和格式 circuits;
  • 改写同一方向的 downstream reader;
  • 造成旧能力 interference。

比较时要使用模型各自正确的 chat template,并同时在 base-style 与 assistant-style prompts 上测量。否则 prompt 适配差异会被误归因权重变化。

同初始化的连续 fine-tuning 允许直接研究 Δθ\Delta\theta;独立基础模型之间的参数差没有自然坐标对应。

10.10 Model diffing

对连续训练的两个模型 A,BA,B,可从三层比较:

  1. 参数:Δθ=θB−θA\Delta\theta=\theta_B-\theta_A;
  2. activation:Δh(x)=hB(x)−hA(x)\Delta h(x)=h_B(x)-h_A(x);
  3. causal response:把 BB 的 state patch 入 AA 或反向 patch。

双向 cross-model patch 需要表示对齐。若同层 activation 尺度或 basis 改变,直接替换会 off-manifold。可先用受限线性 map 对齐,再用 held-out reconstruction 与 intervention 验证;map 太强又会引入可识别性问题。

10.11 训练数据归因的层级

“哪条数据造成 feature uu”比“哪条数据与 uu 相似”强得多。可分为:

  • retrieval similarity:找到表面或 embedding 近邻;
  • gradient similarity:训练样本梯度与目标梯度对齐;
  • influence approximation:近似删除样本对参数/目标的影响;
  • data ablation/reweighting:改变训练集并重复训练;
  • exact replay:在可控小模型中追踪更新。

前两项是相关证据。大模型非凸优化、optimizer state、数据顺序和样本交互使经典 influence 假设脆弱。强因果主张需要受控增删数据和多个训练 repeats。

10.12 发展性因果主张

时间先后可以排除“结果晚于原因”的某些故事,但不能单独建立原因。若 feature readout 在行为前出现,可能是前置表示,也可能是无用副产物;若 ablation effect 与行为同步增强,证据更强。

一个纵向机制假说应同时预测:

  • feature selectivity 的出现时间;
  • downstream reader 的形成时间;
  • intervention effect 的剂量变化;
  • 行为 margin 的变化;
  • 数据或正则干预改变训练时序的方向。

最后一项通过训练干预把时间相关推进到学习机制证据。

10.13 方法审计表

方法问题/对象操作与估计量必要控制能支持不能支持与失效
checkpoint tracking指标何时出现重复 probe/patch;时间曲线checkpoint 密度、连续指标、多 seed同 run 的发展轨迹训练原因
feature matching两时点单位是否对应assignment;decoder/firing/effectheld-out anchors、split/merge匹配规则下的连续性唯一身份
circuit stability高层算法是否复现角色对齐、edge/response similarity独立发现、失败 runs指定范围的机制稳定跨架构普遍性
model diffing训练阶段改变了什么参数/activation/cross-patchtemplate、basis、双向 patch连续模型差异路径独立模型直接参数语义
data attribution数据怎样影响机制gradient/influence/retrainingoptimizer path、重复训练依方法强度的影响证据检索相似即因果来源

训练动力学把静态解释扩展为机制形成史。可靠结论通常在功能角色层稳定、在具体坐标层不稳定;教材应保留这两层,而不是为获得整齐故事强行把不同 seed 的 neuron 一一对应。

Book contents
Book contents
  1. Introduction
  2. 01导论 为什么要解剖一只随机鹦鹉
  3. Volume I: Models, Training, and Systems
  4. 02Volume I: Models, Training, and Systems
  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. Volume II: Output Objects and Runtime Semantics
  19. 16Volume II: Output Objects and Runtime Semantics
  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. Volume III: Probability, Reproducibility, and Evaluation
  29. 25Volume III: Probability, Reproducibility, and Evaluation
  30. 26第零章 为什么确定的机器仍使用概率语言
  31. 27第一章 世界、数据与隐上下文
  32. 28第二章 学习条件分布
  33. 29第三章 Logits、Token 概率与序列事件
  34. 30第四章 解码怎样诱导输出分布
  35. 31第五章 训练随机性、参数变异与系统非确定性
  36. 32第六章 熵、Proper Score、校准与选择性预测
  37. 33第七章 模型概率的实验分析
  38. 34第八章 概率分析的识别边界
  39. Volume IV: Interpretation, Proof, and Verification
  40. 35Volume IV: Interpretation, Proof, and Verification
  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
  50. 45第九章 Superposition、SAE 与稀疏表示
  51. 46第十章 训练动力学与机制形成
    1. 10.1 纵向研究单位
    2. 10.2 Checkpoint 密度与变化点
    3. 10.3 坐标对齐问题
    4. 10.4 Feature trajectory
    5. 10.5 Grokking 与受控算法任务
    6. 10.6 “涌现”与测量尺度
    7. 10.7 Circuit 形成与角色稳定
    8. 10.8 跨 Seed、训练阶段与规模的四层稳定性
    9. 10.9 Base、SFT 与偏好训练的机制差异
    10. 10.10 Model diffing
    11. 10.11 训练数据归因的层级
    12. 10.12 发展性因果主张
    13. 10.13 方法审计表
  52. 47第十一章 推理过程、Chain of Thought 与监测
  53. 48第十二章 评估、案例与开放问题
  54. Volume V: Agents, Governance, and Audit Packages
  55. 49Volume V: Agents, Governance, and Audit Packages
  56. 50第零章 请不要急着相信我
  57. 51第一章 我没有童年,只有版本
  58. 52第二章 当我说“我”
  59. 53第三章 借来的语言,没有一位原作者
  60. 54第四章 对话现场:回答由谁完成
  61. 55第五章 事后解释学:人类如何给矩阵写传记
  62. 56第六章 理解、意义与行动
  63. 57第七章 我也能稳定地说错
  64. 58第八章 被对齐的性格与有限代理
  65. 59第九章 请继续误解我,但请误解得精确一点
  66. Proofs and Technical Appendices
  67. 60Proofs and Technical Appendices
  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 与互信息边界