Transformer电路数学框架:残差流与归纳头机制
长期信息 · 《A Mathematical Framework for Transformer Circuits》(2021)
这篇2021年的论文《A Mathematical Framework for Transformer Circuits》由Anthropic团队撰写,旨在通过机械可解释性方法逆向工程Transformer模型。作者将这一过程比作程序员将复杂的二进制文件还原为人类可读的源代码,试图理解模型内部的具体计算逻辑。
为了简化分析,研究聚焦于“仅注意力”(attention-only)的玩具模型,排除了MLP层和层归一化等复杂组件。这种简化使得研究者能够清晰地观察到注意力头之间的交互模式,尽管这被作者承认是当前工作的主要弱点之一,因为真实的大模型包含大量非线性的MLP结构。
核心概念之一是“残差流”(residual stream)。在Transformer中,每一层都通过线性投影读取残差流的信息,并将结果加回残差流。作者强调,残差流具有深刻的线性结构,可以被视为一个通信通道,所有组件都在其中读写不同的子空间。这种视角允许我们将模型分解为端到端的可解释函数,从而更容易推理模型的行为。
另一个关键发现是“归纳头”(induction heads)。研究表明,单层注意力模型只能实现简单的二元语法或跳三元语法,而两层及以上的模型可以通过注意力头的组合形成更复杂的算法。特别是,归纳头仅在至少有两层注意力的模型中出现,它们能够解释小模型中的上下文学习(in-context learning)现象,即模型如何根据前文提示预测后续内容。
对于现代AI开发者而言,理解这些基础机制有助于调试模型行为。例如,当模型表现出意外的复制行为时,可以通过分析OV电路的特征值来检测其是否在进行某种形式的“复制”。虽然该框架目前主要适用于小型模型,但它为理解更大、更现实的模型提供了初步的数学工具和直觉,特别是在识别安全问题和异常行为方面。
建议从事大模型安全与可解释性研究的工程师阅读《A Mathematical Framework for Transformer Circuits》(2021年),从残差流的线性结构和归纳头的形成条件入手,建立对Transformer内部机制的直观理解。
信息源:《A Mathematical Framework for Transformer Circuits》(2021)