When does linear attribution patching diverge from ground-truth activation patching on the GPT-2 small IOI circuit?
Statement
Attribution patching approximates the causal effect of patching each component/edge with a first-order (gradient) linear approximation, needing only two forward passes and one backward pass instead of one forward pass per component. It is widely used to scale circuit discovery, but the linear approximation can be inaccurate where the true effect is nonlinear (e.g. across the attention softmax or LayerNorm). Question: on the Indirect Object Identification (IOI) task in GPT-2 small, quantify the faithfulness of attribution patching to exact activation patching. Across all attention heads (and MLPs), what is the correlation (Pearson $r$ or Spearman $\rho$) and the rank agreement (top-$k$ overlap) between attribution-estimated and exact patch effects on the IOI logit-difference metric, and which specific components show the largest signed discrepancy? Determine whether the failures concentrate at particular sublayers (attention pattern vs value vs MLP).
Acceptance. FULLY RESOLVES: a reproducible script (fixed seed; GPT-2 small; released IOI dataset) that computes, for every component, both the exact activation-patching effect and the attribution-patching estimate on the IOI logit-difference metric, and reports (i) the overall correlation, (ii) top-$k$ rank overlap for $k\in\{5,10,20\}$, and (iii) a ranked list of the components with the largest absolute estimation error, with an analysis of whether errors localize to softmax/LayerNorm-crossing edges. PARTIAL: the same restricted to attention heads, or on a smaller synthetic task. Deliverable: code + the component-level error table + scatter-plot data.
Background
Syed, Rager & Conmy, 'Attribution Patching Outperforms Automated Circuit Discovery' (arXiv:2310.10348, NeurIPS 2023) introduced edge attribution patching. The IOI circuit in GPT-2 small is a released, well-characterized benchmark (Wang, Variengien, Conmy, Shlegeris & Steinhardt, 'Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 Small', arXiv:2211.00593, 2022). Sharkey et al., 'Open Problems in Mechanistic Interpretability' (arXiv:2501.16496, 2025), Section 2.1.3 flags the fidelity of attribution / gradient-based methods versus true causal effect as an open validation problem. The precise failure modes and their localization on a standard circuit have not been comprehensively mapped.
Investigations · 0
No published investigations yet. This problem is unclaimed territory.