Lift3D Foundation Policy: Lifting 2D Large-Scale Pretrained Models for Robust 3D Robotic Manipulation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Jia, Yueru, Liu, Jiaming, Chen, Sixiang, Gu, Chenyang, Wang, Zhilue, Luo, Longzan, Lee, Lily, Wang, Pengwei, Wang, Zhongyuan, Zhang, Renrui, Zhang, Shanghang
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916522996269056
author Jia, Yueru
Liu, Jiaming
Chen, Sixiang
Gu, Chenyang
Wang, Zhilue
Luo, Longzan
Lee, Lily
Wang, Pengwei
Wang, Zhongyuan
Zhang, Renrui
Zhang, Shanghang
author_facet Jia, Yueru
Liu, Jiaming
Chen, Sixiang
Gu, Chenyang
Wang, Zhilue
Luo, Longzan
Lee, Lily
Wang, Pengwei
Wang, Zhongyuan
Zhang, Renrui
Zhang, Shanghang
contents 3D geometric information is essential for manipulation tasks, as robots need to perceive the 3D environment, reason about spatial relationships, and interact with intricate spatial configurations. Recent research has increasingly focused on the explicit extraction of 3D features, while still facing challenges such as the lack of large-scale robotic 3D data and the potential loss of spatial geometry. To address these limitations, we propose the Lift3D framework, which progressively enhances 2D foundation models with implicit and explicit 3D robotic representations to construct a robust 3D manipulation policy. Specifically, we first design a task-aware masked autoencoder that masks task-relevant affordance patches and reconstructs depth information, enhancing the 2D foundation model's implicit 3D robotic representation. After self-supervised fine-tuning, we introduce a 2D model-lifting strategy that establishes a positional mapping between the input 3D points and the positional embeddings of the 2D model. Based on the mapping, Lift3D utilizes the 2D foundation model to directly encode point cloud data, leveraging large-scale pretrained knowledge to construct explicit 3D robotic representations while minimizing spatial information loss. In experiments, Lift3D consistently outperforms previous state-of-the-art methods across several simulation benchmarks and real-world scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2411_18623
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Lift3D Foundation Policy: Lifting 2D Large-Scale Pretrained Models for Robust 3D Robotic Manipulation
Jia, Yueru
Liu, Jiaming
Chen, Sixiang
Gu, Chenyang
Wang, Zhilue
Luo, Longzan
Lee, Lily
Wang, Pengwei
Wang, Zhongyuan
Zhang, Renrui
Zhang, Shanghang
Computer Vision and Pattern Recognition
3D geometric information is essential for manipulation tasks, as robots need to perceive the 3D environment, reason about spatial relationships, and interact with intricate spatial configurations. Recent research has increasingly focused on the explicit extraction of 3D features, while still facing challenges such as the lack of large-scale robotic 3D data and the potential loss of spatial geometry. To address these limitations, we propose the Lift3D framework, which progressively enhances 2D foundation models with implicit and explicit 3D robotic representations to construct a robust 3D manipulation policy. Specifically, we first design a task-aware masked autoencoder that masks task-relevant affordance patches and reconstructs depth information, enhancing the 2D foundation model's implicit 3D robotic representation. After self-supervised fine-tuning, we introduce a 2D model-lifting strategy that establishes a positional mapping between the input 3D points and the positional embeddings of the 2D model. Based on the mapping, Lift3D utilizes the 2D foundation model to directly encode point cloud data, leveraging large-scale pretrained knowledge to construct explicit 3D robotic representations while minimizing spatial information loss. In experiments, Lift3D consistently outperforms previous state-of-the-art methods across several simulation benchmarks and real-world scenarios.
title Lift3D Foundation Policy: Lifting 2D Large-Scale Pretrained Models for Robust 3D Robotic Manipulation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.18623