核心判断
这条讨论的价值在于把 OPD 从“训练目标”问题推进到“参数轨迹”问题。过去我们常说 OPD 是 on-policy RL 与蒸馏的结合:学生自己生成轨迹,teacher 给密集 token 信号。但论文给出的图像更尖锐:OPD 的终点看起来在 SFT 和 RLVR 之间,路径却很早锁进一个低维通道,并且这个通道对 OPD 训练足够有效。
不是线性折中
OPD 的更新稀疏性、谱漂移、主子空间旋转都介于 SFT 与 RLVR 之间,但它的训练轨迹并不是从 SFT 慢慢滑向 RLVR。
早期低秩锁定
OPD 的累计更新从早期 checkpoint 开始就进入稳定低维带;限制到早期 rank-16 子空间后,OPD 表现基本保留。
泛化双刃剑
低维通道可能是隐式正则化,也可能是 teacher / rollout / objective 早期共同决定的路径依赖瓶颈。
为什么这个问题重要
LLM 后训练里,SFT、RLVR 与 OPD 经常被放在一个二维表里比较:采样是 off-policy 还是 on-policy,反馈是 dense 还是 sparse。这个表很有用,但也容易让人误以为 OPD 只是“RLVR 加 dense teacher reward”或“SFT 换成学生轨迹”。《On the Geometry of On-Policy Distillation》的贡献是把比较对象从 loss function 换成参数空间:训练之后,权重到底在哪里变了?主奇异方向有没有被改写?累计更新能量集中在多少个方向?
| 方法 | 训练状态分布 | 监督信号 | 典型优势 | 典型风险 |
|---|---|---|---|---|
| SFT | 固定外部数据或 teacher 轨迹 | token-level 交叉熵,信号密集 | 容易注入新知识、新格式、新 token 支持 | 训练在 teacher 常见状态上,学生自己跑偏后容易 compounding error;参数改写也更密 |
| RLVR | 学生当前策略的 rollout | 可验证结果奖励,通常是序列级稀疏信号 | 与学生真实错误分布对齐,更像在自己的行为上改错 | credit assignment 困难,奖励稀疏,训练吞吐与稳定性压力大 |
| OPD | 学生当前策略的 rollout | teacher 在学生状态上的 token-level 指导 | 同时获得 on-policy relevance 与 dense feedback | 依赖 teacher 与学生的状态兼容性;可能把学习锁进早期 teacher-guided 通道 |
因此,这篇论文真正追问的是:OPD 在参数空间里的足迹是否和它的表面定义一致?如果 OPD 只是 SFT 与 RLVR 的中间点,那么我们可以把它当作二者的工程折中;如果它有独立几何,那就意味着 OPD recipes 应该监控和控制 update geometry,而不是只调 teacher、token mask、rollout 数量或 KL 系数。
机制拆解:OPD 怎样产生这种几何
OPD 的训练循环可以理解为五步:学生生成自己的轨迹,teacher 在同一 prefix 上计算 token-level 偏差,训练把这些偏差变成 advantage 或 KL 目标,再通过受限更新改变学生参数。和 SFT 的关键差异是状态来自学生自己;和 RLVR 的关键差异是每个 token 都有 teacher 信号。
统一梯度视角
论文用一个抽象形式把 SFT、RLVR、OPD 的更新写在一起:每个 token 的 score function 是 \(\phi_t = \nabla_\theta \log p_\theta(y_t \mid x, y_{\lt t})\),整体梯度可以看作 token score 的加权和。
差别在于权重 \(a_t\)。RLVR 里同一条序列的 tokens 通常共享一个序列级 advantage;OPD 里 \(a_t\) 随 teacher-student token disagreement 变化。也就是说,OPD 保留了 RLVR 的 on-policy 状态分布,但把稀疏结果信号替换成更细粒度的 teacher token 信号。
Three-Gate 的放松版本
论文借用了 RLVR 几何里的 Three-Gate 直觉:第一道门是 policy / distribution anchor,让更新不能离初始化太远;第二道门是预训练模型本身的光谱结构,倾向把受限更新路由到非主方向;第三道门是 bf16 精度,只有超过量化阈值的坐标变化才会在存储权重中可见。OPD 的 teacher dense signal 放宽了这些门:它比 RLVR 改更多坐标,但仍不像 SFT 那样大范围触碰主方向。
关键证据:OPD 的终点与轨迹
论文的证据可以分两层读。第一层是 endpoint localization:训练结束后,OPD 的参数改动在哪里?第二层是 trajectory geometry:训练过程中,累计更新 \(\Delta W_t = W_t - W_0\) 如何演化?rosinality 原帖的观察正是把这两层连在一起:从更新程度看,OPD 介于 RLVR 与 SFT;从轨迹看,OPD 早早表现为低秩。
| 诊断 | 测什么 | 论文里的排序 | 怎么解释 |
|---|---|---|---|
| bf16-aware update sparsity | 有多少权重在 bf16 阈值下近似没变 | SFT 改最多,OPD 居中,RLVR 改最少 | OPD 比 SFT 更节制,但 teacher dense signal 让它比 RLVR 改更多坐标 |
| Principal-angle rotation | 预训练主奇异子空间被旋转多少 | SFT 最大,OPD 小但非零,RLVR 最小 | OPD 仍保留预训练主结构,不像 SFT 那样明显重写 principal directions |
| Spectral drift | 奇异值谱整体偏移多少 | SFT 约 \(10^{-3}\),OPD 约 \(10^{-4}\),RLVR 约 \(10^{-5}\) | OPD 保谱程度接近 RLVR,但不是完全保守 |
| Update-mask overlap | 可见更新落在 principal mask 或低幅值区域的比例 | OPD 和 RLVR 都远离 principal weights,OPD less selective | OPD 继承了 off-principal 偏置,但因为信号更密,支撑集更宽 |
| Stable rank trajectory | 累计更新能量集中在多少有效方向 | OPD 保持窄低秩带,SFT 扩张,RLVR 收缩 | OPD 的低维通道很早形成,不是训练末期偶然结果 |
stable rank 为什么关键
stable rank 用来衡量矩阵有效维度,比普通 rank 更适合分析神经网络更新,因为普通 rank 会被大量微小奇异值干扰。论文使用的定义是:
如果 \(\Delta W_t\) 的能量集中在少数主奇异方向,stable rank 就低。OPD 的 stable rank 早早进入稳定低维区间,说明它不是每一步都在探索全新方向,而是在早期确定的通道内反复累积。
rank-16 投影实验的意义
论文在约 20% 训练进度处提取累计更新的 top-16 右奇异子空间,然后把后续梯度投影到这个子空间。结果是:OPD 在这个瓶颈下基本保持表现,SFT 则明显受损。这是全文最关键的因果性线索之一:早期通道对 OPD 不是事后描述,而是训练功能上足够。
低秩锁定会怎样影响泛化?
rosinality 的问题可以拆成两个相反假设。乐观假设:低秩锁定是隐式正则化,让 OPD 避免破坏预训练主结构,因此泛化更稳。悲观假设:低秩锁定是早期路径依赖,模型过早承诺少数方向,导致 teacher 外、分布外或长链组合任务上缺乏探索能力。两者都可能成立,关键取决于锁住的子空间是否覆盖目标任务真正需要的变化。
作为正则化
如果目标任务主要是修正学生在自身 rollout 中的局部错误,且 teacher 的思维模式与学生兼容,那么低维通道会像一个安全阀:它允许模型快速纠偏,又尽量少动预训练主结构。这类情况下,低秩锁定可能提升稳定性、减少遗忘,并让 OPD 比继续 SFT 更高效。
作为瓶颈
如果任务需要引入新 token 支持、新推理策略或 teacher 没覆盖的状态,早期通道就可能过窄。模型会擅长在熟悉错误上模仿 teacher 的修正,却难以产生真正新的搜索策略;泛化看起来好,可能只是 benchmark 分布与锁定通道高度重合。
| 泛化场景 | 低秩锁定可能带来的好处 | 可能暴露的问题 | 该看什么指标 |
|---|---|---|---|
| 同分布数学推理 | 快速修正常见 chain-of-thought 分叉错误,减少无关参数漂移 | 可能只学会 teacher 偏好的套路,而不是更强搜索 | 按题型、解法类别、错误位置分桶的 pass rate |
| 分布外难题 | 保留 base model 的通用能力,不被 SFT 数据过拟合拖走 | 早期子空间不覆盖新题所需方向,提升停滞 | OOD benchmark、难度外推曲线、失败轨迹多样性 |
| 代码与工具任务 | 可把常见格式、调用习惯、局部修复压进少数方向 | 新 API、新环境、新错误模式可能需要更宽参数支撑 | 环境迁移、工具组合、长时程 agent 成功率 |
| 持续学习 | 低漂移有利于恢复旧行为,减轻 catastrophic forgetting | 多轮任务切换后,锁定通道可能互相冲突或覆盖 | 跨阶段回归测试、subspace similarity、能力保留矩阵 |
从研究设计上看,验证泛化需要超出论文当前的 endpoint benchmark:不仅要看 AIME、代码题或固定 reasoning benchmarks,还要看分布外题型、teacher 不擅长的解法、长链恢复能力、错误前缀干预、不同 teacher 风格下的子空间相似性。如果换 teacher、换 rollout seed、换 curriculum 后早期子空间仍高度一致,低秩锁定更像模型固有的有效通道;如果子空间大幅漂移而 benchmark 仍可被拟合,泛化风险就更大。
工程与研究启发
这篇论文给 OPD recipe 的启发是:不要只把 OPD 当成“更便宜的 RL”或“更贴近学生状态的蒸馏”。一旦 update geometry 成为核心变量,训练系统就应该像监控 loss、reward、KL 一样监控参数更新的谱结构。
训练早期就看子空间
在 10%–20% 进度处跟踪 stable rank、top-K subspace similarity 与 Frobenius norm。若 OPD 很早锁定,后续可考虑低秩约束、早停或更小学习率。
把 objective 当主旋钮
论文控制实验显示 token sparsification 与 off-policy rollout 对 rank trajectory 影响有限,而 OPD/RLVR objective mixing 会改变几何轨迹。
区分规模和方向
更新 norm 变大不一定代表泛化更强;stable rank、谱尾、principal overlap 能告诉我们更新是在扩展能力还是沿窄方向加码。
用诊断驱动 teacher 选择
如果不同 teacher 给出相似子空间,说明学生内在可改方向很强;如果 teacher 诱导完全不同的通道,就要警惕 style imitation 与路径依赖。
一个可落地的 OPD 监控面板
| 面板项 | 目的 | 异常信号 | 可能动作 |
|---|---|---|---|
| stable rank over time | 确认是否过早锁进低维通道 | 过早极低且 eval 停滞 | 提高 objective 多样性、换 teacher、加入 RLVR 成分或重设 curriculum |
| subspace similarity to early checkpoint | 判断通道是否持续稳定 | 频繁旋转但性能不涨 | 降低学习率、检查 rollout 分布漂移、过滤噪声样本 |
| principal-mask overlap | 确认是否破坏预训练主方向 | 接近 SFT 式 principal rewrite | 加 anchor、减小 teacher correction scale、减少 SFT 混合 |
| OOD eval by capability bucket | 验证低维通道是否真泛化 | 同分布涨、分布外不涨 | 扩展 prompt/state 分布,而不是只增加 token 密度 |
实践上,OPD 可能最适合“学生已有能力但经常在自己轨迹上局部走错”的场景;最不适合“学生完全缺少相关支持,需要大量新行为、新知识或新工具模式”的场景。前者需要纠偏,低维通道足够;后者需要扩展支持,继续 SFT、mid-training、representation distillation 或 RL 探索可能更必要。
术语速查
边界与未解决问题
这篇论文的实验集中在 Qwen3-family reasoning 设置上,核心诊断来自保存 checkpoint 的离线参数分析。它有很强的解释力,但还不能直接推出所有模型家族、所有任务和所有 OPD 实现都会出现同样的几何。尤其是 teacher 的 logit 可访问性、KL 方向、teacher-student tokenizer/style mismatch、rollout 长度、学习率与 batch size,都可能改变实际轨迹。
不能证明因果全集
rank-16 投影说明早期通道对 OPD 足够,但没有证明这个通道是唯一原因,也没有证明所有泛化都来自该通道。
泛化证据仍需扩展
论文主要展示参数几何和标准 reasoning benchmark。真正的 OOD、长时程 agent、teacher 外策略创新还需要额外实验。
OPD 实现差异很大
reverse KL、forward KL、sampled-token log-ratio、representation distillation、logit-free OPD 都可能诱导不同更新几何。
低秩不是 LoRA 等价物
这里的低维是训练轨迹显现出的有效更新子空间,不等同于预先规定 adapter rank,也不代表完整模型只能低秩表达。
最终启发
OPD 的下一步研究不应只问“怎样让 teacher 信号更密、更便宜、更稳定”,而应问“怎样让 teacher 信号诱导正确的更新几何”。如果 OPD 的优势来自早期低维通道,那么我们可以把它变成可控对象:监控它、约束它、扰动它、比较不同 teacher 诱导的通道,甚至把 subspace quality 作为选择数据和 objective 的标准。
对泛化来说,最重要的判断是:一个低维通道如果覆盖了任务真正需要的因果变化,它就是高效正则化;如果只覆盖了 benchmark 与 teacher 风格,它就是漂亮的过拟合路径。rosinality 的问题之所以值得认真对待,是因为 OPD 的效率和风险可能来自同一个几何事实。
证据边界与资料索引
本文基于公开材料整理,重点解释 OPD 参数几何与泛化问题。论文中的具体数值、图表和实验设置以公开 arXiv 版本为准;X 动态用于定位讨论焦点,不作为独立实验依据。本文未复现实验,也未检查作者未公开的训练代码或 checkpoint。
- rosinality 关于 OPD 更新程度与低秩轨迹的讨论:https://x.com/rosinality/status/2063887402385523149
- Zhennan Shen 等,《On the Geometry of On-Policy Distillation》:https://arxiv.org/abs/2606.07082
- Thinking Machines Lab,《On-Policy Distillation》:https://thinkingmachines.ai/blog/on-policy-distillation/
- 状态分布视角下的 SFT、RL 与 OPD 讨论:https://nrehiew.github.io/blog/sft_rl_opd/
- OPD 机制与 recipe 相关背景,《Rethinking On-Policy Distillation of Large Language Models》:https://arxiv.org/abs/2604.13016
- OPD 早期低秩与 foresight 相关背景,《Learning to Foresee》:https://arxiv.org/abs/2605.11739