Sparse autoencoders (SAEs) decompose language model activations into interpretable features, but existing methods reveal only which features activate, not which change model outputs when amplified. We introduce Control Reinforcement Learning (CRL), which trains a policy to select SAE features for steering at each token, producing interpretable intervention logs: the learned policy identifies features that change model outputs when amplified. Adaptive Feature Masking encourages diverse feature discovery while preserving singlefeature interpretability. The framework yields new analysis capabilities: branch point tracking locates tokens where feature choice determines output correctness; critic trajectory analysis separates policy limitations from value estimation errors; layer-wise comparison reveals syntactic features in early layers and semantic features in later layers. On Gemma 2 2B across MMLU, BBQ, GSM8K, HarmBench, and XSTest, CRL achieves improvements while providing per-token intervention logs. These results establish learned feature steering as a mechanistic interpretability tool that complements static feature analysis with dynamic intervention probes


翻译:稀疏自编码器(SAE)将语言模型激活分解为可解释特征,但现有方法仅能揭示哪些特征被激活,而无法说明哪些特征在放大时会改变模型输出。我们提出控制强化学习(CRL),该方法训练一个策略以在每个令牌处选择用于引导的SAE特征,并生成可解释的干预日志:学习到的策略能够识别在放大时会改变模型输出的特征。自适应特征掩码在保持单特征可解释性的同时,鼓励多样化的特征发现。该框架提供了新的分析能力:分支点追踪可定位特征选择决定输出正确性的令牌位置;评论家轨迹分析能够区分策略局限性与价值估计误差;分层比较揭示了早期层中的句法特征与深层中的语义特征。在Gemma 2 2B模型上,通过MMLU、BBQ、GSM8K、HarmBench和XSTest等基准测试,CRL在提供逐令牌干预日志的同时实现了性能提升。这些结果表明,学习到的特征引导作为一种机制可解释性工具,能够通过动态干预探测对静态特征分析形成补充。

0
下载
关闭预览

相关内容

《可解释深度强化学习综述》
专知会员服务
40+阅读 · 2025年2月12日
《分布式多智能体强化学习的编码》加州大学等
专知会员服务
55+阅读 · 2022年11月2日
可解释强化学习,Explainable Reinforcement Learning: A Survey
专知会员服务
132+阅读 · 2020年5月14日
基于模型的强化学习综述
专知
42+阅读 · 2022年7月13日
「强化学习可解释性」最新2022综述
专知
12+阅读 · 2022年1月16日
概述自动机器学习(AutoML)
人工智能学家
19+阅读 · 2019年8月11日
TensorFlow 2.0深度强化学习指南
云栖社区
18+阅读 · 2019年2月1日
【干货】深入理解自编码器(附代码实现)
【强化学习】强化学习+深度学习=人工智能
产业智能官
55+阅读 · 2017年8月11日
国家自然科学基金
6+阅读 · 2015年12月31日
国家自然科学基金
42+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
7+阅读 · 2015年12月31日
国家自然科学基金
12+阅读 · 2015年12月31日
国家自然科学基金
23+阅读 · 2009年12月31日
国家自然科学基金
11+阅读 · 2008年12月31日
VIP会员
相关资讯
相关基金
国家自然科学基金
6+阅读 · 2015年12月31日
国家自然科学基金
42+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
7+阅读 · 2015年12月31日
国家自然科学基金
12+阅读 · 2015年12月31日
国家自然科学基金
23+阅读 · 2009年12月31日
国家自然科学基金
11+阅读 · 2008年12月31日
Top
微信扫码咨询专知VIP会员