权重空间消融下的跨层交互:闭式注意力Jacobian界及Qwen2.5验证

Cross-Layer Interaction under Weight-Space Ablation: A Closed-Form Attention Jacobian Bound and a Test on a Real Pretrained Model

精选理由

这篇论文把权重空间消融的分析从单层推到了多层,还在Qwen2.5上验证了闭式注意力界,顺手找出了一个涌现电路,挺有意思。

AI 摘要

该论文将激活修补与权重空间消融的一致性分析从单残差块扩展到多层,并精确分离出跨层余项。作者为注意力子块导出了闭式Jacobian界,在Qwen2.5-1.5B-Instruct真实权重上验证,未出现一次违反。他们还在该模型上发现了一个涌现的间接目标识别电路,五个测试实例中三个表现出可测量的非零交互。论文同时补全了前作未展示的曲率常数。

原文 · arXiv cs.LG

Cross-Layer Interaction under Weight-Space Ablation: A Closed-Form Attention Jacobian Bound and a Test on a Real Pretrained Model

A companion paper studies when activation patching and weight-space ablation agree, inside an idealized model where a conditional computation is carried additively through a residual stream. For the one composition in that model where two carriers are architecturally dependent, an attention head and its own layer's normalization-MLP composition, it derives an exact first-order interaction formula, zero when only the MLP is ablated and second-order bounded when the head is also ablated. That result is confined to a single residual block and checked only on small transformers on a synthetic task. This paper extends the result past both limits. First, the interaction from ablating carriers spanning several layers decomposes exactly into same-block terms, one per touched layer, plus a cross-layer remainder on which the decomposition makes no claim of smallness. Second, we isolate that remainder exactly, for two layers, as a double integral of a mixed second derivative, and name the missing ingredient needed to bound it: a Jacobian bound for the attention sub-block. We derive this bound in closed form and verify it, without a single violation, against Qwen2.5-1.5B-Instruct's real weights, though we do not yet chain it across layers. We also give, in closed form, the curvature constant the companion paper's bound leaves unexhibited. Third, on that same model, we search for and find an emergent circuit for indirect object identification, never designed into it, using the original activation-patching method for this task, and test collapse, dissociation, and interaction on it. The result is mixed: a shared carrier emerges across all five tested instances, collapse and dissociation hold on most but not all, and a nonzero interaction is measurable on three of five, at layer pairs outside the same-block case the companion theorem covers.