ThinkPrior优化RLVR冷启动提示选择

论文AI 评分 65/100arXiv cs.AI
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

ThinkPrior 方法针对使用 GRPO 训练的 RLVR 场景,通过外部锚点在离线阶段构建零回滚难度先验,解决了基于历史的提示选择需浪费目标策略回滚来估计难度的冷启动问题。在 Qwen2.5-Math-7B 模型测试中,该方法将早期静默组减少过半,前 30 步的浪费回滚减少近五分之一,且不影响最终准确率。与 DAPO 组合后,生成回滚量减少 10.6%。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

零优势组带来的浪费

在采用可验证奖励的强化学习训练中,若以组相对策略优化方式更新,奖励优势项取决于同一组内各次采样结果的差异。当一组采样全部正确或全部错误时,组内相对优势会同时归零,这类组无法提供用于更新的梯度信号。论文给出的测量显示,在均匀采样下,一次训练运行中约 39% 的采样落在这种零优势组上。这一比例构成了该方法要处理的核心浪费来源,也解释了为何提示选择会直接影响训练效率。

用外部锚点先验替代试跑

以往依据历史记录挑选提示的做法,必须先消耗目标策略的采样才能估计题目难度,因此在首次采样之前存在冷启动阶段,采样预算会先被用于试探。ThinkPrior 改为在离线阶段一次性借助外部锚点完成难度估计:由验证器对锚点作答打分,得到通过率,再用它初始化一个 Beta 后验分布。这样在目标策略尚未产生任何采样之前,就已经存在一份难度先验,选择时按期望可学习性排序。该方法不修改损失函数,也不改动优化器,只改变提示的挑选顺序。首次选择之后,后验会依据目标策略的实际训练结果继续更新。

早期零优势组减半

在 Qwen2.5-Math-7B 上跨十六个随机种子的实验中,ThinkPrior 使训练早期的零优势组数量减少一半以上,并把截至第 30 步的无效采样削减接近五分之一。最终准确率方面,未检测到差异。由于该实验使用固定的 250 条提示池,预算固定时,效果体现为采样在提示之间的重新分配,而非总量的净节省。另一项组合实验把 ThinkPrior 与 DAPO 搭配,在两组保持相同的 3840 次采样更新预算的前提下,生成的采样总数下降 10.6%。

适用边界与使用条件

该先验的价值集中在首次选择之前,它不需要目标策略先跑一轮,但此后的后验完全依赖目标策略自身的训练结果来修正,因此外部锚点的质量会决定初始排序是否可靠。方法本身只作用于提示选择环节,不触及损失与优化器,便于叠加到现有流程上。需要留意的是,固定提示池下的收益来自预算重分配,若期望的是总采样量的绝对下降,则需结合具体配置判断;论文中报告的 10.6% 降幅来自与 DAPO 的组合设置,并非单独使用时的普遍结论。

为什么值得看

解决 RLVR 训练中的冷启动浪费问题,显著提升计算资源利用率。

Qwen

信源1 家

  1. [1]arXiv cs.AI一手信源ThinkPrior: Zero-Rollout Difficulty Priors for Cold-Start Prompt Selection in RLVR

关键事实

  • ThinkPrior 使用外部锚点在离线阶段构建零回滚难度先验,避免冷启动时的回滚浪费。[1]

  • 在 Qwen2.5-Math-7B 测试中,该方法将早期静默组减少过半,前 30 步浪费回滚减少近五分之一。[1]

  • ThinkPrior+DAPO 组合减少了 10.6% 的生成回滚量,且保持相同的更新预算。[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。