用成功者路径学习奖励,训练更稳效率更高
SR-Reward: Taking The Path More Traveled
- 基于成功者路径的预期状态访问,解耦奖励与策略
- 在D4RL上表现媲美真奖励算法,且抗分布外数据
- 适合数据质量一般时的离线强化学习场景
本文提出一种直接从离线演示中学习奖励函数的新方法。与传统逆强化学习不同,该方法将奖励函数与学习者策略解耦,消除了两者间的对抗性交互,从而实现更稳定高效的训练过程。所提出的奖励函数SR-Reward利用后继表示(Successor Representation, SR),根据演示策略下的未来状态访问期望来编码状态,并通过贝尔曼方程可与多数强化学习算法协同训练,无需修改现有训练流程。此外,引入负采样策略以减少分布外数据的奖励值,缓解过估计误差,天然赋予采用该奖励的强化学习算法保守性偏差。在D4RL基准上的实验表明,其性能与具备真实奖励的离线强化学习算法及行为克隆等模仿学习技术相当。消融实验还揭示了该方法在不同数据量和质量下的优劣边界。
原文摘要 · Abstract (English)
In this paper, we propose a novel method for learning reward functions directly from offline demonstrations. Unlike traditional inverse reinforcement learning (IRL), our approach decouples the reward function from the learner's policy, eliminating the adversarial interaction typically required between the two. This results in a more stable and efficient training process. Our reward function, called \textit{SR-Reward}, leverages successor representation (SR) to encode a state based on expected future states' visitation under the demonstration policy and transition dynamics. By utilizing the Bellman equation, SR-Reward can be learned concurrently with most reinforcement learning (RL) algorithms without altering the existing training pipeline. We also introduce a negative sampling strategy to mitigate overestimation errors by reducing rewards for out-of-distribution data, thereby enhancing robustness. This strategy inherently introduces a conservative bias into RL algorithms that employ the learned reward. We evaluate our method on the D4RL benchmark, achieving competitive results compared to offline RL algorithms with access to true rewards and imitation learning (IL) techniques like behavioral cloning. Moreover, our ablation studies on data size and quality reveal the advantages and limitations of SR-Reward as a proxy for true rewards.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。