arXiv:2410.01643cs.LGcs.AI2024-10ICML

用相似性核稳定离线强化学习的价值函数估计。

Stable Offline Value Function Learning with Bisimulation-based Representations

  • 基于状态动作对相似性构建核表示,提升表征稳定性。
  • 相比基线方法,价值误差更低,且收敛更稳定。
  • 适合需要可靠离线评估的RL研究与应用。

在强化学习中,离线价值函数学习是指利用离线数据集估算每个状态在遵循固定目标策略时的期望折扣回报。该过程的稳定性(即是否收敛到固定点)关键取决于状态-动作对的表征质量。若表征学习不佳,可能导致价值函数学习不稳定甚至发散。因此,通过显式设计状态-动作表征来稳定价值函数学习至关重要。近期,基于双仿真(bisimulation)的算法在控制任务中展现出塑造表征的潜力。然而,这类方法能否真正稳定价值函数学习仍不明确。本文回答了这一问题并给出了肯定答案:提出一种名为核表示离线策略评估(krope)的双仿真算法。krope 使用核函数构造状态-动作表征,使得具有相似即时奖励且在目标策略下导致相似下一状态-动作对的状态-动作对,其表征也趋于相似。我们证明了krope能够:1)学习稳定的表征;2)相较于基线方法降低价值误差。理论分析揭示了双仿真方法的稳定性机制,提示从业者可借助此类方法提升离线评估的稳定性与准确性。

原文摘要 · Abstract (English)

In reinforcement learning, offline value function learning is the procedure of using an offline dataset to estimate the expected discounted return from each state when taking actions according to a fixed target policy. The stability of this procedure, i.e., whether it converges to its fixed-point, critically depends on the representations of the state-action pairs. Poorly learned representations can make value function learning unstable, or even divergent. Therefore, it is critical to stabilize value function learning by explicitly shaping the state-action representations. Recently, the class of bisimulation-based algorithms have shown promise in shaping representations for control. However, it is still unclear if this class of methods can \emph{stabilize} value function learning. In this work, we investigate this question and answer it affirmatively. We introduce a bisimulation-based algorithm called kernel representations for offline policy evaluation (\textsc{krope}). \textsc{krope} uses a kernel to shape state-action representations such that state-action pairs that have similar immediate rewards and lead to similar next state-action pairs under the target policy also have similar representations. We show that \textsc{krope}: 1) learns stable representations and 2) leads to lower value error than baselines. Our analysis provides new theoretical insight into the stability properties of bisimulation-based methods and suggests that practitioners can use these methods to improve the stability and accuracy of offline evaluation of reinforcement learning agents.

强化学习离线评估表征学习双仿真

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