RVT-2: Learning Precise Manipulation from Few Demonstrations

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Goyal, Ankit, Blukis, Valts, Xu, Jie, Guo, Yijie, Chao, Yu-Wei, Fox, Dieter
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916284904505344
author Goyal, Ankit
Blukis, Valts
Xu, Jie
Guo, Yijie
Chao, Yu-Wei
Fox, Dieter
author_facet Goyal, Ankit
Blukis, Valts
Xu, Jie
Guo, Yijie
Chao, Yu-Wei
Fox, Dieter
contents In this work, we study how to build a robotic system that can solve multiple 3D manipulation tasks given language instructions. To be useful in industrial and household domains, such a system should be capable of learning new tasks with few demonstrations and solving them precisely. Prior works, like PerAct and RVT, have studied this problem, however, they often struggle with tasks requiring high precision. We study how to make them more effective, precise, and fast. Using a combination of architectural and system-level improvements, we propose RVT-2, a multitask 3D manipulation model that is 6X faster in training and 2X faster in inference than its predecessor RVT. RVT-2 achieves a new state-of-the-art on RLBench, improving the success rate from 65% to 82%. RVT-2 is also effective in the real world, where it can learn tasks requiring high precision, like picking up and inserting plugs, with just 10 demonstrations. Visual results, code, and trained model are provided at: https://robotic-view-transformer-2.github.io/.
format Preprint
id arxiv_https___arxiv_org_abs_2406_08545
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RVT-2: Learning Precise Manipulation from Few Demonstrations
Goyal, Ankit
Blukis, Valts
Xu, Jie
Guo, Yijie
Chao, Yu-Wei
Fox, Dieter
Robotics
Artificial Intelligence
Computer Vision and Pattern Recognition
In this work, we study how to build a robotic system that can solve multiple 3D manipulation tasks given language instructions. To be useful in industrial and household domains, such a system should be capable of learning new tasks with few demonstrations and solving them precisely. Prior works, like PerAct and RVT, have studied this problem, however, they often struggle with tasks requiring high precision. We study how to make them more effective, precise, and fast. Using a combination of architectural and system-level improvements, we propose RVT-2, a multitask 3D manipulation model that is 6X faster in training and 2X faster in inference than its predecessor RVT. RVT-2 achieves a new state-of-the-art on RLBench, improving the success rate from 65% to 82%. RVT-2 is also effective in the real world, where it can learn tasks requiring high precision, like picking up and inserting plugs, with just 10 demonstrations. Visual results, code, and trained model are provided at: https://robotic-view-transformer-2.github.io/.
title RVT-2: Learning Precise Manipulation from Few Demonstrations
topic Robotics
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.08545