arXiv:2509.23209cs.LGcs.AI2025-09

解决上下文强化学习中性能无法持续提升的难题。

Towards Monotonic Improvement in In-Context Reinforcement Learning

  • 引入上下文价值作为理想性能信号,缓解上下文歧义问题。
  • 在暗室和迷你网格环境中实现性能稳定提升,避免退化。
  • 适合关注快速适应与稳定改进的强化学习研究者。

上下文强化学习(ICRL)通过利用过往经验作为上下文,使智能体在不更新参数的情况下快速适应新任务。现有方法在在线强化学习数据上训练大序列模型,期望测试时表现持续提升。然而实验发现,这些模型在测试阶段无法像训练数据那样实现持续改进。理论上,我们识别出该现象源于上下文歧义:模型自身随机动作生成的交互历史可能误导性地模拟低效策略,导致恶性循环。为此,我们提出上下文价值感知的ICRL(CV-ICRL),在训练阶段引入上下文价值,作为给定上下文下理论最优性能的显式信号。随着上下文扩展,上下文价值可包含更多任务相关资讯,理想性能应非递减。我们证明,上下文价值能收紧相对于理想单调提升策略的性能差距下界,并提出两种训练与测试阶段的估计方法。在暗室(Dark Room)和迷你网格(Minigrid)环境上的实验表明,CV-ICRL有效缓解性能下降,显著提升各类任务与环境下的整体能力。

原文摘要 · Abstract (English)

In-Context Reinforcement Learning (ICRL) has emerged as a promising paradigm for developing agents that can rapidly adapt to new tasks by leveraging past experiences as context, without updating their parameters. Recent approaches train large sequence models on monotonic policy improvement data from online RL, aiming to a continue improved testing time performance. However, our experimental analysis reveals a critical flaw: these models cannot show a continue improvement like the training data during testing time. Theoretically, we identify this phenomenon as Contextual Ambiguity, where the model's own stochastic actions can generate an interaction history that misleadingly resembles that of a sub-optimal policy from the training data, initiating a vicious cycle of poor action selection. To resolve the Contextual Ambiguity, we introduce Context Value into training phase and propose Context Value Informed ICRL (CV-ICRL). CV-ICRL use Context Value as an explicit signal representing the ideal performance theoretically achievable by a policy given the current context. As the context expands, Context Value could include more task-relevant information, and therefore the ideal performance should be non-decreasing. We prove that the Context Value tightens the lower bound on the performance gap relative to an ideal, monotonically improving policy. We fruther propose two methods for estimating Context Value at both training and testing time. Experiments conducted on the Dark Room and Minigrid testbeds demonstrate that CV-ICRL effectively mitigates performance degradation and improves overall ICRL abilities across various tasks and environments. The source code and data of this paper are available at https://github.com/Bluixe/towards_monotonic_improvement .

强化学习上下文学习性能提升

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