提出新方法提升语言模型的可控生成能力,效果接近提示词但更省参数。
Improved Representation Steering for Language Models
- 用双向偏好优化实现无参考的概念引导与抑制
- 在Gemma模型上优于现有方法,27B版本逼近提示词效果
- 兼具可解释性与抗攻击能力,适合需要安全可控的场景
语言模型的控制方法旨在通过调整输入、权重或表示来实现细粒度且可解释的生成调控。近期研究发现,相比提示词,调整权重或表示的效果较差,尤其是在引入或抑制特定概念时。本文提出无参考偏好控制(RePS),一种双向偏好优化目标,可同时实现概念引导与抑制。我们在AxBench大规模评测基准上训练并评估了RePS的三种参数化形式。在2B至27B规模的Gemma模型上,RePS显著优于基于语言建模目标训练的现有方法,大幅缩小与提示词方法的差距,同时保持高可解释性并最小化参数量。在抑制任务中,RePS在Gemma-2上表现等同于语言建模目标,在更大版本Gemma-3上则更优,且对基于提示的越狱攻击具有强鲁棒性。结果表明,RePS为概念控制提供了可解释且稳健的替代方案。
原文摘要 · Abstract (English)
Steering methods for language models (LMs) seek to provide fine-grained and interpretable control over model generations by variously changing model inputs, weights, or representations to adjust behavior. Recent work has shown that adjusting weights or representations is often less effective than steering by prompting, for instance when wanting to introduce or suppress a particular concept. We demonstrate how to improve representation steering via our new Reference-free Preference Steering (RePS), a bidirectional preference-optimization objective that jointly does concept steering and suppression. We train three parameterizations of RePS and evaluate them on AxBench, a large-scale model steering benchmark. On Gemma models with sizes ranging from 2B to 27B, RePS outperforms all existing steering methods trained with a language modeling objective and substantially narrows the gap with prompting -- while promoting interpretability and minimizing parameter count. In suppression, RePS matches the language-modeling objective on Gemma-2 and outperforms it on the larger Gemma-3 variants while remaining resilient to prompt-based jailbreaking attacks that defeat prompting. Overall, our results suggest that RePS provides an interpretable and robust alternative to prompting for both steering and suppression.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。