Search papers, labs, and topics across Lattice.
This paper investigates the robustness of contrastive steering, a technique for controlling LLM generation, against dataset corruption. The authors find that while contrastive steering is initially robust, significant data alteration can introduce unwanted side effects. They propose using a robust mean estimator in place of standard mean computation during steering direction learning to mitigate the impact of malicious corruption.
Replacing standard mean computation with a robust mean estimator can significantly mitigate the negative effects of malicious data corruption on contrastive steering of LLMs.
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.
