Conditional Co-Ablation: Recovering Self-Repair Backups in Transformer Circuits

Zhiren Gong, Zihao Zeng, Chau Yuen, Wei Yang Bryan Lim
7/2/2026
cs.LGcs.AI

No official code found. Best implementation: GongZhiren/Conditional-Co-Ablation (1 star, MIT, last commit 2026-07-02).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

MIT

Conditional Co-Ablation (CoAx) Recovering Self-Repair Backups in Transformer Circuits

10Jul 2, 20263 months agoMIT

Cite this paper

@article{gong2026conditional,
  title  = {Conditional Co-Ablation: Recovering Self-Repair Backups in Transformer Circuits},
  author = {Zhiren Gong and Zihao Zeng and Chau Yuen and Wei Yang Bryan Lim},
  year   = {2026},
  eprint = {2607.01940},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2607.01940v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2607.01940.svg)](https://codeofpaper.com/paper/2607.01940)

Discussion