不依赖保守策略,用贝叶斯方法实现长时序离线强化学习
Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism
- 基于后验建模世界模型,通过历史依赖策略最大化期望回报
- 在7个数据集上达到新SOTA,支持数百步的长时序回放
- 适合低质量或覆盖不足的数据集,超越传统保守算法
主流离线强化学习方法依赖显式保守性,惩罚数据外动作或限制回放时长。本文质疑该原则的普适性,提出一种互补的贝叶斯视角用于测试时适应。通过建模世界模型的后验分布,并训练历史依赖代理以最大化期望回报,该方法直接应对认知不确定性,无需显式保守性。在老虎机任务中,贝叶斯方法在低质量数据上表现优于保守方法。扩展至真实任务,我们发现移除保守性后,长时序回放对控制价值过估计至关重要。提出NEUBAY算法,通过设计选择实现长时序回放的同时缓解模型误差累积。在D4RL和NeoRL基准上,NEUBAY性能媲美领先保守算法,在7个数据集上取得新SOTA,回放时长达数百步。最后,通过数据质量与覆盖度分析,明确指出NEUBAY适用场景。
原文摘要 · Abstract (English)
Popular offline reinforcement learning (RL) methods rely on explicit conservatism, penalizing out-of-dataset actions or restricting rollout horizons. We question the universality of this principle and revisit a complementary Bayesian perspective for test-time adaptation. By modeling a posterior over world models and training a history-dependent agent to maximize expected return, the Bayesian approach directly addresses epistemic uncertainty without explicit conservatism. We first illustrate in a bandit setting that Bayesianism excels on low-quality datasets where conservatism fails. Scaling to realistic tasks, we find that long-horizon rollouts are essential to control value overestimation once conservatism is removed. We introduce design choices that enable learning from long-horizon rollouts while mitigating compounding model errors, yielding our algorithm, NEUBAY, grounded in the neutral Bayesian principle. On D4RL and NeoRL benchmarks, NEUBAY is competitive with leading conservative algorithms, achieving new state-of-the-art on 7 datasets with rollout horizons of several hundred steps. Finally, we characterize datasets by quality and coverage to identify when NEUBAY is preferable to conservative methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。