通过迭代优化策略,让强化学习在离线数据上更稳定高效。
An Optimal Discriminator Weighted Imitation Perspective for Reinforcement Learning
- 用判别器权重引导模仿学习,逐步逼近最优策略分布
- 在所有测试数据集上超越主流方法,性能与稳定性双提升
- 适合处理带噪声或低质量示范数据的离线强化学习任务
我们提出迭代双重强化学习(IDRL),从最优判别器加权模仿视角解决强化学习问题。实验发现,仅用离线数据集和额外专家数据集训练判别器,再进行加权行为克隆即可取得良好效果。该最优判别器权重与双强化学习(Dual-RL)中学习的访问分布比非常接近,但现有方法未能正确估计该比值。IDRL提出一种修正机制,通过迭代方式逼近离线数据中的最优访问分布比,无需额外专家数据。每轮迭代中,利用前一轮学习的比值剔除零权重的次优转移,并在剩余子数据集上运行Dual-RL。这相当于用优化后的访问分布替代原始行为分布,理论上形成逐步趋近最优判别器权重的课程学习。我们在多种离线数据集上验证了IDRL的有效性,包括D4RL数据集和更具现实性的污染示范数据。结果表明,IDRL在所有数据集上均优于强基线模型Primal-RL和Dual-RL,在性能与稳定性方面表现更优。
原文摘要 · Abstract (English)
We introduce Iterative Dual Reinforcement Learning (IDRL), a new method that takes an optimal discriminator-weighted imitation view of solving RL. Our method is motivated by a simple experiment in which we find training a discriminator using the offline dataset plus an additional expert dataset and then performing discriminator-weighted behavior cloning gives strong results on various types of datasets. That optimal discriminator weight is quite similar to the learned visitation distribution ratio in Dual-RL, however, we find that current Dual-RL methods do not correctly estimate that ratio. In IDRL, we propose a correction method to iteratively approach the optimal visitation distribution ratio in the offline dataset given no addtional expert dataset. During each iteration, IDRL removes zero-weight suboptimal transitions using the learned ratio from the previous iteration and runs Dual-RL on the remaining subdataset. This can be seen as replacing the behavior visitation distribution with the optimized visitation distribution from the previous iteration, which theoretically gives a curriculum of improved visitation distribution ratios that are closer to the optimal discriminator weight. We verify the effectiveness of IDRL on various kinds of offline datasets, including D4RL datasets and more realistic corrupted demonstrations. IDRL beats strong Primal-RL and Dual-RL baselines in terms of both performance and stability, on all datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。