用视频教机器人学任务,让强化学习更省时间。
On-Robot Reinforcement Learning with Goal-Contrastive Rewards
- 从无动作视频中自动学习密集奖励信号
- 实验显示比基线方法多解决约一倍任务
- 适合希望减少实机训练成本的研究者
强化学习有望让机器人在真实世界中自主学习。但稀疏奖励导致探索效率低,训练耗时过长。现有密集奖励设计依赖人工标注,成本高且需领域知识。本文提出GCR(目标对比奖励),一种可基于被动视频演示训练的密集奖励学习方法。该方法仅需无动作标签的视频,无需动作信息即可扩展使用。GCR结合隐式价值损失与目标对比损失,分别建模成功轨迹的奖励提升和区分成功/失败轨迹。我们在RoboMimic与MimicGen模拟操控任务,以及真实世界中的Franka机械臂和Spot四足机器人上进行测试。结果表明,GCR显著提升样本效率,使无模型强化学习解决的任务数量约为基线方法的两倍。此外,还验证了跨机器人形态的正向迁移效果,即利用人类或其它机器人视频可有效指导新机器人学习。
原文摘要 · Abstract (English)
Reinforcement Learning (RL) has the potential to enable robots to learn from their own actions in the real world. Unfortunately, RL can be prohibitively expensive, in terms of on-robot runtime, due to inefficient exploration when learning from a sparse reward signal. Designing dense reward functions is labour-intensive and requires domain expertise. In our work, we propose GCR (Goal-Contrastive Rewards), a dense reward function learning method that can be trained on passive video demonstrations. By using videos without actions, our method is easier to scale, as we can use arbitrary videos. GCR combines two loss functions, an implicit value loss function that models how the reward increases when traversing a successful trajectory, and a goal-contrastive loss that discriminates between successful and failed trajectories. We perform experiments in simulated manipulation environments across RoboMimic and MimicGen tasks, as well as in the real world using a Franka arm and a Spot quadruped. We find that GCR leads to a more-sample efficient RL, enabling model-free RL to solve about twice as many tasks as our baseline reward learning methods. We also demonstrate positive cross-embodiment transfer from videos of people and of other robots performing a task. Website: https://gcr-robot.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。