Conditional Co-Ablation: Recovering Self-Repair Backups in Transformer Circuits
About
Mechanistic interpretability often relies on component-level interventions to discover how a model produces a behavior. This guides attribution, capability knockout, and model pruning downstream to operate by scoring each unit by the effect of ablation in isolation. Such first-order scoring is natural when component importance is additive, but becomes misleading when a transformer self-repairs: after a primary component is removed, a dormant backup can take over, muting the primary's measured effect while the backup itself appears irrelevant on the intact model. We recast this failure as a recovery task, conditional circuit completion, and introduce Conditional Co-Ablation (CoAx), a label-free, output-grounded score that asks how much each remaining unit's ablation effect grows once a primary set has been removed. This conditional growth exposes the second-order interaction that single-unit scores discard. On the GPT-2-small IOI circuit, CoAx raises backup-head recovery from 0.33 to 0.91 ROC-AUC, outperforming all baselines, including self-repair-aware gradient scores (best 0.82); counterfactual patching verifies that the recovered heads causally carry the repair. The same label-free procedure transfers to induction across eight models. Beyond discovery, the recovered backups correct self-repair-masked attribution, identify the components required for capability knockout, and yield repair-aware structured pruning scaling from 124M to 7B. Component importance is therefore not merely an isolated-unit property: in robust circuits, the components that matter can become visible only under the interventions that make them necessary.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText-2 (test) | PPL50 | 2416 | |
| Circuit Discovery | Duplicate-token | ROC-AUC98 | 24 | |
| Circuit Discovery | Previous-token | ROC-AUC68 | 24 | |
| Circuit Discovery | Induction | ROC-AUC79 | 24 | |
| Language Modeling | WikiText-2 (val) | PPL (10% sparsity)21 | 17 | |
| Backup-head discovery | GPT-2-small IOI circuit node-level (eight documented backups) | ROC-AUC91 | 6 | |
| Backup discovery | Controlled-redundancy synthetic benchmark 40 independent trials n=100 | ROC-AUC0.9 | 4 | |
| Circuit Clustering | GPT-2-small dup-token circuit VS-ACTIVE (pair-level same-circuit) | Cluster AUC97 | 4 | |
| Circuit Clustering | GPT-2-small induction circuit VS-ACTIVE (pair-level same-circuit) | Cluster AUC0.94 | 4 | |
| Circuit Clustering | GPT-2-small prev-token circuit VS-ACTIVE (pair-level same-circuit) | Cluster AUC87 | 4 |