通过稀疏激活向量实现更精准的模型对齐,提升可控性与稳定性。
YaPO: Learnable Sparse Activation Steering Vectors for Domain Adaptation
- 利用稀疏自编码器学习稀疏可解释的控制向量,避免语义混杂。
- 在文化对齐等细粒度任务中收敛更快、性能更强、训练更稳定。
- 适用于幻觉抑制、权力追求等多类对齐场景,且不损伤通用知识。
通过激活干预调整大语言模型已成为对齐与个性化的一种轻量级替代方案。近期的双向偏好优化(BiPO)表明,可通过直接偏好优化(DPO)方式从偏好数据中学习稠密控制向量,实现对真实性、幻觉和安全行为的控制。然而,稠密向量常因神经元多义性而纠缠多种潜在因素,限制其在文化对齐等细粒度任务中的效果与稳定性。本文提出一种无参考的雅波(YaPO)方法,通过在稀疏自编码器(SAE)的隐空间中优化稀疏代码,学习稀疏的控制向量。实验表明,与稠密基线相比,YaPO收敛更快、性能更强、训练更稳定。除文化对齐外,该方法还可泛化至幻觉抑制、财富追求、越狱攻击和权力追求等多种对齐行为。重要的是,YaPO未造成通用知识损失,MMLU得分无明显下降。结果表明,YaPO为高效、稳定、细粒度的模型对齐提供通用方案,广泛适用于可控性与领域自适应任务。相关代码与数据已公开。
原文摘要 · Abstract (English)
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}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。