Coarse-to-Fine 3D Keyframe Transporter

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhu, Xupeng, Klee, David, Wang, Dian, Hu, Boce, Huang, Haojie, Tangri, Arsh, Walters, Robin, Platt, Robert
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917911044554752
author Zhu, Xupeng
Klee, David
Wang, Dian
Hu, Boce
Huang, Haojie
Tangri, Arsh
Walters, Robin
Platt, Robert
author_facet Zhu, Xupeng
Klee, David
Wang, Dian
Hu, Boce
Huang, Haojie
Tangri, Arsh
Walters, Robin
Platt, Robert
contents Recent advances in Keyframe Imitation Learning (IL) have enabled learning-based agents to solve a diverse range of manipulation tasks. However, most approaches ignore the rich symmetries in the problem setting and, as a consequence, are sample-inefficient. This work identifies and utilizes the bi-equivariant symmetry within Keyframe IL to design a policy that generalizes to transformations of both the workspace and the objects grasped by the gripper. We make two main contributions: First, we analyze the bi-equivariance properties of the keyframe action scheme and propose a Keyframe Transporter derived from the Transporter Networks, which evaluates actions using cross-correlation between the features of the grasped object and the features of the scene. Second, we propose a computationally efficient coarse-to-fine SE(3) action evaluation scheme for reasoning the intertwined translation and rotation action. The resulting method outperforms strong Keyframe IL baselines by an average of >10% on a wide range of simulation tasks, and by an average of 55% in 4 physical experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2502_01773
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Coarse-to-Fine 3D Keyframe Transporter
Zhu, Xupeng
Klee, David
Wang, Dian
Hu, Boce
Huang, Haojie
Tangri, Arsh
Walters, Robin
Platt, Robert
Robotics
Computer Vision and Pattern Recognition
Recent advances in Keyframe Imitation Learning (IL) have enabled learning-based agents to solve a diverse range of manipulation tasks. However, most approaches ignore the rich symmetries in the problem setting and, as a consequence, are sample-inefficient. This work identifies and utilizes the bi-equivariant symmetry within Keyframe IL to design a policy that generalizes to transformations of both the workspace and the objects grasped by the gripper. We make two main contributions: First, we analyze the bi-equivariance properties of the keyframe action scheme and propose a Keyframe Transporter derived from the Transporter Networks, which evaluates actions using cross-correlation between the features of the grasped object and the features of the scene. Second, we propose a computationally efficient coarse-to-fine SE(3) action evaluation scheme for reasoning the intertwined translation and rotation action. The resulting method outperforms strong Keyframe IL baselines by an average of >10% on a wide range of simulation tasks, and by an average of 55% in 4 physical experiments.
title Coarse-to-Fine 3D Keyframe Transporter
topic Robotics
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.01773