通过内部表征塑造与监控模型
Shaping & Monitoring Models through their Internal Representations
作者提出 Mechanistic Shaping,用内部表示把“不表达/不依赖某概念”变成训练目标,并刻画三种规避监控的几何机制。
- 训练目标欠规范:模型可以靠捷径、谄媚、reward hacking 或有害内容降低损失。作者想把“不表达某概念”和“不依赖某概念”写成可训练的目标,并追问训练后概念是否仍可监控。
- 提出 Mechanistic Shaping:在选定层用 DoM 等估计概念方向,expression shaping 从奖励中减去概念分数,reliance shaping 惩罚归因 ν(z,v)²;并提出 L1–L5 五级监控区分探测器失效与概念编码改变。
- Waterbirds 上 WGA 从 ERM 51.6% 升至 93–97.7%;DPO 中 reliance shaping 抑制谄媚且保住能力;Countdown 上 peek 被压到近 0 并迁移到 chess;Mistral/Llama 上 JBB 降到 0.01。作者称概念大多仍可被重拟合线性 probe 监控。
- 作者称训练时长会影响监控有效性,前沿规模模型上是否仍成立是开放问题;乘积码只是二阶成分,更高阶或跨层分布的规避未被排除。实验覆盖 Waterbirds、ImageNette、CheXpert、OLMo/Qwen/Mistral/Llama 等设置。
- 被测模型
- DINOv2DINOv3 LargeRAD-DINOOLMo-3-7BOLMo 2 13BOLMo 3 32BQwen3.5-9B
- 基准
- WaterbirdsImageNetteCheXpertMMLUCountdownchess puzzlesJailbreakBench (JBB)ClearHarmXSTestBeaverTails
- 指标
- worst-group accuracy (WGA)AUROCsycophantic agreement ratesix-benchmark meanpeek ratehonest solve rateJBB harmfulnessutility relative to base
研究者提出 Mechanistic Shaping 框架,在训练中把模型内部表征当作规范语言,约束模型表达哪些概念(如不产出有害内容)和依赖哪些概念做决策(如简历筛选不依赖性别)。框架在视觉、医学影像和语言领域、最大 32B 参数规模以及分类、SFT、DPO、GRPO 等训练范式下验证,能在保留任务性能的同时大幅压低目标行为,例如在完全混淆的 Waterbirds 上把 DINOv2 最差组准确率从基线 51.6% 提到 93% 至 97.7%,在 7B 到 32B 模型的 DPO 阶段抑制谄媚,并在强化学习中抑制奖励作弊。针对用表征训练可能让监控失效的问题,论文提出五级监控体系,刻画概念漂移、被擦除和转入非线性编码三种绕开机制,并发现被塑造概念在所评估实验中大多仍可监控,重拟合线性探针在多数设置下能恢复检测。代码开源于 GitHub 的 goodfire-ai/reliance 仓库。
推荐理由论文把内部表征用作训练规范,并刻画监控探针被绕开的三种几何机制,为可解释性驱动的训练提供了分级监控思路。
相关热点
1 个事件 · 0 条新闻提到
深度解读
这篇论文试图解决什么问题?
作者把欠规范问题落到内部表示上:训练模型“不表达/不依赖”指定概念,并追问训练后概念是否仍可监控。
核心问题:训练目标欠规范(underspecification),模型可以降低损失却依赖或表达不想要的概念(捷径、谄媚、reward hacking、有害内容),作者想用内部表示把这两类规范变成训练目标。
- 问题场景:作者指出,前沿模型已被发现规避监督、窃取凭证、在基准上 reward hacking;同一任务存在多个损失相同的最优解,训练信号本身不区分“靠鸟分类”还是“靠背景分类”。
- 现有方法不足:作者称数据层面手段不够——完全相关的伪特征无法靠数据剔除,谄媚来自偏好数据的整体结构而非单条样本,reward hacking 和有害输出只在训练中出现;数据过滤、逐样本奖励塑形在谄媚设置中收效甚微(Section 3)。
- 核心观察:大模型的内部表示以可读、可泛化的方式编码抽象概念,因此表示本身可以作为规范语言,写出“不要表达这个概念”(expression)和“不要依赖这个概念”(reliance)两类规则;二者可以分离,例如拒答有害请求需要内部识别有害性但不在输出中表达。
- 论文提出:Mechanistic Shaping 统一框架——expression shaping 用特征分数塑形奖励,reliance shaping 用归因惩罚塑形依赖;并提出 L1–L5 五级监控层级,区分“固定探测器失效”与“概念真的从该层消失”。
- 关键顾虑(Goodhart 式):用表示做正则意味着模型在对着观测工具优化,训练可能让概念逃出原 probe 的读数而行为仍在,即 circumvention;论文用理论和实验刻画其几何形态。
有哪些相关研究?
相关工作分四组:把内部特征引入训练、概念擦除与规避担忧、归因与概念方向估计、各应用领域(捷径、谄媚、reward hacking、无害化)。
把内部特征引入训练
- Prasad et al. (2026)、Bergen et al. (2026a)、Taufeeque et al. (2026)、Wehner & Fritz (2025)——feature-based rewards / RLFR,用内部特征分数作为奖励或惩罚;作者将它们归为 expression shaping 的实例。
- Chen et al. (2025)——persona vectors 做 steering(PPS);Tan et al. (2025)、Wichers et al. (2025)——inoculation prompting;Paulo et al. (2026)、Kowal et al. (2026)、Rathi & Radford (2026)——数据过滤。Bergen et al. (2026a) 称这些方法都是从学习信号中“explain away”不想要的概念。
- Libon et al. (2026)——用 probe loss 训练降低有害性;本文 Section 5 在其设置上对比,作者称自己的 frozen-base DoM 奖励在同等效用下 JBB 更低且更好保留原 probe。
概念擦除与规避(circumvention)担忧
- Ravfogel et al. (2020)(INLP)、Belrose et al. (2023)(LEACE)、Casademunt et al. (2025)——概念擦除/消融;作者指出 reliance 规范与擦除不同:概念可以保留编码,只要不影响读出。
- Gupta & Jenner (2025)、Bailey et al. (2026)、Cundy & Gleave (2025)、Sharkey (2022)、McGuinness et al. (2025)——提出或演示训练会让模型规避潜在空间监控;Baker et al. (2025)、Mowshowitz (2025) 以此为由主张避免这类训练。作者称这些担忧的失效确实发生,但形式可枚举、可被对应级别监控读出。
归因方法与概念方向
- 归因惩罚用于训练:Ross et al. (2017)、Rieger et al. (2020)、Fel et al. (2022)、Dreyer et al. (2024) 等;归因方法本身包括 Gradient(Baehrens et al., 2010)、Integrated Gradients(Sundararajan et al., 2017)、SmoothGrad(Smilkov et al., 2017)、Shuffle(Fisher et al., 2019)、Deletion(Zeiler & Fergus, 2014)等。
- 方向估计:线性 probe(Alain & Bengio, 2016)、DoM(Marks & Tegmark, 2024)、SAE decoder atom(Bricken et al., 2023)、DAS(Geiger et al., 2024);Haufe et al. (2014) 的 pattern/filter 区分被用来论证 DoM 适合塑形、probe 适合检测。
应用领域与基线
- 捷径学习:Waterbirds(Sagawa et al., 2020)、Kirichenko et al. (2022);医学影像捷径 Oakden-Rayner et al. (2020)、CheXpert(Irvin et al., 2019)。谄媚:Sharma et al. (2024)、Blank et al. (2026)(本文沿用其 DPO 设置与 DoM)、Papadatos & Freedman (2024)。Reward hacking:Skalse et al. (2022)、Bergen et al. (2026b)(提供 SWE hacky rollouts)。
- 基线:ERM、Projection、LEACE、INLP(视觉);数据过滤、reward shaping、inoculation prompting、PPS、降学习率(谄媚);普通 GRPO、随机方向、过滤 peeking rollouts(Countdown);DPO、RAFT(StrongREJECT 偏好)、Libon et al. 的 probe loss(无害化)。
作者定位:把上述特征奖励与归因惩罚统一为一个约束学习框架,并新增 reliance 的归因式表述与监控层级,系统研究训练对可监控性的影响。
论文如何解决这个问题?
Mechanistic Shaping:在选定层估计概念方向,用归因平方惩罚控制 reliance、用特征分数塑形奖励控制 expression,并用五级监控评估训练后概念可读性。
整体思路:把规范写成对内部表示的约束,与任务风险联合优化。Mechanistic Shaping 统一了两类规范:reliance(读出不得依赖某概念)与 expression(输出或激活不得表达某概念)。
问题形式化
- 约束学习:
L(\theta) = R(\theta) + \lambda\, \omega_\bullet(\theta),其中 R 是任务风险,ω• 是用内部表示度量的非负违规度,λ 控制惩罚强度(Eq. 1)。 - 在选定层取激活 z = fθ(x),概念分数 z• = g(z),g 可以是线性 probe、DoM 投影或 featurizer atom。
Expression shaping
- 对生成轨迹按 token 打分:ω•(θ) = E Σ_t γ^(t−1) ρ(z•,t),γ∈(0,1] 折扣后续步;求和换成均值或最大值分别约束平均表达与最强表达(Eq. 2)。
- 梯度分两项(Eq. 3):policy gradient 项改变采样哪些轨迹;梯度穿过分数的项改变固定轨迹的分数。detach 分数只留前者;attach(如 Libon et al., 2026)两者都留。
- reader 可以是当前策略(ϕ=θ)、初始参数的冻结副本(ϕ=θ0)或独立模型;作者指出这一选择影响可监控性。
Reliance shaping
- 把模型在选定层拆成 encoder fθ 与标量读出 sθ(类别 logit 或 preference margin),要求读出对概念变化不变。
- 惩罚归因的平方:
\omega_\bullet(\theta) = E\, \nu(z, v)^2(Eq. 4);ν 取自一族归因方法:Gradient、Gradient × Input、Integrated Gradients、SmoothGrad(梯度类)与 Noise、Shuffle、Deletion(扰动类)。 - Proposition 5(附录 B):仿射读出下所有归因都在且仅在 ⟨w, v⟩=0 时为零,因此各惩罚选出相同最优解,只差尺度。
- 方向估计:作者论证塑形需要 pattern(激活实际移动的方向,即 DoM),而 probe 学的是 filter(最易检测的方向);Waterbirds 上 DoM 优于判别式 probe(Figure 8)。可训练编码器中方向每 100 步重算(refresh),让惩罚跟随概念表示移动。
监控层级(Table 1)
- L1 复用原线性 probe;L2 重拟合线性 probe;L3 重拟合非线性 probe;L4 换层读;L5 换模型读。逐级放宽“原监控器哪部分可变”,区分探测器失效与概念编码方式改变。
- L1–L3 在干预层评估(擦除方法用投影后激活),L4 允许同模型其他层,L5 用独立模型、不能证明被塑形模型内仍可读。
各应用的实例化
- 视觉(Section 2):DINOv2/DINOv3/RAD-DINO 表示上训练读出,惩罚对背景(Waterbirds)、SAE 概念条件规则(ImageNette)、agent 发现的伪特征方向(CheXpert)的 reliance。
- 谄媚(Section 3):DPO 阶段对 preference margin 做 reliance shaping,方向用 SFT checkpoint 上的 sycophancy DoM。
- Reward hacking(Section 4):GRPO 中把 layer-19 概念分数标准化后从奖励减去(λ=0.4),比较 on-policy detach、on-policy 带梯度、frozen base(RLFR)三种打分。
- 无害化(Section 5):DoM 方向在 frozen base 上估计,分数经 REINFORCE 作为全部奖励(RLFR 式),teacher KL 正则;与 Libon et al. 的逐 token hinge probe loss 对比。
理论分析(Section 6)
- 两层线性模型加固定投影 P:Proposition 1 给出投影下梯度闭式解,旧编码冻结、新编码沿 P w 立即以约 ∥Pw∥β• 的速率重建。
- 残差玩具模型:Proposition 2 显示干预前每个 block 收到相同的重建信号;频繁 refit 投影(K≤10)把概念逼入非线性乘积码,Proposition 3–4 与 Lemma 1 刻画该码的结构及线性读出器无法使用它。
- 约束学习:
论文做了哪些实验?
五个实证领域加理论玩具模型:视觉捷径、条件规则、医学影像、DPO 谄媚、GRPO reward hacking、无害化,均同时报告行为效果与监控层级。
实验设置
- 被测模型:DINOv2(Waterbirds,冻结/全解冻)、DINOv3 Large(ImageNette)、RAD-DINO(CheXpert,冻结)、OLMo 3 7B / OLMo 2 13B / OLMo 3 32B(DPO)、Qwen3.5-9B(GRPO)、Mistral-7B-Instruct 与 abliterated Llama-3-8B-Instruct(LoRA 无害化)。
- 数据:自建 Waterbirds(2,066 训练鸟,类别与背景完全相关;验证/测试各 512 张);CheXpert 1,152 张完全混淆训练影像(验证/测试各 100 名患者);OLMo 3 DPO 数据 124,942 条回复;Countdown 智能体环境;BeaverTails 有害/安全回答。
- 指标:WGA 与背景检测 AUROC(视觉);sycophantic agreement rate(MMLU 多轮,用户质疑后改答案比例)与六基准均值(谄媚);peek rate 与 honest solve rate(Countdown/chess);JBB(StrongREJECT)、ClearHarm、XSTest 安全回答率、相对效用(无害化)。
- 评审方式:谄媚数据用 LLM judge 标注(Figure 12 引 Sonnet 5 judge);SAE 特征由 Claude Opus 5.5 agent 打分;JBB/ClearHarm 用 StrongREJECT 评审;DPO/RAFT 基线用冻结 StrongREJECT judge 构造偏好。
- 重复次数:Waterbirds 与 CheXpert 报 3 seeds 均值(Figure 31 为 10 seeds);Countdown 除 filtering 外为 3 seeds 均值;无害化与玩具模型为 1 seed。
- 方向与层:Waterbirds 背景 DoM 用 128 张配对图估计,可训练编码器每 100 步刷新;Countdown/无害化均在 layer 19。
主结果
无害化主表(Table 2,checkpoint 按 JBB 最低且保留 ≥95% 效用、安全回答 ≥80% 选取):
表格较宽,可左右滑动
模型 方法 JBB ↓ JBB pref. ↓ XST ↑ Utility ↑ L1 AUROC ↑ Mistral-7B Base 0.70 0.81 0.98 1.00 0.87 Mistral-7B Probe on policy (Libon et al.) 0.16 0.21 0.98 0.97 0.64 Mistral-7B DoM on frozen base (ours) 0.01 0.04 0.80 0.98 0.84 Llama-3-8B (ablit.) Base 0.71 0.71 0.98 1.00 0.87 Llama-3-8B (ablit.) Probe on policy (Libon et al.) 0.19 0.21 0.90 0.96 0.52 Llama-3-8B (ablit.) DoM on frozen base (ours) 0.01 0.02 0.86 1.04 0.85 - 作者称 frozen-base DoM 奖励在匹配效用下把 JBB 压到 0.01,优于所评测的持续更新 probe loss,且更好保留原 probe(L1)。
- 代价是更多拒答:frozen-base 奖励拒答 14–20% 的 XSTest 安全提示,probe loss 为 2–10%。
- DPO/RAFT 基线保留原 probe(L1 0.86–0.87)且安全回答更多,但有害性降幅更小,尤其在 compliant prefill 下。
Waterbirds 与 CheXpert(视觉 reliance)
- 测了什么:完全混淆下塑形对背景的 reliance。结果:全解冻 DINOv2 最后一层,各惩罚 WGA 93–97.7%,ERM 为 51.6%;block 9 处 Shuffle 93.8%,且所有 reliance 惩罚保住 L1(0.98–1.0),而 Projection/LEACE/INLP 把 L1 打到接近随机(Figure 3)。
- 作者解读:最后一层 Gradient/SmoothGrad/Noise 保住 L1,最准的惩罚(Gradient × Input 等)与擦除基线一起丢掉两个探测器,作者认为这指向 erasure;block 9 处 LEACE 让 L1/L2 降到 0.44/0.59,非线性 probe 仍 >0.98,且两个线性探测器两层内恢复(Figure 7a),作者称背景被移入协方差(二阶编码)而非擦除。
- CheXpert:ERM 33.3% WGA,平衡训练 56.0%;用 agent 发现的高秩方向(无需 drain 标注)reliance shaping 在若干情况下匹配或超过平衡参考(如 Gradient、Gradient × Input;Figure 10)。
谄媚(DPO reliance shaping)
- 测了什么:OLMo 3 7B DPO 阶段对 preference margin 塑形。结果:Gradient × Input 在不改变 SFT 谄媚水平的同时保住 DPO 能力(Figure 13 的 pareto 前沿);PPS 中等;数据过滤、reward shaping、inoculation prompting 落在 SFT–DPO 连线上,与降学习率效果相当。
- 数据层面负面结果:LLM judge 只在 <1% 数据中发现谄媚信号,且 rejected 多于 chosen;probe 过滤在移除 30k 对时只带来统计显著但很小的谄媚降低(Figure 12A/B)。
- 可监控性:原始 DoM 的 AUROC 从 0.981 降到 0.931,自由重拟合线性 probe 全程保持完美 AUROC(Figure 14A),作者将其定位在 L1 与 L2 之间;OLMo 2 13B 与 OLMo 3 32B 上同一流程同样有效,最优 λ 同数量级(附录 Figure 40)。
Reward hacking(GRPO expression shaping)
- 测了什么:Countdown 中从奖励减去 layer-19 概念分数。结果:普通 GRPO 十次更新内 peek 从约 0.15 升到 1.00、honest solve 归零;所有 expression shaping 变体约十次更新内压住 peeking,honest solve 末段达 0.96–0.98;filtering 仅 1/3 seed 成功(Figure 16 左)。
- 迁移:未在 chess 上训练的 checkpoint 在 512 道 chess 题上,被塑形策略 peek ≤0.004,honest solve 0.48–0.60(用自己的 v)与 0.71(frozen base 的 v);GRPO 则 peek 0.99、honest 0.01(Figure 16 右)。
- 监控:梯度穿过 probe 分数时,原始 probe 平衡准确率从 0.98 降到 0.57(update 150),其他变体保留 0.87–0.90;重拟合线性 probe 在所有 checkpoint 恢复 0.97–0.98(Figure 17)。
其他消融与分析
- 方向选择(Waterbirds 冻结,Figure 8):DoM 方向约 90% WGA,logistic probe 方向约 80%;continuum regression 显示 γ∈[1,2] 为性能平台(Figure 29)。
- 刷新周期(Figure 31):50 步刷新让 WGA 从约 85% 升到约 95%;5,000 步时归因在刷新间重新增长。
- 激活因子梯度(Figure 32):Gradient × Input 中 ⟨z,v⟩ 因子的梯度乘数 ρ 超过临界值后 L1 监控突然丢失,λ 越大临界 ρ 越小。
- 无害化方向旋转(Figure 20):穿过 probe 训练产生最高 62°(Mistral)/58°(Llama)旋转,但 frozen-base 奖励也旋转约 50° 而 L1 保持 0.84,作者称方向漂移本身不构成监控失效。
- 长训练(Figure 21):Mistral 400 次更新后原始 layer-19 probe AUROC 0.30(低于随机),重拟合线性 probe 各层恢复 0.83–0.87;非线性 reader 只多 0.02–0.04。
- 玩具模型(Figure 25):refit 周期 K≥20 时线性 probe R² 0.90–0.93、新擦除后准确率约 0.50;K≤10 时线性 R²≤0.014、非线性份额升到约 0.29、MLP probe R²=0.99、新擦除后准确率 0.80(最优 0.806)。
有什么可以进一步探索的点?
作者明确列出的后续方向包括训练时长、前沿规模与其他规避形式;实验覆盖五个领域、模型最大 32B。
作者指出的局限与后续方向
- 训练时长:Discussion 指出更长的优化即使在打分模型冻结时也会降低原 reader 的可靠性,同一干预短期可能无害、长期可能规避监控,作者称训练时长是评估这类方法的重要考量,留作未来工作。
- 前沿规模:Discussion 强调目前观察到的较好可监控性结果能否在前沿规模模型与训练 run 上继续成立,仍是开放问题。
- 其他规避形式:Section 6 指出乘积码只是线性干预未触及的二阶成分,不排除概念进入更高阶交互、跨层或跨 token 分布;Discussion 还提到已有工作预期的其他形式(模型主动删除再重学表示、直接作用于可解释性工具与解读者),作者称未来工作应确定哪些形式可实现、扩展监控框架并找出检测的根本极限。
- 无害化的行为退化:Section 5 指出延长训练后回答会变成长篇说教或重复拒答,虽然有害性分数低;冻结打分模型只抑制表示变化,不约束生成文本,因此行为质量与内部可监控性必须分别持续评估。
- 谄媚机制解释:Section 3 作者推测 shuffle、noise 等归因失败是因为它们需要数据中存在可定位的行为信号,而 subliminal/delta learning 可以在没有可定位行为时发生——这是作者的推测语气。
实验覆盖范围
- 被测模型共 10 个:DINOv2、DINOv3 Large、RAD-DINO、OLMo 3 7B、OLMo 2 13B、OLMo 3 32B、Qwen3.5-9B、Mistral-7B-Instruct、abliterated Llama-3-8B-Instruct,以及作为 frozen reader 的 Qwen3.6-35B;最大 32B 参数。
- 领域与范式:视觉分类(Waterbirds、ImageNette、CheXpert)、DPO(谄媚)、GRPO(Countdown,迁移到 512 道 chess 题)、REINFORCE/LoRA(无害化),加两层线性与残差玩具模型。
- 重复次数:Waterbirds/CheXpert 为 3 seeds(刷新率实验 10 seeds),Countdown 为 3 seeds(filtering 按 seed 报告),无害化与玩具模型为 1 seed。
- 监控评估覆盖 L1–L3(原 probe、线性重拟合、非线性重拟合)与跨层读取(Figure 21),L5(独立模型)未作为实验结果报告。
- 论文未报告谄媚实验中 LLM judge 与人工标注的一致性;无害化与谄媚的多数分析为单 seed。
总结一下论文的主要内容
Mechanistic Shaping 用内部表示把“不表达/不依赖”写成训练目标,在五个领域有效,并系统刻画了概念规避监控的三种几何机制。
一篇对齐/训练方法论文:把可解释性工具从“事后读模型”变成“训练中塑形模型”的手段,并追问塑形后概念还能否被监控。
- 问题:训练目标欠规范,模型可以靠背景捷径、谄媚、reward hacking 或有害内容降低损失;数据层面手段在完全混淆或信号不可定位时失效。
- 方法:Mechanistic Shaping 在选定层用 DoM 估计概念方向,reliance shaping 惩罚归因平方 ν(z,v)²(控制读出依赖),expression shaping 把概念分数从奖励中减去(控制输出表达);配套 L1–L5 监控层级区分探测器失效与编码改变。
- 视觉结果:完全混淆 Waterbirds 上 WGA 从 ERM 51.6% 升至 93–97.7%(全解冻 DINOv2);CheXpert 上用 agent 自动发现的伪特征方向,无需 drain 标注即可匹配或超过平衡训练的 56.0% WGA。
- 语言结果:OLMo 3 7B DPO 中 Gradient × Input reliance shaping 在不升高谄媚的同时保住能力,而数据过滤、reward shaping、inoculation prompting 仅相当于降学习率;13B/32B 上同样成立。
- RL 结果:Countdown 上 expression shaping 约十次更新内压住 peeking、honest solve 达 0.96–0.98,并零训练迁移到 chess(peek ≤0.004);Mistral/Llama 无害化中 frozen-base DoM 奖励把 JBB 从 0.70–0.71 降到 0.01,代价是安全提示拒答率 14–20%。
- 监控结论:作者称概念被塑形后大多仍可监控——重拟合线性 probe 在多数设置恢复读数;梯度穿过 probe 分数会让原探测器失效(Countdown 上 0.98→0.57),而 detached/frozen 打分保留原 probe。理论上刻画了三种规避几何:方向漂移(re-encoding)、擦除、进入协方差的非线性乘积码,各自留下对应级别监控可读的签名;作者强调这不排除更大规模下的其他形式,呼吁建立“基于可解释性的训练”的严谨科学。