跳到正文
arXiv cs.AI· Tiejin Chen, Maunil Vyas, Huaiyuan Yao, Hua Wei·· 1 天前AI 评分34

将引导向量用作 LLM 后训练的训练信号

Steering Vectors as a Training Signal in LLM Post-Training

AI 导读

研究提出把引导向量(SVs)从推理时的后置控制手段转为后训练阶段的训练信号,在 SFT、蒸馏和 RLVR 中验证其有效性。监督设置下 SV 注入优于普通 SFT 与蒸馏,从 thinking-mode 教师蒸馏时提升最高达 12%,且跨三个随机种子稳定。RL 中将正确性对比方向用于错误 rollout 的优势修复,可恢复零梯度全错组的学习信号并迁移至留出数学基准。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:引导向量(SVs)是残差流方向,在推理时将其加到隐藏状态上可改变 LLM 的行为。以往工作主要将 SVs 用作事后控制,我们则探究相同的方向是否也能在训练后阶段引导参数更新。我们在监督学习(包括 SFT 和蒸馏)以及带可验证奖励的强化学习(RLVR)中研究这一问题。我们的框架考虑两种干预机制:教师强制监督训练期间的隐藏状态钩子,以及 GRPO 中对错误 rollout 的优势修复机制。对每种机制,我们分析三个设计轴,即引导方向、其放置位置,以及用于将其与训练结合的操作。在监督设置中,SV 注入优于普通 SFT 和蒸馏,增益随离策略差距增大而增大,当从思考模式教师蒸馏时提升最高可达 $12\%$。这些增益在三个随机种子下均成立。在 RL 设置中,将正确性对比方向作为优势修复应用于错误 rollout(包括原本零梯度的全错组),可恢复一种学习信号,该信号可迁移到留出数学基准,并作为一种可组合的辅助信号,叠加到现有的零方差补救措施(如熵感知优势塑造)之上。我们进一步识别出三个失效边界。具体而言,代码生成可能破坏方向,跨家族层迁移可能破坏放置位置,在经 SV 训练的检查点上进行推理时应用可能破坏干预机制。总之,这些结果确立了引导向量作为 LLM 训练后阶段一种有条件但有效的训练信号。我们的代码可在此处获取。
评论: 被 NeurIPS 2026 接收
主题: 机器学习(cs.LG);人工智能(cs.AI)
引用为: arXiv:2610.40998 [cs.LG]
  (或 arXiv:2610.40998v1 [cs.LG] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.40998

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Tiejin Chen [查看邮箱]
[v1] 2026 年 10 月 29 日星期四 21:28:51 UTC(3,109 KB)

来源:arXiv cs.AI · arxiv.org