YaPO: Learnable Sparse Activation Steering Vectors for Domain Adaptation
By: Abdelaziz Bounhar , Rania Hossam Elmohamady Elbadry , Hadi Abdine and more
Steering Large Language Models (LLMs) through activation interventions has emerged as a lightweight alternative to fine-tuning for alignment and personalization. Recent work on Bi-directional Preference Optimization (BiPO) shows that dense steering vectors can be learned directly from preference data in a Direct Preference Optimization (DPO) fashion, enabling control over truthfulness, hallucinations, and safety behaviors. However, dense steering vectors often entangle multiple latent factors due to neuron multi-semanticity, limiting their effectiveness and stability in fine-grained settings such as cultural alignment, where closely related values and behaviors (e.g., among Middle Eastern cultures) must be distinguished. In this paper, we propose Yet another Policy Optimization (YaPO), a \textit{reference-free} method that learns \textit{sparse steering vectors} in the latent space of a Sparse Autoencoder (SAE). By optimizing sparse codes, YaPO produces disentangled, interpretable, and efficient steering directions. Empirically, we show that YaPO converges faster, achieves stronger performance, and exhibits improved training stability compared to dense steering baselines. Beyond cultural alignment, YaPO generalizes to a range of alignment-related behaviors, including hallucination, wealth-seeking, jailbreak, and power-seeking. Importantly, YaPO preserves general knowledge, with no measurable degradation on MMLU. Overall, our results show that YaPO provides a general recipe for efficient, stable, and fine-grained alignment of LLMs, with broad applications to controllability and domain adaptation. The associated code and data are publicly available\footnote{https://github.com/MBZUAI-Paris/YaPO}.
Similar Papers
Steering Large Language Model Activations in Sparse Spaces
Machine Learning (CS)
Teaches AI to follow instructions better.
SAE-SSV: Supervised Steering in Sparse Representation Spaces for Reliable Control of Language Models
Computation and Language
Guides AI to follow instructions better.
Textual Steering Vectors Can Improve Visual Understanding in Multimodal Large Language Models
Machine Learning (CS)
Makes AI better at seeing and understanding.