提出OSC方法,解决离线强化学习中策略约束过保守问题。
Policy Constraint by Only Support Constraint for Offline Reinforcement Learning
- 仅约束策略支持集,不额外限制支持内动作选择
- 在多个基准上显著提升性能,缓解分布偏移问题
- 结合扩散模型刻画行为策略支持集,适合离线强化学习研究者
离线强化学习旨在利用预收集数据集优化策略以最大化累积奖励。然而,由于学习策略与行为策略间存在分布偏移,导致对分布外(OOD)动作的Q值计算出现误差。现有策略约束方法通常通过限制学习策略分布或将动作限制在行为策略支持集中来缓解此问题,但易导致过度保守,阻碍策略超越行为策略性能。本文提出仅支持约束(OSC),基于最大化学习策略在行为策略支持集上的总概率推导而来,仅限制策略支持集而不对支持内动作施加额外约束。此外,为充分挖掘新约束潜力,OSC使用扩散模型有效表征行为策略的支持集。在多个离线强化学习基准上的实验表明,OSC显著提升性能,缓解分布偏移挑战并减轻策略约束的保守性。代码已开源。
原文摘要 · Abstract (English)
Offline reinforcement learning (RL) aims to optimize a policy by using pre-collected datasets, to maximize cumulative rewards. However, offline reinforcement learning suffers challenges due to the distributional shift between the learned and behavior policies, leading to errors when computing Q-values for out-of-distribution (OOD) actions. To mitigate this issue, policy constraint methods aim to constrain the learned policy's distribution with the distribution of the behavior policy or confine action selection within the support of the behavior policy. However, current policy constraint methods tend to exhibit excessive conservatism, hindering the policy from further surpassing the behavior policy's performance. In this work, we present Only Support Constraint (OSC) which is derived from maximizing the total probability of learned policy in the support of behavior policy, to address the conservatism of policy constraint. OSC presents a regularization term that only restricts policies to the support without imposing extra constraints on actions within the support. Additionally, to fully harness the performance of the new policy constraints, OSC utilizes a diffusion model to effectively characterize the support of behavior policies. Experimental evaluations across a variety of offline RL benchmarks demonstrate that OSC significantly enhances performance, alleviating the challenges associated with distributional shifts and mitigating conservatism of policy constraints. Code is available at https://github.com/MoreanP/OSC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。