让大模型学会生成多样化解法,提升推理时搜索效果。
Vector Policy Optimization: Training for Diversity Improves Test-Time Search
- 用向量奖励训练模型输出多解,适应不同任务需求。
- 在四个任务中优于或持平最强标量强化学习基线。
- 特别适合需要多样解的进化搜索场景,如代码生成。
大语言模型需在未见环境中即插即用,并适配如AlphaEvolve等基于推理时搜索的流程,该流程依赖于多种任务特定奖励函数来选择轨迹。然而,当前主流的大模型后训练范式仅优化单一标量奖励,导致响应分布熵值低,难以满足推理时搜索所需的多样性。本文提出向量策略优化(VPO),一种强化学习算法,显式训练模型预测多样化的下游奖励函数并生成多样化解法。VPO利用实际中奖励常为向量的特点,如代码生成中的每测试案例正确率,或不同用户角色/奖励模型。VPO本质上是GRPO优势估计器的即插即用替代方案,但训练模型输出一组解,每解针对向量奖励空间中的不同权衡进行专业化。在四个任务上,VPO在测试时搜索(如pass@k、best@k)表现匹配或超越最强标量强化学习基线,且随着搜索预算增加,差距进一步扩大。对于进化搜索,使用VPO的模型可解决GRPO模型完全无法处理的问题。随着测试时搜索日益标准化,优化多样性可能需成为默认的后训练目标。
原文摘要 · Abstract (English)
Language models must now generalize out of the box to novel environments and work inside inference-scaling search procedures, such as AlphaEvolve, that select rollouts with a variety of task-specific reward functions. Unfortunately, the standard paradigm of LLM post-training optimizes a pre-specified scalar reward, often leading current LLMs to produce low-entropy response distributions and thus to struggle at displaying the diversity that inference-time search will require. We propose Vector Policy Optimization (VPO), an RL algorithm that explicitly trains policies to anticipate diverse downstream reward functions and to produce diverse solutions. VPO exploits that rewards are often vector-valued in practice, like per-test-case correctness in code generation or, say, multiple different user personas or reward models. VPO is essentially a drop-in replacement for the GRPO advantage estimator, but it trains the LLM to output a set of solutions where individual solutions specialize to different trade-offs in the vector reward space. Across four tasks, VPO matches or beats the strongest scalar RL baselines on test-time search (e.g. pass@k and best@k), with the gap widening as the search budget grows. For evolutionary search, VPO models unlock problems that GRPO models cannot solve at all. As test-time search becomes more standardized, optimizing for diversity may need to become the default post-training objective.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。