用生成模型避免离群动作,让离线强化学习更安全可靠。
Generative OOD-regularized Model-based Policy Optimization

- 引入生成密度模型,限制策略更新在数据密集区域。
- 在真实医疗数据集上性能提升17%,优于顶尖基线方法。
- 适合动态稳定环境,对密度估计精度要求高。
我们研究基于离线强化学习的序列决策问题。传统离线RL策略在仅依赖稀疏离线表示时,可能导致分布外(OOD)动作。为确保稀疏状态-动作空间中的安全策略,我们探索将密度估计模型融入基于模型的RL,以避开OOD区域。生成模型能显式建模稀疏状态-动作空间的密度。基于此,我们提出生成式OOD正则化模型基础策略优化(GORMPO),一种利用生成密度建模限制策略更新至高密度数据区域的离线RL算法。进一步,我们检验更好的OOD检测是否带来更优的基于模型离线策略。在真实医疗数据集和稀疏离线RL数据集上,比较了多种密度估计器的OOD检测能力及其在GORMPO框架中的表现。理论上,在温和假设下保证了GORMPO的性能。实验证明,GORMPO在真实医疗数据集上比最先进基线高出17%,并在离线RL数据集上提升了基础模型性能。结果表明,在动态稳定的环境中,更好的OOD检测通常带来更优策略;而在动态不确定时,保守惩罚结合差的密度估计反而更优。
原文摘要 · Abstract (English)
We study sequential decision-making with offline reinforcement learning (RL). Traditional offline RL policies may result in out-of-distribution (OOD) actions when training relies only on sparse offline representations. To ensure safe offline policies in a sparse state-action space, we explore how density estimation models can be integrated into model-based RL methods to avoid the OOD regions. Generative models are capable of explicitly modeling the density in sparse state-action spaces. Building on this, we introduce Generative OOD-regularized Model-based Policy Optimization (GORMPO), a density-regularized offline RL algorithm that uses generative density modeling to restrict policy updates to high-density areas of the dataset. Furthermore, we examine whether better OOD detection corresponds to better model-based offline policies. We compare (1) the OOD detection capabilities of various density estimators and (2) their performance within the GORMPO framework on a real-world medical dataset and sparse offline RL datasets. We theoretically guarantee GORMPO's performance under mild assumptions. Empirically, GORMPO outperforms state-of-the-art baselines by 17% on a real-world medical dataset and enhances the base model on the offline RL datasets. Our empirical findings show that better OOD detection generally results in improved policies in environments with stable dynamics, while conservative penalties with poor density estimation are favored when dynamics are uncertain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。