Skip to content
The Daily Triptych239 / 365
Ablation as a causal check

A component is neutralised and the model is rerun on the same inputs. What changes supports necessity under that intervention; what survives weakens a purely correlational story.

II · THE IDEA · ARTIFICIAL INTELLIGENCE

Ablation: Reading Function by Removing Parts

interpretability · component ablation · BERT interpretability illusion · 2104.07143

▶ Listen · narrated

A neuron that lights up whenever French appears may not be a part the model needs. Ablation, switching a part off and rerunning the model, tests that claim.

At a glance

What it is
Removing neurons, heads or directions to test causal role
Core move
Compare behaviour with a component present versus absent
Main risk
Apparent importance need not survive a causal check
Key caution
BERT can show an interpretability illusion under ablation

Think of a radio with many small parts. You notice that one part gets warm whenever jazz plays. That is a correlation: the part is busy when the thing you care about happens. It does not tell you the part makes the jazz. Ablation is the test. Unscrew the part and play jazz again. If the music stops, the part was needed. If the music carries on, the warmth was a red herring, and something else was doing the work.

In a neural network the removable parts are neurons, attention heads — the pieces that let one word draw information from another — and particular patterns in the model's internal numbers. Researchers switch one off, run the same sentences through again, and check whether the skill survives. Stories built by watching which parts light up can fail this test outright. Work on the language model BERT showed exactly that: structure that looked like the home of a grammatical concept turned out, once removed, not to be something the model depended on. So ablation is less a way of naming circuits than a way of stress-testing claims about them.

Look closer

  1. Removal is the measurement

    Ablation treats a model like a machine with removable parts. A neuron is zeroed, an attention head is dropped, or a direction in representation space is projected out. The same inputs are run again. Whatever changes in the output is taken as evidence of that part’s role. The method is appealing because the intervention is explicit and the comparison is direct.

  2. Activation is not necessity

    A component can respond strongly to a concept and still be dispensable. Other components may carry the same information, or the model may reroute around the damage. Conversely, a quiet component can matter once removed. Ablation is useful because it can contradict the story told by activations alone; that contradiction is data, not a failed experiment.

  3. The BERT illusion

    Work on BERT showed that directions which appear, by several interpretability criteria, to encode a linguistic concept need not behave as a simple encoding picture would predict under intervention. Erasing or altering those directions does not always remove the model’s use of the concept. The geometry can create an interpretability illusion: the map looks clear; the causal test does not match it.

The story

A trained neural network is a very large collection of numbers. Nobody wrote those numbers by hand: training set them, and nobody knows in advance which of them do which jobs. Interpretability is the effort to find out. The hopeful picture is that small groups of parts — single neurons, or attention heads, the sub-modules that let one word in a sentence pull in information from another — each perform one recognisable step of the model's work. If that picture is right, it can be tested directly. Switch a part off and see what the model can no longer do. That test is called ablation, and its logic is the logic of removal: if the part was needed for a job, the job should fail without it.

The procedure has five steps. First, choose a candidate part: a neuron, an attention head, or a direction in the model's internal numbers — one particular pattern in the values flowing through it. Second, neutralise it. There is more than one way to do this, and the choice matters. You can force the part's output to 0, which deletes its signal entirely. You can replace its output with the average value it takes across other data, which removes its message about this particular input while leaving something typical in its place. You can mask an attention head so it passes nothing on. Or you can subtract out one direction, erasing a single pattern while leaving every other pattern intact. These are different operations, and they can give different answers. Third, run the same inputs through the damaged model that you ran through the intact one. Fourth, compare the two sets of outputs. Fifth — the hard step — decide what the comparison means.

If accuracy on one task falls sharply while other tasks hold steady, that is evidence the removed part was doing something specific to that task. If nothing changes, the part was not necessary — for those inputs, removed in that particular way. Both conclusions are hedged, and the hedges are not politeness. They mark real limits on what the experiment can show.

Networks resist the removal test for a plain reason: they are not machines assembled from parts with one job each. They have far more capacity than their tasks require. They are trained all at once rather than built piece by piece, so nothing forces a function to live in one place. And they are full of duplicate routes to the same answer. Cut one route and another may quietly carry the load, so the output barely moves and the cut looks harmless. Subtract one direction that tracks a concept and the concept may still be recoverable from everything left behind. In each case the experiment reports, truthfully, what the altered model does. Turning that report into a claim about how the original model works is a further step, and it is where mistakes happen.

The clearest warning comes from BERT, a language model whose internal representations have been studied heavily. Researchers found subspaces — sets of directions in the model's internal numbers — that looked, by several standard interpretability tests at once, like the place where a particular piece of grammar was stored. The tests agreed with each other, which is normally reassuring. Then the researchers did the obvious next thing: they intervened on those directions. If the grammar really lived there, erasing the directions should have erased the model's use of it. It did not, at least not reliably. The model went on using the concept. The authors named the pattern an interpretability illusion: several apparently independent criteria can agree about where a concept sits and still be wrong about what the model actually uses.

This is exactly what ablation is for. It is not a complete method for reading out a circuit, and a successful ablation does not prove the removed piece was the circuit. What it does is put pressure on stories built from correlation alone. A neuron that fires on French text, and whose removal leaves the model's French intact, was never a sufficient account of how French is handled — the firing was real, but it was not the mechanism. A head whose removal collapses one narrow behaviour is a candidate worth studying further. A candidate, not a finished explanation: the cut may have severed one link in a larger chain rather than removed the chain itself.

So the discipline is bookkeeping rather than drama. Name the part. State exactly how you switched it off, because zeroing, averaging and subtracting a direction mean different things. Report what changed and what did not, on which inputs, by which measure. Try several related cuts, to see whether an effect is stable or an artefact of one particular incision. And note what remains unknown: how far the BERT illusion generalises to other architectures, including today's larger models, is not settled by that work alone. What the field does know is the shape of the trap. Seeing structure is not the same as testing function, and ablation — used carefully, stated precisely — is how the second gets done.

Why it mattered then

As probes and visualisation tools made a model's internal structure easier to plot, a tidy pattern began to look like an explanation on its own: this part lights up for that concept, so this part must handle it. Ablation arrived as a correction. It asks a different question — not "does this part respond to the concept?" but "can the model still do the job once this part is gone?" The BERT interpretability-illusion results gave that correction its force. They showed that several interpretability signals could all point the same way and still fail the causal test. After that, showing that a part correlates with a concept was no longer enough. You had to show the model needs it.

Why it matters now

Open-weight models — models whose internal numbers anyone can download — put this kind of experiment within reach of anyone who can run one and attach hooks, small pieces of code that read or overwrite values in the middle of a computation. Switching off attention heads, patching activations from one run into another, and subtracting directions are now routine work in mechanistic interpretability. The old temptation survives alongside the new tools: reading a handful of cleanly firing units as a circuit, and treating one successful ablation as a finished account. The earlier caution still holds. An ablation tests whether a part is necessary under one specific way of removing it. It does not describe how the computation is arranged, and illusions of the BERT kind are exactly what a careful causal check exists to catch.

The surprising detail

The unsettling result is not that switching a part off sometimes changes nothing. It is that a set of directions can pass several independent interpretability tests — each suggesting it stores a grammatical concept — and yet the model keeps using the concept after those directions are erased. Agreement between tools felt like confirmation. It was only agreement. Knowing where something appears to sit is not the same as knowing what causes what.

What is disputed

How far the BERT interpretability illusion generalises to other architectures and to larger decoder-only models is not settled by that work alone. Ablation results are also sensitive to how a component is removed — zeroing, mean replacement, and directional projection are not equivalent interventions — so a single null or positive result should be read in light of the method used.

Remember this

Ablation tests whether a part is necessary under one stated way of removing it. On its own it does not prove what the part was for.

Test yourself

A direction in a model’s representation space lights up cleanly for a grammatical feature, and several probes agree it encodes that feature. You project the direction out and the model’s behaviour on the feature barely changes. What should you conclude, and what should you not conclude?

Go deeper

Image: Original diagram, The Daily Triptych. Licence: Original work. Source.

← Back to day 239