带可验证奖励的策略内蒸馏

Hugging Face Daily Papers 论文

摘要

论文介绍了OPDVR,这是一种结合策略内蒸馏与可验证奖励的方法,使用ReLU门控重配来增强大型语言模型的推理能力,无需额外超参数,在六个基准测试上优于标准方法。

带可验证奖励的强化学习(RLVR)和策略内蒸馏(OPD)已成为训练大型语言模型后的两种广泛采用的范式。然而,RLVR面临稀疏的任务级反馈问题,而OPD提供密集的令牌级指导,但忽略了轨迹正确性,将其性能限制在教师模型的水平。将它们结合是一个有前景的方向:OPD提供密集的监督信号,而RLVR提供任务级正确性。然而,现有的集成方法通常依赖于加权组合或启发式切换,引入了额外的超参数和权衡。我们提出带可验证奖励的策略内蒸馏(OPDVR),这是一种简单而有效的方法,能够无缝结合OPD和RLVR,无需添加任何超参数。我们首先基于轨迹正确性重新表述采样令牌OPD的隐式奖励,然后应用ReLU门控机制,确保正确轨迹获得非负奖励,错误轨迹获得非正奖励——从而将蒸馏信号与任务成功对齐,同时保留教师模型的分布指导。此外,我们的修改将采样令牌OPD转化为一种适当的RLVR方法,使其易于与任何策略梯度算法(如GRPO)结合。在六个推理基准上的实验表明,OPDVR始终优于标准OPD。我们的代码可在 https://github.com/LeapLabTHU/OPDVR 获取。
查看原文
查看缓存全文

缓存时间: 2026/08/26 03:15

论文页面 - 基于可验证奖励的在线蒸馏

来源:https://huggingface.co/papers/2608.24696

摘要

OPDVR 通过基于轨迹正确性的隐式奖励重构和 ReLU 门控机制,将在线蒸馏与可验证奖励相结合,在无需引入额外超参数的情况下提升了模型的推理性能。

基于可验证奖励的强化学习(RLVR)与在线蒸馏已成为大语言模型后训练的两大主流范式。然而,RLVR 存在任务级反馈稀疏的问题,而在线蒸馏虽能提供密集的词元级指导,却忽略了轨迹正确性,其性能上限受限于教师模型。将两者结合是一个有前景的方向:在线蒸馏提供密集监督信号,RLVR 提供任务级正确性验证。然而,现有集成方法多依赖加权组合或启发式切换,需引入额外超参数并面临权衡问题。

我们提出基于可验证奖励的在线蒸馏,这是一种简单有效的方法,无需增加任何超参数即可无缝融合在线蒸馏与 RLVR。首先,我们基于轨迹正确性重构采样词元在线蒸馏的隐式奖励,随后应用 ReLU 门控机制:确保正确轨迹获得非负奖励,错误轨迹获得非正奖励,从而使蒸馏信号与任务成功度对齐,同时保留教师模型的分布指导。此外,该改进将采样词元在线蒸馏转化为标准的 RLVR 方法,使其可与任何策略梯度算法(如 GRPO)直接结合。

在六个推理基准上的实验表明,OPDVR 的性能始终优于标准在线蒸馏。代码开源地址:https://github.com/LeapLabTHU/OPDVR。

查看 arXiv 页面 (https://arxiv.org/abs/2608.24696) 查看 PDF (https://arxiv.org/pdf/2608.24696) GitHub (https://github.com/LeapLabTHU/OPDVR) 添加至收藏 (https://huggingface.co/login?next=%2Fpapers%2F2608.24696)

在智能体中获取本文:

hf papers read 2608.24696

未安装最新 CLI?请执行:curl -LsSf https://hf.co/cli/install.sh | bash

引用本文的模型 0

暂无模型关联此论文

在模型 README.md 中引用 arxiv.org/abs/2608.24696 即可从本页链接。

引用本文的数据集 0

暂无数据集关联此论文

在数据集 README.md 中引用 arxiv.org/abs/2608.24696 即可从本页链接。

引用本文的空间 0

暂无空间关联此论文

在空间 README.md 中引用 arxiv.org/abs/2608.24696 即可从本页链接。

包含本文的合集 0

暂无合集收录此论文

将本文添加至合集 (https://huggingface.co/new-collection) 即可从本页链接。

相似文章

论同策略蒸馏的几何结构

Hugging Face Daily Papers

本文刻画了大语言模型中同策略蒸馏(OPD)独特的参数空间动力学,表明其具有松弛的非主方向更新和子空间锁定特性,从而与监督微调和基于可验证奖励的强化学习区分开来。

同策略Delta蒸馏(OPD^2)

Hugging Face Daily Papers

本文提出同策略Delta蒸馏(OPD^2),一种新的蒸馏奖励——差异信号,该信号捕捉教师模型与其基础模型在推理调优前的差异,为迁移推理能力提供更直接的信号。在数学、科学和代码基准上的实验表明,OPD^2始终优于传统的同策略蒸馏。

超越模仿:基于推理进展过滤策略蒸馏

arXiv cs.AI

论文提出了Reasoning-Progress-Aware Reward Filtering for On-Policy Distillation (R2-OPD) 来解决教师派生奖励与真实推理进展之间的不匹配问题,提高语言模型训练中的推理性能。

利用在线策略逆蒸馏实现弱到强泛化

Hugging Face Daily Papers

论文提出了在线策略逆蒸馏(OPRD)方法,该方法通过沿教师策略偏移方向放大验证器支持的策略梯度,使更强的AI模型能够超越较弱的监督者,在蒸馏场景中以更少的更新次数获得更高性能。

同策略蒸馏中的增益与崩溃:一个强化学习视角

Hugging Face Daily Papers

本文从强化学习视角分析了用于语言模型后训练的同策略蒸馏(OPD),解释了OPD既能提升性能又可能崩溃为重复生成的原因。研究发现,在训练中屏蔽不健康响应以及采用SFT初始化可以缓解崩溃,从而将关注点从教师的生成质量转移到其评估学生轨迹(rollout)的能力上。