Post

Inside the refusal circuit of a large language model

A paper on arXiv in late August dissects how aligned language models refuse harmful requests. Three stages, detection heads, safety neurons and refusal heads, chain together across six LLMs, and scaling that circuit improves attack robustness by 26.5% at a 1.7% accuracy cost.

Alignment today is mostly done by training: a model is shown enough examples of right and wrong behaviour that refusal becomes its default, and nobody can fully say why the refusal works. A paper posted to arXiv on 30 August, by Kuan-Lin Chu, Chung-En Sun and Tsui-Wei Weng, opens the box on that process and argues that refusal is not a vague disposition but a specific circuit, built from identifiable parts, wired the same way across very different models.

The decomposition has three stages. First come the harmful detection heads, attention heads that light up when the input is itself harmful, before anything like a reply is formed. Then come the safety neurons, individual neurons in the residual stream that take that detection signal and mediate it, stabilizing it between layers. Finally come the refusal heads, the output-side heads that translate the stabilized signal into the actual refusal wording. The point of the paper is not just to name these parts but to show they cause one another. When the authors suppressed the upstream detection heads, the downstream refusal behaviour fell apart. When they intervened on the middle neurons, the link between detection and refusal weakened. That chain of causal evidence is what turns a description of correlated activations into a circuit.

What makes the result travel is that the same decomposition recurred across six different large language models and under several adversarial attack settings. Safety alignment is usually treated as a property that each model acquires in its own opaque way, unique to its own training. Finding the same three-stage circuit in model after model suggests there is a shared, recoverable grammar of refusal, not a private solution that has to be rediscovered from scratch every time.

The practical step is deliberately simple. Rather than editing weights wholesale or running another round of reinforcement learning, the authors used weight scaling, an architecture-preserving probe that turns up or down the contribution of the identified circuit. Guided by the circuit map, that scaling improved safety rates under attack by 26.5% across the six models, while dragging accuracy on four standard benchmarks down by only 1.7%. The appeal is the ratio: a large robustness gain bought with a small general-purpose cost, by acting on the parts the map identifies rather than by retraining the whole network.

The limits are real and the authors are upfront about them. A circuit discovered on a particular set of attacks may not generalize to the next way of bypassing a model, and weight scaling is a local intervention, not a complete theory of alignment. Mechanistic maps of this kind are also always provisional, revised as better tools find features the last pass missed.

What the paper contributes, beyond its numbers, is an attitude. Safety has been treated as a training problem, solvable by collecting better examples and scaling up the loss. This work argues that it is also an engineering problem with a diagram attached. If refusal really is a circuit with detectable heads and mediating neurons, then aligning a model can stop being a matter of hoping the training worked and start being a matter of inspecting the wiring, strengthening the parts that matter, and measuring what changed.

This post is licensed under CC BY 4.0 by the author.