RACER: Rich Language-Guided Failure Recovery Policies for Imitation Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dai, Yinpei, Lee, Jayjun, Fazeli, Nima, Chai, Joyce
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913513090318336
author Dai, Yinpei
Lee, Jayjun
Fazeli, Nima
Chai, Joyce
author_facet Dai, Yinpei
Lee, Jayjun
Fazeli, Nima
Chai, Joyce
contents Developing robust and correctable visuomotor policies for robotic manipulation is challenging due to the lack of self-recovery mechanisms from failures and the limitations of simple language instructions in guiding robot actions. To address these issues, we propose a scalable data generation pipeline that automatically augments expert demonstrations with failure recovery trajectories and fine-grained language annotations for training. We then introduce Rich languAge-guided failure reCovERy (RACER), a supervisor-actor framework, which combines failure recovery data with rich language descriptions to enhance robot control. RACER features a vision-language model (VLM) that acts as an online supervisor, providing detailed language guidance for error correction and task execution, and a language-conditioned visuomotor policy as an actor to predict the next actions. Our experimental results show that RACER outperforms the state-of-the-art Robotic View Transformer (RVT) on RLbench across various evaluation settings, including standard long-horizon tasks, dynamic goal-change tasks and zero-shot unseen tasks, achieving superior performance in both simulated and real world environments. Videos and code are available at: https://rich-language-failure-recovery.github.io.
format Preprint
id arxiv_https___arxiv_org_abs_2409_14674
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RACER: Rich Language-Guided Failure Recovery Policies for Imitation Learning
Dai, Yinpei
Lee, Jayjun
Fazeli, Nima
Chai, Joyce
Robotics
Computation and Language
Computer Vision and Pattern Recognition
Developing robust and correctable visuomotor policies for robotic manipulation is challenging due to the lack of self-recovery mechanisms from failures and the limitations of simple language instructions in guiding robot actions. To address these issues, we propose a scalable data generation pipeline that automatically augments expert demonstrations with failure recovery trajectories and fine-grained language annotations for training. We then introduce Rich languAge-guided failure reCovERy (RACER), a supervisor-actor framework, which combines failure recovery data with rich language descriptions to enhance robot control. RACER features a vision-language model (VLM) that acts as an online supervisor, providing detailed language guidance for error correction and task execution, and a language-conditioned visuomotor policy as an actor to predict the next actions. Our experimental results show that RACER outperforms the state-of-the-art Robotic View Transformer (RVT) on RLbench across various evaluation settings, including standard long-horizon tasks, dynamic goal-change tasks and zero-shot unseen tasks, achieving superior performance in both simulated and real world environments. Videos and code are available at: https://rich-language-failure-recovery.github.io.
title RACER: Rich Language-Guided Failure Recovery Policies for Imitation Learning
topic Robotics
Computation and Language
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.14674