无需预设奖励函数,通过人类偏好学习奖励并训练智能体。
TROFI: Trajectory-Ranked Offline Inverse Reinforcement Learning
- 用人类偏好构建奖励函数,再用于标注离线数据集。
- 在D4RL上表现优于基线,接近真实奖励效果。
- 适合无奖励函数的现实场景,如游戏开发。
在离线强化学习中,智能体仅使用由源策略生成的固定转移数据集进行训练,但通常需要数据集带有奖励函数标签。在视频游戏开发等实际场景中,奖励函数未必可得。本文提出轨迹排序的离线逆强化学习(TROFI),一种无需预定义奖励函数即可有效学习策略的新方法。TROFI首先从人类偏好中学习奖励函数,再将其用于标注原始数据集,使其可用于策略训练。与现有方法不同,本方法不依赖最优轨迹。在D4RL基准上的实验表明,TROFI持续优于基线,性能接近使用真实奖励函数训练的结果。此外,我们在3D游戏环境中验证了该方法的有效性。对奖励模型的研究进一步表明:为确保价值函数与未来折扣奖励一致,必须设计良好且易于学习的奖励函数。
原文摘要 · Abstract (English)
In offline reinforcement learning, agents are trained using only a fixed set of stored transitions derived from a source policy. However, this requires that the dataset be labeled by a reward function. In applied settings such as video game development, the availability of the reward function is not always guaranteed. This paper proposes Trajectory-Ranked OFfline Inverse reinforcement learning (TROFI), a novel approach to effectively learn a policy offline without a pre-defined reward function. TROFI first learns a reward function from human preferences, which it then uses to label the original dataset making it usable for training the policy. In contrast to other approaches, our method does not require optimal trajectories. Through experiments on the D4RL benchmark we demonstrate that TROFI consistently outperforms baselines and performs comparably to using the ground truth reward to learn policies. Additionally, we validate the efficacy of our method in a 3D game environment. Our studies of the reward model highlight the importance of the reward function in this setting: we show that to ensure the alignment of a value function to the actual future discounted reward, it is fundamental to have a well-engineered and easy-to-learn reward function.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。