arXiv:2509.25424cs.LGcs.AI2025-09被引 9

让强化学习保持多样策略,避免陷入单一输出陷阱。

Polychromic Objectives for Reinforcement Learning

  • 设计多色目标函数,主动鼓励策略多样性
  • 在多个环境中提升成功率并更好应对扰动
  • 适合需要探索多种解法的复杂任务场景

强化学习微调(RLFT)是提升预训练策略在下游任务表现的主流方法。这些预训练策略在大规模数据上训练,能生成多种有潜力但不成熟的动作行为。然而,RLFT常出现策略多样性丧失、收敛到少数易被利用输出的问题,阻碍探索,限制预训练策略能力扩展和测试时计算资源的放大效益。为此,本文提出一种显式促进多样化生成的策略梯度目标,称为多色目标。我们进一步将近端策略优化(PPO)适配此目标:(1)采用藤蔓采样收集在线轨迹;(2)修改优势函数以反映新目标下的优势。在BabyAI、Minigrid和算法创造力任务上的实验表明,该方法显著提升成功解决环境配置的比例,并在大扰动下具备更强泛化能力。此外,在pass@$k$实验中,策略展现出更高覆盖率,证明其能有效维持并利用多样化的策略库。

原文摘要 · Abstract (English)

Reinforcement learning fine-tuning (RLFT) is a dominant paradigm for improving pretrained policies for downstream tasks. These pretrained policies, trained on large datasets, produce generations with a broad range of promising but unrefined behaviors. Often, a critical failure mode of RLFT arises when policies lose this diversity and collapse into a handful of easily exploitable outputs. This convergence hinders exploration, which is essential for expanding the capabilities of the pretrained policy and for amplifying the benefits of test-time compute scaling. To address this, we introduce an objective for policy gradient methods that explicitly enforces the exploration and refinement of diverse generations, which we call a polychromic objective. We then show how proximal policy optimization (PPO) can be adapted to optimize this objective. Our method (1) employs vine sampling to collect on-policy rollouts and (2) modifies the advantage function to reflect the advantage under our new objective. Experiments on BabyAI, Minigrid, and Algorithmic Creativity show that our method improves success rates by reliably solving a larger set of environment configurations and generalizes better under large perturbations. Moreover, when given multiple attempts in pass@$k$ experiments, the policy achieves substantially higher coverage, demonstrating its ability to maintain and exploit a diverse repertoire of strategies.

强化学习策略多样性PPO

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。