Anthropic 提出电路追踪方法:用跨层 transcoder 揭示语言模型的计算图
Circuit Tracing: Revealing Computational Graphs in Language Models
AI 导读
Anthropic 可解释性团队提出一套电路追踪方法,用跨层 transcoder(CLT)把语言模型的 MLP 替换为可解释的替代模型,并据此构建归因图,描述模型在特定提示下产生某个 token 的计算步骤。CLT 的每个特征从某一层的残差流读取输入,并可向其后所有 MLP 层输出,团队在 18 层小模型和 Claude 3.5 Haiku 上训练了 300K 到 30M 规模的特征。归因图节点包括活跃特征、提示 token 嵌入、重建误差和输出 logit,边表示线性归因,团队通过冻结注意力模式与归一化分母保证特征间直接交互线性,并用剪枝把节点数约减少 10 倍而只损失约 20% 的行为解释。
推荐理由
Anthropic 可解释性团队公开了用跨层 transcoder 构建归因图的完整方法,并给出验证与局限,适合关注机制可解释性工程细节的读者。