arXiv:2605.08315cs.LG2026-05

让大模型基于完整轨迹改进策略,比只看得分更准更快。

Reflective Prompted Policy Optimization: Trajectory-Grounded Revision and Salience Bias

论文配图:Reflective Prompted Policy Optimization: Trajectory-Grounded Revision and Salience Bias
图 1 · 摘自论文原文
  • 用两个大模型分工:一个提策略,一个分析完整执行过程并优化
  • 在10个环境中平均表现最佳,CartPole任务500次内达接近最优
  • 解决模型过度关注失败案例的偏见问题,适合做策略优化研究

现有基于大模型的策略优化器仅接收标量奖励(如得分0.45),无法获取智能体是否卡在循环、第三步掉进坑里,或19次成功仅一次崩溃等行为细节。本文提出反射式提示策略优化(R2PO),一种两阶段大模型框架,将标量奖励与轨迹级行为证据结合。搜索模型提出候选策略参数,环境执行后,批评模型分析生成的多轮轨迹,基于观察到的状态、动作和奖励提出具体修正。在十个环境中,消融实验表明,分离全局搜索与行为驱动修订,并使用筛选机制过滤高方差修改是性能提升的关键。我们发现主导失败模式为显著性偏见:当面对多个轨迹时,批评模型会执着于修复单一失败案例,即使多数轨迹表现良好。在三轨迹版本中,该行为解释了CartPole任务76.6%的性能退化。R2PO通过综合统计、中位轨迹选择和修订规则缓解此问题。使用200亿参数开源模型,R2PO在所有十项任务中达到最高均值最优奖励,在部分任务中约500轮内即达近优表现,且训练稳定性远超深度强化学习及以往基于大模型的方法。结果表明,将轨迹视为上下文中的第一类证据,而非简化为标量回报,能显著提升小规模大模型在策略空间中的搜索效率,使其学习更快、诊断更准、可靠改进外部控制器。

原文摘要 · Abstract (English)

Existing LLM-based policy optimizers see only scalar rewards: that a policy scored 0.45, but not whether the agent got stuck in a loop, fell into a hole on the third step, or performed well on 19 out of 20 rollouts and failed catastrophically on one. We propose Reflective Prompted Policy Optimization (R2PO), a two-stage LLM framework for policy search over compact policy classes that augments scalar reward feedback with trajectory-level behavioral evidence. A Search-LLM proposes candidate policy parameters; the environment executes them; a Critic-LLM inspects the resulting rollouts and proposes targeted revisions grounded in observed states, actions, and rewards. Across ten environments, ablations show R2PO's gains require separating global search from behavior-grounded revision and using selection to filter high-variance edits. We further identify a dominant failure mode, salience bias: when presented with multiple rollouts, the Critic-LLM fixates on improving a single failure even when most trajectories succeed. In a three-trajectory variant where the Critic-LLM sees the best, worst, and median rollout, this behavior explains 76.6% of regressions on CartPole. R2PO mitigates this by reasoning over aggregate rollout statistics, median-trajectory selection, and a revision rule. Using a 20B open-weight model, R2PO achieves the highest mean best reward across all ten environments, reaches near-optimal performance substantially earlier (e.g., near-maximum CartPole reward within ~500 episodes), and trains far more stably than both deep RL and prior LLM-based methods. These results show that treating trajectories as first-class in-context evidence, rather than artifacts reduced to scalar returns, changes how even comparatively small LLMs search over policy spaces, enabling them to learn faster, diagnose more precisely, and reliably improve external controllers.

策略优化大模型强化学习轨迹分析

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