arXiv:2411.05273cs.ROcs.AI2024-11中稿 · CoRL被引 29

用视觉语言模型自动标注离线数据奖励,让机器人学会真实世界穿衣任务。

Real-World Offline Reinforcement Learning from Vision Language Model Feedback

  • 通过视觉语言模型的偏好反馈自动生成奖励标签
  • 在真实机器人穿衣任务中实现有效策略,优于行为克隆和逆强化学习
  • 无需人工标注,适合安全敏感的现实场景

离线强化学习可在不进行在线交互的情况下,从预先收集的次优数据集中学习策略,特别适用于真实世界机器人和高安全性场景。然而,现有方法通常依赖人工标注的任务奖励,过程耗时且困难,尤其当真实状态难以确定时。本文基于先前工作RL-VLM-F,提出一种新系统:利用视觉语言模型的偏好反馈与任务文本描述,自动为离线数据集生成奖励标签,并使用该标签数据进行离线强化学习。我们在复杂的真实世界机器人协助穿衣任务中验证了该系统,首先通过视觉语言模型在次优离线数据上学习奖励函数,再结合隐式Q学习构建有效的穿衣策略。该方法在刚性与柔性物体操作的仿真任务中也表现优异,显著优于行为克隆和逆强化学习等基线方法。本研究提出一种可自动完成奖励标注与策略学习的系统,适用于无标签、次优的离线数据。

原文摘要 · Abstract (English)

Offline reinforcement learning can enable policy learning from pre-collected, sub-optimal datasets without online interactions. This makes it ideal for real-world robots and safety-critical scenarios, where collecting online data or expert demonstrations is slow, costly, and risky. However, most existing offline RL works assume the dataset is already labeled with the task rewards, a process that often requires significant human effort, especially when ground-truth states are hard to ascertain (e.g., in the real-world). In this paper, we build on prior work, specifically RL-VLM-F, and propose a novel system that automatically generates reward labels for offline datasets using preference feedback from a vision-language model and a text description of the task. Our method then learns a policy using offline RL with the reward-labeled dataset. We demonstrate the system's applicability to a complex real-world robot-assisted dressing task, where we first learn a reward function using a vision-language model on a sub-optimal offline dataset, and then we use the learned reward to employ Implicit Q learning to develop an effective dressing policy. Our method also performs well in simulation tasks involving the manipulation of rigid and deformable objects, and significantly outperform baselines such as behavior cloning and inverse RL. In summary, we propose a new system that enables automatic reward labeling and policy learning from unlabeled, sub-optimal offline datasets.

离线强化学习视觉语言模型机器人任务自动标注

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。