Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Zhiren Gong, Zihao Zeng, Chau Yuen, Wei Yang Bryan Lim• 2026

Related benchmarks

TaskDatasetResultRank
Language ModelingWikiText-2 (test)
PPL50
2416
Circuit DiscoveryDuplicate-token
ROC-AUC98
24
Circuit DiscoveryPrevious-token
ROC-AUC68
24
Circuit DiscoveryInduction
ROC-AUC79
24
Language ModelingWikiText-2 (val)
PPL (10% sparsity)21
17
Backup-head discoveryGPT-2-small IOI circuit node-level (eight documented backups)
ROC-AUC91
6
Backup discoveryControlled-redundancy synthetic benchmark 40 independent trials n=100
ROC-AUC0.9
4
Circuit ClusteringGPT-2-small dup-token circuit VS-ACTIVE (pair-level same-circuit)
Cluster AUC97
4
Circuit ClusteringGPT-2-small induction circuit VS-ACTIVE (pair-level same-circuit)
Cluster AUC0.94
4
Circuit ClusteringGPT-2-small prev-token circuit VS-ACTIVE (pair-level same-circuit)
Cluster AUC87
4
Showing 10 of 15 rows

Other info

Follow for update