arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.16080cs.CLcs.CYcs.HCcs.LG
收藏

通过生成因果中介进行激活引导

Activation Steering via Generative Causal Mediation

CSAIL, MIT(麻省理工学院计算机科学与人工智能实验室);Stanford University(斯坦福大学);Goodfire Research

Aruna Sankaranarayanan, Amir Zur, Atticus Geiger, Dylan Hadfield-Menell

AI总结:

研究提出生成因果中介(GCM)方法,通过对比长形式响应选择模型组件,以局部化和控制扩散行为,如诗歌式 vs. 论文式表达,在三种语言模型上验证其有效性。

摘要:

在语言模型(LM)中,应在哪里干预以局部化和控制跨多个token的扩散行为?我们引入生成因果中介(GCM),一种通过对比长形式响应选择模型组件(如注意力头)来引导此类扩散概念(如诗歌式 vs. 论文式表达)的程序。在GCM中,我们首先构建对比行为输入和长形式响应的数据集。然后量化模型组件如何中介概念并选择最强的中介体进行引导。我们在三种行为——拒绝、阿谀奉承和风格转换——上评估GCM,跨三种语言模型。GCM成功局部化长形式响应中的概念,并在使用稀疏注意力头引导时优于相关性探针基线。这些结果表明,GCM为从LM的长形式响应中局部化和控制提供了有效方法。

英文摘要:

Where should we intervene in a language model (LM) to localize and control behaviors that are diffused across many tokens of a long-form response? We introduce Generative Causal Mediation (GCM), a procedure for selecting model components (e.g., attention heads) from contrastive long-form responses, to steer such diffuse concepts (e.g., talk in verse vs. talk in prose). In GCM, we first construct a dataset of contrasting behavioral inputs and long-form responses. Then, we quantify how model components mediate the concept and select the strongest mediators for steering. We evaluate GCM on three behaviors--refusal, sycophancy, and style transfer--across three language models. GCM successfully localizes concepts expressed in long-form responses and outperforms correlational probe-based baselines when steering with a sparse set of attention heads. Together, these results demonstrate that GCM provides an effective approach for localizing from and controlling the long-form responses of LMs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑