Neyman Orthogonal Activation Steering
Organizations: Adobe Research · MIT · University of California, Berkeley
Abstract
Activation steering is a popular approach for shifting LLM output toward a desired behavior, and increasingly for generating realistic synthetic data. The standard recipe estimates a steering direction by contrasting LLM activations on prompts with and without a target concept, such as truthfulness. In this paper, we separate the estimand (the steering direction these methods target) from the estimator used to recover it, two notions the literature often conflates. We frame nearly all existing methods as plug-in estimators of the average gradient of an outcome regression, with the binary contrastive label fixed as the outcome. Using this framing, we can immediately incorporate flexible modeling approaches and extend existing methods to allow for continuous outcomes such as log-probabilities or oracle ratings. Flexible ML models that fit the outcome well, however, might not estimate the gradient well, which motivates estimators that target the gradient directly. Drawing on the debiased machine learning literature, we develop a complementary estimator built on the Stein score, the Riesz representer of the gradient functional. Combining the two routes gives an orthogonalized estimator with classical double-robustness properties. We illustrate the framework on activation steering applied to a number of benchmark datasets.
Figures & tables
| Method or construction | Outcome model or assumption | Direction |
|---|---|---|
| CAA / difference of means | Linear, isotropic working covariance | |
| OLS | Linear regression of | |
| LDA | Gaussian classes, shared covariance | |
| PCA-projected difference of means | Isotropic working slope, projected onto PCs | |
| ITI mean-difference direction | Per-head linear, selected attention heads | |
| Logistic probe / ITI probe direction | Logistic, global or per-head |
| TV | KL | Gain (%) | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Model | Target | Unsteered | CAA | Neyman | Unsteered | CAA | Neyman | TV | KL |
| Llama-3.1-8B | Dem | 0.271 | 0.265 | 0.263 | 0.254 | 0.229 | 0.216 | 0.6 | 5.7 |
| Rep | 0.301 | 0.276 | 0.259 | 0.303 | 0.237 | 0.218 | 6.0 | 8.0 | |
| Qwen3-4B-Base | Dem | 0.241 | 0.232 | 0.231 | 0.209 | 0.192 | 0.176 | 0.6 | 8.1 |
| Rep | 0.315 | 0.270 | 0.269 | 0.356 | 0.266 | 0.257 | 0.3 | 3.5 | |
| Qwen3-8B | Dem | 0.354 | 0.308 | 0.284 | 0.725 | 0.442 | 0.323 | 7.8 | 26.9 |
| Model | Unst. | Probe | Stein | CAA | Ney. | Gain |
| Llama-3.1-8B | .492 | .463 | .465 | .459 | .444 | 3.2 |
| Q3-4B-B | .495 | .423 | .334 | .323 | .281 | 12.8 |
| Q3-8B | .475 | .394 | .225 | .298 | .195 | 34.6 |
| Q3-8B-B | .452 | .424 | .350 | .393 | .283 | 28.0 |
| Q3.5-9B | .432 | .367 | .312 | .294 | .271 | 7.9 |
| Q3.5-9B-B | .461 | .415 | .336 | .372 | .312 | 16.0 |
| Linear | KRR | |||||
| Model | Plug | Ney. | Gain | Plug | Ney. | Gain |
| Llama-3.1-8B | .471 | .448 | 4.9 | .469 | .444 | 5.2 |
| Q3-4B-B | .430 | .322 | 25.1 | .432 | .326 | 24.6 |
| Q3-8B | .367 | .213 | 42.0 | .383 | .208 | 45.8 |
| Q3-8B-B | .404 | .324 | 19.8 | .390 | .313 | 19.5 |
| Q3.5-9B | .378 | .288 | 23.8 | .366 | .286 | 21.7 |
| Genre | Prompt | CAA | Neyman | Gain |
|---|---|---|---|---|
| Text (332) | 0.523 | 0.665 | 0.740 | [.044,.106] |
| Code (122) | 0.335 | 0.559 | 0.559 | — |
| Math (46) | 0.409 | 0.499 | 0.499 | — |
| All (500) | 0.466 | 0.624 | 0.673 | [.029,.070] |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Linear | KRR | RFM | |||||||
| Model | Plug | Ney. | Gain | Plug | Ney. | Gain | Plug | Ney. | Gain |
| Llama-3.1-8B | .268 | .260 | 2.8 | .268 | .259 | 3.1 | .268 | .260 | 3.0 |
| Qwen3-4B-Base | .257 | .255 | 0.2 | .256 | .252 | 1.6 | .256 | .252 | 1.7 |
| Qwen3-8B | .331 | .281 | 14.6 | .338 | .276 | 18.2 | .338 | .275 | 18.5 |
| Qwen3-8B-Base | .246 | .237 | 3.6 | .247 | .235 | 4.7 | .246 | .235 | 4.3 |
| Qwen3.5-9B | .295 | .261 | 11.3 | .291 | .269 | 6.9 | .291 | .269 | 7.2 |
| TV | KL | |||
|---|---|---|---|---|
| Dataset | CAA | Neyman | CAA | Neyman |
| Cities | 0.221 | 0.121 | 0.295 | 0.154 |
| Sp-En Trans. | 0.350 | 0.264 | 0.550 | 0.346 |
| Neg Cities | 0.365 | 0.297 | 0.506 | 0.377 |
| Neg Sp-En Trans. | 0.406 | 0.370 | 0.591 | 0.510 |
| Cities Conj. | 0.280 | 0.195 | 0.379 | 0.254 |
Language Model Steering
Language model steering controls a model’s behavior at inference time by intervening in its representations, components, or decoding process.
See all 283 papers