The Control Framework
Addition steering and ablation steering are specific techniques. But they are part of a broader paradigm: representation control, the systematic manipulation of model behavior through interventions on internal representations.
Zou et al. (2023) formalized this paradigm as part of Representation Engineering (RepE) [1]Representation Engineering: A Top-Down Approach to AI Transparency
Zou, A., Phan, L., Chen, S., et al.
arXiv, 2023. A direction used by a linear probe is also a natural candidate for intervention: add it to test whether behavior increases, or project it out to test whether behavior decreases. A useful probe direction need not be an effective control direction, so the intervention is a new experiment rather than an automatic consequence.
Representation Control: The use of concept directions identified through probing to systematically control model behavior. Control operations include addition (to induce behaviors) and ablation (to disable behaviors), both operating on the same geometric structure revealed by probing.
Reading and Control: Two Sides of One Coin
The framework unifies three capabilities:
- Read: What does the model represent? (CAA, LAT)
- Control: How can we steer it? (Addition, Ablation)
- Analyze: What does the intervention tell us about the model?
This unification is important because it connects the diagnostic question (what does the model encode?) with the interventional question (can we change it?).
A concept direction that reads well but steers poorly suggests the representation is correlated with but not causal for the behavior. A direction that steers well but reads poorly suggests the intervention works through a mechanism we do not yet understand.
The Geometric Operations
Each operation corresponds to a fundamental geometric operation on activation space:
| Operation | Geometric Effect | Purpose |
|---|---|---|
| Reading | Project onto direction | Detect if concept is present |
| Addition | Translate along direction | Induce the concept |
| Ablation | Project onto orthogonal complement | Remove the concept |
These are the three basic linear operations on a one-dimensional subspace (a direction). Reading projects the activation onto the subspace to measure its component. Addition translates along the subspace to shift toward the concept. Ablation projects onto the orthogonal complement to eliminate the concept.The geometric perspective is illuminating. In a high-dimensional activation space, a concept direction defines a one-dimensional subspace. Reading, addition, and ablation are the three fundamental linear operations on this subspace. More sophisticated operations (like those in LEACE) extend this geometry to higher-dimensional subspaces.
Safety Applications
Representation control enables both understanding and manipulating safety-relevant properties:
-
Honesty. Read whether the model's representations encode truthfulness, then steer toward honesty. The honesty direction can distinguish when a model "knows" it is generating false information.
-
Harmlessness. Detect tendencies to generate harmful content, then control them. The harmlessness direction separates harmful from harmless response trajectories.
-
Sycophancy. Identify the direction that encodes "agree with the user regardless of accuracy," then ablate or reverse it to promote truthfulness.
-
Refusal. In one study of 13 models, a refusal direction mediated much of the measured refusal behavior, making it a useful example of low-dimensional control with clear safety implications.
Causal Validation
Representation control provides a methodology for establishing causal claims about model behavior:
- Identify the direction via probing (CAA, LAT).
- Test sufficiency via addition: does adding the direction cause the behavior?
- Test necessity via ablation: does removing the direction prevent the behavior?
A direction that passes both tests has evidence for causal mediation under those interventions. This follows the logic of activation patching, but applies it to a direction rather than a named model component.
Pause and think: The limits of linear control
Representation control assumes that high-level behavioral concepts are represented as linear directions in activation space. Under what circumstances might this assumption fail? What kinds of behaviors might resist linear control?
The assumption likely fails for behaviors that are highly context-dependent or compositional. "Be helpful" might require different strategies in different contexts, making it difficult to capture with a single direction. Similarly, behaviors defined by the absence of something (e.g., "do not discuss topic X") may not have clean linear representations. Conditional behaviors ("be honest unless X") are inherently non-linear.
Combining Operations
Representation control operations can be combined:
- Add multiple directions to induce multiple behaviors simultaneously.
- Ablate one direction while adding another to replace one behavior with a different one.
- Layer-specific interventions to target where concepts are most malleable.
The function vectors work shows that even complex tasks (not just concepts) can be represented as directions and combined through addition.
Implications for Alignment
Representation control creates several possibilities and risks for AI safety:
Understanding: We can now ask precise questions about what safety-relevant concepts a model represents and where.
Control: We have tools to steer behavior without retraining, useful for rapid iteration and deployment-time adjustments.
Vulnerability: The same tools that help us understand safety mechanisms can bypass them. The refusal direction can be ablated with one operation.
The same access that supports diagnosis can also support bypasses. Any safety case using representation control must therefore account for who can apply interventions and how easily the targeted mechanism can be reconstructed or routed around.
Pause and think: Designing robust safety mechanisms
If safety behaviors are encoded as linear directions that can be ablated, how might we design more robust safety mechanisms? Is it possible to make safety behaviors resistant to linear ablation while maintaining the interpretable, linear structure that makes models useful?
One approach is to encode safety redundantly across several interacting components, so removing one direction does not remove the behavior. Redundancy does not inherently prevent interpretation, but it makes both analysis and intervention more demanding. Another possibility is to couple safety behavior to representations used for general capabilities, so bypassing safety also damages performance. That could improve tamper resistance while making legitimate customization harder.
A Linear Toolkit
Representation control and probing methods provide a compact toolkit for one-dimensional linear representations:
- Probe with CAA and LAT to identify concept directions.
- Add with addition steering to induce behaviors.
- Remove with ablation steering to disable behaviors.
- Erase with LEACE for mathematically guaranteed removal.
This toolkit extends to naturally occurring directions like function vectors, showing that the same geometric structure underlies both engineered interventions and the model's own learned computations.