3D-MVP: 3D Multiview Pretraining for Robotic Manipulation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qian, Shengyi, Mo, Kaichun, Blukis, Valts, Fouhey, David F., Fox, Dieter, Goyal, Ankit
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908278609412096
author Qian, Shengyi
Mo, Kaichun
Blukis, Valts
Fouhey, David F.
Fox, Dieter
Goyal, Ankit
author_facet Qian, Shengyi
Mo, Kaichun
Blukis, Valts
Fouhey, David F.
Fox, Dieter
Goyal, Ankit
contents Recent works have shown that visual pretraining on egocentric datasets using masked autoencoders (MAE) can improve generalization for downstream robotics tasks. However, these approaches pretrain only on 2D images, while many robotics applications require 3D scene understanding. In this work, we propose 3D-MVP, a novel approach for 3D Multi-View Pretraining using masked autoencoders. We leverage Robotic View Transformer (RVT), which uses a multi-view transformer to understand the 3D scene and predict gripper pose actions. We split RVT's multi-view transformer into visual encoder and action decoder, and pretrain its visual encoder using masked autoencoding on large-scale 3D datasets such as Objaverse. We evaluate 3D-MVP on a suite of virtual robot manipulation tasks and demonstrate improved performance over baselines. Our results suggest that 3D-aware pretraining is a promising approach to improve generalization of vision-based robotic manipulation policies. Project site: https://jasonqsy.github.io/3DMVP
format Preprint
id arxiv_https___arxiv_org_abs_2406_18158
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle 3D-MVP: 3D Multiview Pretraining for Robotic Manipulation
Qian, Shengyi
Mo, Kaichun
Blukis, Valts
Fouhey, David F.
Fox, Dieter
Goyal, Ankit
Robotics
Computer Vision and Pattern Recognition
Recent works have shown that visual pretraining on egocentric datasets using masked autoencoders (MAE) can improve generalization for downstream robotics tasks. However, these approaches pretrain only on 2D images, while many robotics applications require 3D scene understanding. In this work, we propose 3D-MVP, a novel approach for 3D Multi-View Pretraining using masked autoencoders. We leverage Robotic View Transformer (RVT), which uses a multi-view transformer to understand the 3D scene and predict gripper pose actions. We split RVT's multi-view transformer into visual encoder and action decoder, and pretrain its visual encoder using masked autoencoding on large-scale 3D datasets such as Objaverse. We evaluate 3D-MVP on a suite of virtual robot manipulation tasks and demonstrate improved performance over baselines. Our results suggest that 3D-aware pretraining is a promising approach to improve generalization of vision-based robotic manipulation policies. Project site: https://jasonqsy.github.io/3DMVP
title 3D-MVP: 3D Multiview Pretraining for Robotic Manipulation
topic Robotics
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.18158