归纳头:Transformer 上下文学习的第一块机制拼图

归纳头:「学会学」的回路级解释

大模型的少样本能力有一个朴素的描述:给几个例子,模型就能照着做。但「照着做」在计算上意味着什么?2022 年 Anthropic 的《In-context Learning and Induction Heads》(Catherine Olsson、Nelson Elhage、Neel Nanda、Tom Henighan、Tristan Hume 等,Transformer Circuits 系列)第一次给出了精确到注意力头级别的答案:上下文学习在很大程度上由一种叫归纳头(induction heads)的回路实现,而这种回路是在训练中突然形成的。这篇论文是机制可解释性领域被引用最多的工作之一,也是涌现能力争论里「真实相变」一方的关键证据。要理解它,得先回到这个系列的起点。

从数学框架到归纳头假说

2021 年,Anthropic 的 Elhage 等人发表《A Mathematical Framework for Transformer Circuits》,为机制分析准备了一套语言:把每个注意力头拆成两部分——QK 回路决定「往哪里看」,OV 回路决定「看了之后做什么」;再定义头与头之间的三种组合方式,即 Q 组合、K 组合与 V 组合:一个头的输出写入残差流后,可以被下游头的 Query、Key 或 Value 读取,两个头由此组成一条「虚拟注意力头」回路。归纳头的假说正是在这个框架里被提出的:当序列中出现过相邻对 A B、而 A 再次出现时,模型应当预测下一个 token 是 B——这就是归纳头要实现的模式补全。

具体机制是两层结构下的 K 组合。浅层有一个「前一个 token 头」:它在当前位置把上一位置的 token 身份写入残差流,充当 Key 的来源。深层的归纳头则用当前位置的 Query 去匹配这条被改写过的 Key,于是注意力被精准投放到「上一次当前 token 出现位置的后一个位置」;OV 回路随后把那个位置的 token 复制成预测。整个过程不需要任何显式监督,纯粹是注意力结构的天然组合:前一个 token 头提供索引,归纳头完成查找与复制。在最简单的两层注意力模型里,这条两头回路就是上下文中「照着例子做」的全部秘密。

训练动力学:一条会隆起的损失曲线

Olsson 等人的实验设计非常干净。他们训练了几十个两层的 attention-only transformer,同时监控一个专门的探针指标:构造 X Y 加省略号再加 X 形式的重复随机序列,测量模型在第二个 X 出现后对 Y 的预测概率,以此作为「归纳分数」。结果清晰得罕见:训练到约 25 亿 token 时,归纳分数陡然上升,与此同时总损失曲线出现一个先变差再变好的凸起——模型先短暂退化,随后整体能力上台阶,这是回路重构的典型动力学特征。他们随后在更大的真实模型上重复测量,发现归纳头大约在 20 亿参数规模形成,而且形成的时间点与四十多项少样本评测成绩的集体跳变相吻合。

这篇论文由此成为第一篇把训练动力学相变与能力涌现连起来的机制研究:如果很多「涌现」能力其实共享同一个形成时刻——归纳回路成形的时刻——那么涌现就不是一堆无法解释的巧合,而是一次机制事件的多个投影。Meta FAIR 的 Bietti 等人在 2023 年 NeurIPS 的《Birth of a Transformer: A Memory Viewpoint》从理论上补齐了时间线:小 transformer 先形成 bigram 记忆回路,再在训练后期形成归纳记忆回路;归纳回路的形成要求训练序列长度至少三个 token,词汇表规模与序列长度共同决定相变时机;基于信号传播的分析能定量预测归纳头何时出现,理论预测与实验观察一致,「归纳头相变」由现象升级为机制结论。

证据链:消融、模糊归纳头与边界条件

因果证据来自消融实验。把归纳头从模型中掐掉之后,模型在多个少样本任务上的成绩显著下滑——这说明归纳头与上下文学习之间不只是相关关系,而是被验证过的因果链条。功能外推的证据来自「模糊归纳头」:真实文本里极少出现逐 token 的精确重复,归纳头之所以有用,在于它泛化出了语义级的模式补全——少样本翻译中复制结构映射、变量绑定任务中复制指代关系,这些行为都由模糊化的归纳回路承担,模型把「精确复制」这个底层操作抽象成了「按模式延续」的通用能力。

边界条件的证据同样重要。已知干扰模式会显著破坏归纳头:给模型重复但无规律的序列,归纳头的预测质量骤降,这构成了上下文学习的可复现脆弱点。同时必须给这个故事划清边界:上下文学习并不完全等于归纳头。Stanford 的 Xie 等人 2022 年提出隐式贝叶斯解释——ICL 是模型对预训练分布中「潜在概念」的贝叶斯推断,提示只是收窄概念后验的证据;Garg 等人 2022 年则证明 transformer 能在上下文中做函数类逼近,例如对稀疏线性回归的目标函数进行隐式拟合,这超出了归纳头的复制能力。比较公允的图景是分工:归纳头负责「格式与模式」层面的照猫画虎,更深层的任务理解由其他回路承担。两者不冲突,而是构成 ICL 的不同层次。

对两个研究方向的持久影响

第一个影响落在涌现之争上。归纳头相变提供了「真实相变」的机制标本:能力突变对应的是回路突然成形,而不是评测假象。值得注意的是,Schaeffer 等人的度量批评并不消解这类证据——在连续指标下,归纳头形成前后的行为差异依然清晰可测。对涌现持怀疑态度的阵营也接受这一点:度量假象解释了大部分报告的涌现曲线,但机制相变解释了剩下那部分真实存在的突变,两者在不同层面各自成立。

第二个影响落在机制可解释性本身。归纳头确立了「先找回路、再谈能力」的研究范式:Wang 等人 2022 年在 GPT-2 small 上完整解剖了间接宾语识别任务(IOI)的回路,识别出 name mover heads 等专门构件;随后的复制抑制、特征叠加与稀疏自编码器工作都沿用这一范式,把「能力归因到具体头与具体回路」变成标准动作。工程层面,归纳分数已经成为诊断模型上下文学习健康度的实用指标:剪枝、量化、蒸馏或继续预训练之后,监测归纳回路是否受损,能提前发现少样本能力的退化,避免等到线上评测才暴雷。开放问题也很明确:混合专家模型与长上下文模型里归纳头的形态是否改变;RLHF 对齐是否会重写归纳回路;归纳头与 grokking 回路能否统一解释;以及能否主动诱导回路更快成形,从而缩短训练早期的低效阶段。

flowchart TD
    A["输入序列:X Y ... A B ... A"] --> B["前一个 token 头:把上一位置的 token 写入 Key"]
    B --> C["归纳头:用当前 token A 的 Query 做 K 组合匹配"]
    C --> D["注意力指向上一次 A 之后的位置"]
    D --> E["OV 回路复制其后续 token B 作为预测"]
    E --> F["效果:模式补全,上下文学习的回路基础"]

后续机制研究:从单回路到回路词典

归纳头之后,回路级分析迅速铺开成一套方法论。Wang 等人 2022 年的《Interpretability in the Wild》在 GPT-2 small 上完整解剖了间接宾语识别(IOI)任务:模型要回答「John 给 Mary 一个苹果,她给了谁」这类问题,背后是重复归纳头、S 抑制头、名字移动头(name mover heads)等多个构件组成的协作回路,其中名字移动头负责把候选名字复制到输出位置。这是机制可解释性第一次把「句子级语义归因」落到具体注意力头上,方法论上正是归纳头组合分析的自然延伸。McDuffie 等人 2023 年对复制抑制头的全解析则展示了硬币的另一面:某个深层注意力头专门负责抑制对上下文内容的重复复制,与归纳头的复制功能互为补充——「复制」与「抑制复制」在模型里是一对被分别回路化的功能,两条研究合起来构成了 ICL 行为的完整微观经济学。

另一条支线解释了为什么回路分析会遇到多义性障碍。Elhage 等人 2022 年的《Toy Models of Superposition》证明:神经网络会用少于神经元数量的维度存储更多特征,多个特征叠加在同一组神经元上——这解释了单个注意力头为何常常多义,也给机制分析划定了分辨率极限。突破来自稀疏自编码器:Bricken 等人 2023 年在小型模型上提取出数百万个单义特征,Templeton 等人 2024 年的《Scaling Monosemanticity》进一步在 Claude 3 Sonnet 上提取约 3400 万个可解释特征,覆盖概念层面的抽象。特征图谱与回路分析由此形成互补:回路回答「信息怎么流动」,特征回答「流动的是什么」。

更 speculative 但影响很大的一步来自 von Oswald 等人 2023 年:他们证明训练好的 transformer 可以在上下文中隐式执行类似梯度下降的参数更新,为「归纳头之上的更高阶上下文学习」给出机制猜想——归纳头负责最外层的模式复制,更深的回路可能实现某种上下文内的学习算法。工程层面,这一整套研究已经落地为回归测试:对模型做量化、剪枝或继续预训练之后,跑一遍归纳分数与 ICL 评测套件,能在基准大盘波动之前发现回路层面的损伤,这大概是机制可解释性目前最直接的生产价值。

小结

归纳头研究示范了机制可解释性的完整闭环:数学框架提出假说,训练动力学捕捉相变,消融实验确立因果,理论工作解释时机。它的核心洞见——上下文学习的很大一部分是序列模式的回路化补全,而这条回路在训练中突然成形——同时服务了两个方向:为涌现争论提供了真实相变的机制证据,把「突然变聪明」从黑箱现象落到了具体注意力头的成形时刻;为工程实践提供了上下文学习健康度的监测手段。对于想进入机制可解释性领域的研究者,Transformer Circuits 系列的这两篇论文至今仍是最好的入门材料。

← 返回资讯列表

读者留言

COMMENTS 暂无
仅本站原创文章开放留言 · 请勿留下手机号、邮箱等个人信息

还没有留言,来说第一句?