用预训练价值模型提升稀疏强化学习的稳定性和效率
$V_{0.5}$: Generalist Value Model as a Prior for Sparse RL Rollouts
- 结合预训练价值模型先验与实时采样均值,动态调整策略
- 在每组仅4个样本的极端稀疏条件下,误差降低超10%
- 适合追求高效、稳定强化学习训练的研究者和工程师
在可验证奖励的强化学习(RLVR)中,构建稳健的优势基线对策略梯度至关重要,能有效引导策略模型强化期望行为。近期研究提出通用价值模型(如 $V_0$),通过上下文编码模型能力实现预训练价值估计,无需与策略模型同步更新。本文提出 $V_{0.5}$,自适应融合该价值模型预测的先验基线与稀疏回溯所得经验均值,构建兼具计算效率与极低方差的鲁棒基线。具体引入实时统计检验与动态预算分配机制,平衡稀疏采样带来的高方差与价值模型固有的系统性偏差(或幻觉)。通过实时检验先验可靠性,系统按需分配额外回溯预算,最小化基线估计的均方误差(MSE),确保在极端稀疏条件(组大小为4)下仍保持稳定策略梯度。六项数学推理基准测试表明,$V_{0.5}$ 显著优于 GRPO 与 DAPO,收敛更快,性能提升超过10%。
原文摘要 · Abstract (English)
In Reinforcement Learning with Verifiable Rewards (RLVR), constructing a robust advantage baseline is critical for policy gradients, effectively guiding the policy model to reinforce desired behaviors. Recent research has introduced Generalist Value Models (such as $V_0$), which achieve pre-trained value estimation by explicitly encoding model capabilities in-context, eliminating the need to synchronously update the value model alongside the policy model. In this paper, we propose $V_{0.5}$, which adaptively fuses the baseline predicted by such value model (acting as a prior) with the empirical mean derived from sparse rollouts. This constructs a robust baseline that balances computational efficiency with extremely low variance. Specifically, we introduce a real-time statistical testing and dynamic budget allocation. This balances the high variance caused by sparse sampling against the systematic bias (or hallucinations) inherent in the value model's prior. By constructing a hypothesis test to evaluate the prior's reliability in real-time, the system dynamically allocates additional rollout budget on demand. This mechanism minimizes the baseline estimator's Mean Squared Error (MSE), guaranteeing stable policy gradients, even under extreme sparsity with a group size of 4. Extensive evaluations across six mathematical reasoning benchmarks demonstrate that $V_{0.5}$ significantly outperforms GRPO and DAPO, achieving faster convergence and over some 10% performance improvement.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。