用想象力限制策略避免离线强化学习中动作估值过高
Imagination-Limited Q-Learning for Offline Reinforcement Learning
- 用动态模型模拟未知动作价值,再与行为数据最大值裁剪
- 在D4RL基准上达到当前最优性能,且未知状态动作估值偏差可控
- 适合追求高精度离线策略的科研与工业应用
离线强化学习旨在仅从历史数据中学习更优策略,但常因对分布外(OOD)动作的估值过于乐观而受阻。现有方法多依赖策略约束或保守价值正则,但可能施加过强限制或引入偏差,抑制性能提升。为此,我们提出想象有限Q学习(ILQ),在合理范围内保留对分布外动作的乐观评估。具体地,利用动态模型生成分布外动作的价值估计,并以行为数据中的最大值进行裁剪。该设计在最大程度保持对分布外动作合理评估的同时,避免其过度乐观。理论上,我们在表格马尔可夫决策过程下证明了ILQ的收敛性,表明分布外状态-动作对的估值误差与分布内情况同阶,说明估值偏差得到有效缓解。实验上,本方法在D4RL基准的多种任务中均取得当前最优表现。
原文摘要 · Abstract (English)
Offline reinforcement learning seeks to derive improved policies entirely from historical data but often struggles with over-optimistic value estimates for out-of-distribution (OOD) actions. This issue is typically mitigated via policy constraint or conservative value regularization methods. However, these approaches may impose overly constraints or biased value estimates, potentially limiting performance improvements. To balance exploitation and restriction, we propose an Imagination-Limited Q-learning (ILQ) method, which aims to maintain the optimism that OOD actions deserve within appropriate limits. Specifically, we utilize the dynamics model to imagine OOD action-values, and then clip the imagined values with the maximum behavior values. Such design maintains reasonable evaluation of OOD actions to the furthest extent, while avoiding its over-optimism. Theoretically, we prove the convergence of the proposed ILQ under tabular Markov decision processes. Particularly, we demonstrate that the error bound between estimated values and optimality values of OOD state-actions possesses the same magnitude as that of in-distribution ones, thereby indicating that the bias in value estimates is effectively mitigated. Empirically, our method achieves state-of-the-art performance on a wide range of tasks in the D4RL benchmark.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。