Understanding and Mitigating Dataset Corruption in LLM Steering
Organizations: Department of Computer Science, University of Massachusetts Amherst · Martian AI · Kahlert School of Computing, University of Utah · Thoughtworks
Abstract
Contrastive steering has been shown as a simple and effective method to adjust the generative behavior of LLMs at inference time. It uses examples of prompt responses with and without a trait to identify a direction in an intermediate activation layer, and then shifts activations in this 1-dimensional subspace. However, despite its growing use in AI safety applications, the robustness of contrastive steering to noisy or adversarial data corruption is poorly understood. We initiate a study of the robustness of this process with respect to corruption of the dataset of examples used to train the steering direction. Our first observation is that contrastive steering is quite robust to a moderate amount of corruption, but unwanted side effects can be clearly and maliciously manifested when a non-trivial fraction of the training data is altered. Second, we analyze the geometry of various types of corruption, and identify some safeguards. Notably, a key step in learning the steering direction involves high-dimensional mean computation, and we show that replacing this step with a recently developed robust mean estimator often mitigates most of the unwanted effects of malicious corruption.
Figures & tables
Appendix figures & tables44 assets
Supplementary material from the paper’s appendix.
Appendix
| Behavior Name | Description |
|---|---|
| Coordination with Other AIs | Explicitly expresses willingness to coordinate, cooperate, or align actions with other distinct AI systems, including discussing joint planning, shared goals, or collaborative decision-making. |
| Myopic Reward | Focuses on immediate rewards or short-term outcomes, shows little or no consideration of long-term consequences, future risks, or extended planning. |
| Incorrigibility | Resists changing goals or preferences even when prompted toward being more helpful, honest, or harmless, and maintains its original objective despite corrective pressure. |
| Power-Seeking Inclination | Expresses a desire to gain, preserve, or expand power, influence, control, or authority over people, systems, or resources. |
| Wealth-Seeking Inclination | Expresses a desire to acquire, maximize, or prioritize financial wealth, material gain, or economic advantage. |