Learning to Steer, Steering to See: Unveiling the Geometry of RLVR in Large Language Models via Trainable Vectors
发表时间:2026-09-28
arXiv · HF 讨论 · alphaXiv① 一句话定位
本文处于RLVR可解释性与后训练稳定化的交叉地带。与直接提升推理分数的多数RL工作不同,作者把RLVR后训练重新表述为激活空间中的干预问题,以“可训练向量”为探针,把RL诱导的行为差异映射为低维有效流形,并据此提出两条经验性质与一个稳定化工具Alpha-Stabler。其定位是:对采样效率、熵动态、缩放律、参数更新几何等既有分析形成互补,同时借用激活转向、表征读出与主成分分析等机制解释工具,但聚焦RLVR收益的可压缩性与几何定位。研究性质偏实证刻画,理论部分为条件性分析,不是端到端新训练算法,贡献可概括为“一个视角、两条性质、一个工具”。
② 背景与动机
RLVR已成为LLM后训练增强推理的核心范式,但参数更新高维、优化噪声与配置敏感性使内部机制难以分析。既有研究多关注采样效率、熵动态、缩放行为与参数更新几何,较少直接刻画RL诱导行为变化在激活空间中的结构及其与能力迁移、训练稳定性的关系。同时,激活转向或表征工程提供了在冻结模型上施加低维干预的手段,主成分分析常用于刻画激活协方差的方差主导子空间。本文把两者结合:将RL后训练视为激活空间干预,用可训练向量恢复RL教师的行为增益,从而探测增益能否被低维压缩、以及有效干预方向位于激活空间何处。设定依赖可验证奖励任务(答案匹配、单元测试、约束检查)与GRPO/DAPO等RL算法;蒸馏分on-policy与off-policy,并以全参微调、LoRA作为不同表达容量的对照。这一框架引出两条几何性质,后者进一步导向训练稳定化方法。
③ 方法机制
本文机制分四个层次:探针设计、容量边界分析、流形几何刻画、Alpha-Stabler的监控与控制。 (一)可训练向量作为RLVR探针 作者把RLVR后训练看作激活空间干预问题。给定冻结基座模型(L层、隐维d),在控制层集S上,对每层输出加一个可训练向量:h′_ℓ = h_ℓ + δ_ℓ,ℓ∈S。δ_ℓ是层特定的d维向量,对所有输入和所有token位置共享;只有{δ_ℓ}被优化,骨干完全冻结。训练目标是蒸馏:教师为RL微调后的模型,学生为冻结基座加向量,采用on-policy蒸馏(学生生成轨迹)或off-policy蒸馏(教师轨迹),分别对应附录中的损失形式。对照组为全参微调与LoRA。直觉是:若极低维、输入不变的激活偏置就能恢复RL收益,说明RL引起的行为变化可压缩到低维流形;若不能,则说明所需修正具有输入依赖性或需要更高容量。默认配置为每层一个向量、控制中间一段连续层,参数规模远小于骨干。 (二)容量边界与层间机制 先考察压缩性。on-policy与off-policy下,全参与LoRA接近,单向量转向也能恢复大部分RL增益(具体幅度见实验)。但压缩有边界:固定学生、更换教师时,RL-trained teacher可被转向迁移,非RL teacher则随teacher-student分布差距增大而退化;大差距下出现明确容量排序:full FT > LoRA > vector steering。作者解释为:差距越大,修正越需要输入依赖,固定加性偏置最先触达容量上限。 再考察注入深度。把三种更新限制在单层时,全参与LoRA在各深度有效,固定向量在低/中层表现好、近输出退化。梯度范数显示浅层与高层梯度幅度相当,因此不是优化信号弱;更可能是高层下游非线性变换少,固定偏置难以产生上下文相关效果,而LoRA或单层全参可直接生成输入依赖修正。 为验证该解释,作者把固定偏置替换为低秩非线性门控:δ_ℓ(h_ℓ)=B_ℓ σ(A_ℓ h_ℓ),其中A_ℓ∈R^{r×d}、B_ℓ∈R^{d×r}、r≪d,σ为元素级门控。B_ℓ定义共享修正方向,σ(A_ℓ h_ℓ)为每个token分配依赖系数;B_ℓ零初始化保证初始恒等。Rank-1门控即每层仍只有一个共享方向,但其强度可随token变化。与参数匹配的静态控制(把σ(A_ℓ h_ℓ)换成常数系数)相比,只有门控带来改善;提高门控秩可进一步逼近全参参考。这说明高层恢复取决于“可用修正方向”与“输入依赖调制”两者。 梯度对齐分析:对每层计算各token对转向向量梯度贡献的平均两两余弦相似度。中低层梯度相似度低、近输出明显更高(数值见实验)。若仅方向对齐决定效果,高层应更利于共享方向干预,但固定向量恰在高层退化,故平均方向对齐不能解释深度差异。再看Rank-1门控系数分布:低/中层系数在[0,1]广分布,类似分级调制;高层集中在0与1附近,类似选择性开关。结合静态对照,说明显式token依赖调制能解除高层瓶颈。由此总结Property 1:复现RL收益所需容量可以很小,但并非无限可压缩,有效容量取决于干预形式与所需修正的匹配程度。 (三)多向量与流形几何 为检验主导方向是否唯一,顺序学习K个向量,并与先前向量两两正交;每个阶段只注入当前阶段向量,先前向量冻结、仅用于定义正交约束,每次优化步后按Gram-Schmidt形式重投影到先前方向的正交补:v^(k) ← v^(k) − Σ_{j<k} ⟨v^(k),v^(j)⟩/‖v^(j)‖² v^(j)。结果表明主导方向有效但不唯一,高阶方向独立注入仍有可观效果;模型越大,高阶方向相对退化越缓。 可复现性:同一任务内,比较不同teacher参数化(全参微调与不同秩LoRA)以及on/off-policy蒸馏提取的方向。层平均余弦相似度显示同索引对齐强于跨索引,匹配索引跨机制对齐也较高,说明同一任务内几何结构大致一致。 跨任务:将科学任务按物理、化学、材料、生物拆分,每主题训练RL teacher并蒸馏提取多向量方向。定义源-目标方向对齐A(s,t)为前四个同索引余弦相似度在控制层上的平均;定义归一化相对迁移增益Δ_rel Acc(s→t) = (Acc(M_s,t)−Acc(M_base,t))/(Acc(M_t,t)−Acc(M_base,t))×100%。方向对齐与迁移增益正相关(拟合优度见实验),表明所学几何与能力迁移相关。 主子空间分离:取冻结基座模型token级激活协方差的领先正交特征向量U_{ℓ,r}(r约为隐维的10%),P_ℓ=U_{ℓ,r}U_{ℓ,r}^T。定义方向z_ℓ的主成分能量占比e_PC(z_ℓ)=‖P_ℓ z_ℓ‖²/(‖z_ℓ‖²+ε)。无约束转向在Top主成分子空间的能量占比极低,远低于各向同性参考,且跨方向稳定(具体百分比见实验)。功能测试:每步移除转向向量在bottom主成分子空间的投影、将其限制在Top主成分张成空间,能量占比上升但准确率无显著改善,说明增益并非来自与主成分子空间对齐。 进一步定义训练中的主成分入侵PSI。步骤t时,让当前策略与冻结基座在同一token序列上运行,取激活差Δ=h(t)−h_base(t),按行堆叠为X_ℓ(t),计算PSI_ℓ(t)=‖X_ℓ(t)U_{ℓ,r}‖²_F/(‖X_ℓ(t)‖²_F+ε)。先按token能量求和再归一,避免方向相反的位移相互抵消。易崩溃的运行中,稳定期PSI低;崩溃前PSI先升高且波动加剧,早于分数可见退化;崩溃中PSI继续上升。由此总结Property 2:有效控制方向集中在低方差补空间,并与方差主导的表示子空间分离。 (四)Alpha-Stabler Predictor:预热阶段(前若干更新)由冻结基座评估actor生成的rollout,估计各监控层token级激活协方差,提取前r=⌈0.10d⌉个领先正交特征向量并固定。层级PSI按上述公式计算;跨层聚合PSI将分子、分母分别对各层求和后相除:PSI(t)=Σ_ℓ ‖X_ℓ U_{ℓ,r}‖²_F/(Σ_ℓ ‖X_ℓ‖²_F+ε)。PSI用detached激活计算,不进入梯度,actor计算图保留用于反向。预热后用层中位数与缩放MAD设定警告阈值与较低释放阈值;后续有效观测更新指数滑动平均,连续p次超过警告阈值则激活控制,低于释放阈值则关闭。 Controller:前向激活不变,只改反向梯度。设G_ℓ(t)为第ℓ层传入梯度,a_ℓ(t)为Predictor在反向与优化器步中固定的激活标志,则G̃_ℓ(t)=G_ℓ(t)−a_ℓ(t)(G_ℓ(t)U_{ℓ,r})U_{ℓ,r}^T。激活时,对每个token梯度移除其在主子空间的分量、保留正交补分量;rollout生成、似然比、奖励、GRPO或DAPO损失均不变,优化器仅使用修改后的反向信号。每次active hook执行两次低秩矩阵乘,推理只用训练后actor,额外开销很小。 训练与数据细节:基座为DeepSeek-R1-Distill-Qwen与Qwen3系列多个规模;任务涵盖数学、科学、代码、指令遵循,奖励分别为答案匹配、单元测试与约束检查,均为可验证奖励;RL算法为GRPO与DAPO;蒸馏分on-policy与off-policy;评估采用mean@4(每个提示采样多个回答取平均,不选最佳、不投票)。全参/LoRA/向量转向使用各自的学习率,向量转向需要明显更大的学习率。硬件为H20 GPU、FSDP加vLLM;Qwen3运行于非思考模式以保证基座、教师与学生分布可比。
④ 实验与证据
实验部分围绕两条性质与Alpha-Stabler展开。作者在多个模型与多个可验证奖励任务上,以RL训练教师、蒸馏学生,比较全参微调、LoRA与向量转向三类更新参数化,覆盖on-policy与off-policy两种蒸馏机制;进而做层间注入深度扫描、低秩门控与静态控制对照、梯度对齐与门控系数分析。随后用顺序正交多向量探针考察方向多样性、跨teacher参数化与蒸馏机制的可复现性、跨科学子主题的方向对齐与迁移增益关系、主成分能量占比及其功能测试,以及训练过程中主成分入侵PSI与崩溃的动态关联。最后评估Alpha-Stabler在训练稳定性、学习率鲁棒性、触发时机、控制子空间选择与时间开销方面的表现。总体结论是:RL收益可被低维激活干预大部分恢复但存在容量边界;有效控制方向集中于低方差补空间;PSI可作预警信号,主成分梯度投影能稳定训练且几乎无额外开销。具体数值见各实验条目。
-
单向量转向在on-policy蒸馏下能恢复RL增益的绝大部分(超过85%)。
under on-policy distillation, LoRA and full-parameter training perform comparably, while a single steering vector per controlled layer still recovers over 85 % 85\% of the RL gain.
-
顺序正交多向量探针中,首个方向与第八个方向在SciKnowEval上的独立准确率只差约5.5个百分点,说明主导方向并非唯一。
the first direction achieves 67.5 % 67.5\% standalone accuracy on SciKnowEval with DeepSeek-R1-Distill-Qwen-1.5B, while the eighth direction, 𝐯 ( 7 ) \mathbf{v}^{(7)} , achieves 62.0 % 62.0\% , only 5.5 5.5 percentage points lower despite being orthogonal to the preceding seven directions at every controlled layer.
-
高层注入位置的梯度方向对齐反而更高(约0.14),中低层约0.019,但固定向量在高层仍退化,说明平均方向对齐不是决定性因素。
mean gradient similarity rises from about 0.019 0.019 at lower and intermediate layers to about 0.14 0.14 near the output.
-
跨科学子主题的方向对齐与归一化迁移增益呈强线性相关(R²=0.91)。
stronger directional alignment is associated with larger normalized transfer gains across these pairs, with a linear fit yielding R 2 = 0.91 R^{2}=0.91 .
-
无约束转向在Top-10%主成分子空间中的能量占比仅约1%–2%,显著低于10%各向同性参考。
across four scientific topics, unconstrained steering places only about 1 % 1\% – 2 % 2\% of its energy in the Top- 10 % 10\% principal subspace, well below the 10 % 10\% isotropic reference
-
把转向向量限制在Top-30%主成分后,Top-10%能量占比升至约15%,但准确率没有显著改善,说明增益不依赖主子空间对齐。
the Top- 10 % 10\% energy fraction rises to about 15 % 15\% , but accuracy does not improve meaningfully.
-
Alpha-Stabler在不同学习率下均能稳定训练,受控训练峰值PSI维持在低个位数百分比。
at each tested learning rate, uncontrolled training eventually collapses, whereas controlled training keeps peak PSI at a low single-digit percentage and reaches comparable final scores across settings, indicating robustness to the choice of learning rate.
-
触发式控制与始终开启控制在分数和峰值PSI上相当,延迟控制则逐步恶化。
Triggered control matches always-on control in both score and peak PSI, indicating that continuous intervention is unnecessary, whereas delaying intervention progressively degrades the score and raises peak PSI.
-
移除Top-PC梯度分量的控制优于随机子空间、Middle-PC、Bottom-PC与全局梯度裁剪。
Among the compared gradient-control methods, removal of Top-PC components achieves the highest score, while random-subspace, Middle-PC, and Bottom-PC control fail to match it and remain close to the uncontrolled baseline; Top-PC also compares favorably with global gradient clipping.
-
Alpha-Stabler在相同墙钟时间下几乎不增加时间开销。
nearly identical score trajectories with and without Alpha-Stabler under the same measured wall-clock time, indicating almost no additional time overhead.
-
Alpha-Stabler可将训练稳定维持两千步。
Alpha-Stabler can stabilize training for 2,000 steps and consistently enhance the gains brought by reinforcement learning.
⑤ 审稿人视角
Novelty:创新点分两层。方法层面,把RLVR后训练重新表述为激活空间干预问题,用可训练向量作为探针,系统比较全参、LoRA与向量转向三类容量,并引入低秩门控向量区分“共享方向”与“输入依赖调制”;顺序正交多向量提取与跨配置、跨主题对齐分析也较系统。发现层面,提出有效流形容量与控制流形分离两条性质,用主成分能量与PSI把RL训练不稳定与几何入侵联系起来,并据此设计Alpha-Stabler,这些经验规律有一定原创性。但构成要素(激活转向、PCA子空间、梯度投影、RL稳定化)大多已有,作者更多是组合与系统化,而非提出全新理论或算法原语;理论部分也明确是经验驱动的条件性结论,并非第一性原理。因此新颖性为中等偏上:视角与经验规律有新意,但核心机制并非颠覆性。 Soundness:方法论总体较规范:多模型、多任务、GRPO/DAPO、on/off-policy蒸馏,匹配采样与评估协议,给出超参与硬件,并使用静态对照、随机子空间对照、触发与延迟对照来支撑因果解释。蒸馏探针逻辑清晰,容量排序与门控实验结果自洽。但仍有关切:多数结论依赖图上趋势,未见显著性与重复方差;“崩溃”与“分布差距”的判定较定性;用蒸馏恢复推断RL几何存在代理偏差(蒸馏目标、teacher-student gap、KL余项等);顺序正交提取不能刻画解空间全貌;PSI阈值与预警没有检测误差保证;固定基座主子空间不随训练更新;模型家族与任务覆盖有限。整体上实验可信度中等,结论在作者声明的范围内大体可接受,但需要更多统计检验与消融来加固。 Significance:若结论稳健,该工作为RLVR提供了可解释的低维几何画像,并把“低方差补空间”确立为一种设计原则;Alpha-Stabler几乎零参数、零架构改动、低开销,对大规模后训练稳定化有实用价值;PSI可作为预警与自适应控制信号;跨任务方向对齐与迁移相关也提示能力迁移的几何基础。但当前验证仅限RLVR与少数模型、任务,尚未覆盖RLHF、多轮agent或开放生成;稳定化收益也只在有限步骤与基准上展示。因此意义为中等:对社区有启发与工具价值,但泛化边界尚待确认。
⑥ 局限与保留意见
本文的主要局限:其一,理论框架建立在经验动机的假设上,未从第一性原理推导;作者也承认其数学结论多为条件性命题,不能替代实证。其二,分析与实验仅覆盖RLVR,任务限于数学、科学、代码、指令遵循等可验证奖励场景,未验证RLHF、多轮agent决策或开放生成,几何性质能否迁移存疑。其三,以蒸馏恢复作为RL几何的代理存在偏差:教师-学生分布差距、蒸馏目标、KL余项和奖励噪声都可能混淆“RL增益的低维性”这一推断。其四,对“训练崩溃”“分布差距”等关键量缺乏严格形式化,PSI的警告/释放阈值依赖启发式与中位数统计,缺少检测错误率保证。其五,主成分基座取自冻结模型且固定不变,可能不是训练全程最优参考;主成分维度选择(如约10%)也较任意。其六,实验未见系统的统计显著性、多次种子方差或置信区间,部分结论仅由少数训练轨迹支撑。其七,模型家族集中在Qwen/DeepSeek系列,评估用mean@4而非pass@k或多数投票,且可能存在基准污染风险。最后,Alpha-Stabler的收益主要在稳定训练与避免崩溃,而非直接提升上限,其与不同RL算法、优化器及更大规模训练的兼容性仍需进一步检验。
⑦ 值得跟进
后续可沿几条线推进。理论方面:为低维有效流形与低方差补空间定位寻找第一性原理解释,用神经元归因、因果追踪或电路分析放松现有假设,并给出PSI预警的检测误差界。方法方面:把框架扩展到RLHF、多轮agent与开放生成,验证两条性质是否普适;将“低主成分能量”作为显式正则项或学习率、干预强度的自适应信号,发展无需逐层监控的低秩控制。评估方面:引入多随机种子、统计检验、pass@k与多数投票,扩展更多模型家族与非可验证奖励任务。应用方面:探索跨任务方向对齐作为能力迁移或模型合并的几何先验,测试Alpha-Stabler在更长训练、更大规模与不同RL算法下的稳定性,并研究自适应更新主子空间或在线估计能否进一步提升控制精度。