One-Policy-Fits-All: Geometry-Aware Action Latents for Cross-Embodiment Manipulation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Mu, Juncheng, Yang, Sizhe, Bae, Hojin, Jia, Feiyu, Ben, Qingwei, Li, Boyi, Xu, Huazhe, Pang, Jiangmiao
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866908887779639296
author Mu, Juncheng
Yang, Sizhe
Bae, Hojin
Jia, Feiyu
Ben, Qingwei
Li, Boyi
Xu, Huazhe
Pang, Jiangmiao
author_facet Mu, Juncheng
Yang, Sizhe
Bae, Hojin
Jia, Feiyu
Ben, Qingwei
Li, Boyi
Xu, Huazhe
Pang, Jiangmiao
contents Cross-embodiment manipulation is crucial for enhancing the scalability of robot manipulation and reducing the high cost of data collection. However, the significant differences between embodiments, such as variations in action spaces and structural disparities, pose challenges for joint training across multiple sources of data. To address this, we propose One-Policy-Fits-All (OPFA), a framework that enables learning a single, versatile policy across multiple embodiments. We first learn a Geometry-Aware Latent Representation (GaLR), which leverages 3D convolution networks and transformers to build a shared latent action space across different embodiments. Then we design a unified latent retargeting decoder that extracts embodiment-specific actions from the latent representations, without any embodiment-specific decoder tuning. OPFA enables end-to-end co-training of data from diverse embodiments, including various grippers and dexterous hands with arbitrary degrees of freedom, significantly improving data efficiency and reducing the cost of skill transfer. We conduct extensive experiments across 11 different end-effectors. The results demonstrate that OPFA significantly improves policy performance in diverse settings by leveraging heterogeneous embodiment data. For instance, cross-embodiment co-training can improve success rates by more than 50% compared to single-source training. Moreover, by adding only a few demonstrations from a new embodiment (e.g., eight), OPFA can achieve performance comparable to that of a well-trained model with 72 demonstrations.
format Preprint
id arxiv_https___arxiv_org_abs_2603_14522
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle One-Policy-Fits-All: Geometry-Aware Action Latents for Cross-Embodiment Manipulation
Mu, Juncheng
Yang, Sizhe
Bae, Hojin
Jia, Feiyu
Ben, Qingwei
Li, Boyi
Xu, Huazhe
Pang, Jiangmiao
Robotics
Cross-embodiment manipulation is crucial for enhancing the scalability of robot manipulation and reducing the high cost of data collection. However, the significant differences between embodiments, such as variations in action spaces and structural disparities, pose challenges for joint training across multiple sources of data. To address this, we propose One-Policy-Fits-All (OPFA), a framework that enables learning a single, versatile policy across multiple embodiments. We first learn a Geometry-Aware Latent Representation (GaLR), which leverages 3D convolution networks and transformers to build a shared latent action space across different embodiments. Then we design a unified latent retargeting decoder that extracts embodiment-specific actions from the latent representations, without any embodiment-specific decoder tuning. OPFA enables end-to-end co-training of data from diverse embodiments, including various grippers and dexterous hands with arbitrary degrees of freedom, significantly improving data efficiency and reducing the cost of skill transfer. We conduct extensive experiments across 11 different end-effectors. The results demonstrate that OPFA significantly improves policy performance in diverse settings by leveraging heterogeneous embodiment data. For instance, cross-embodiment co-training can improve success rates by more than 50% compared to single-source training. Moreover, by adding only a few demonstrations from a new embodiment (e.g., eight), OPFA can achieve performance comparable to that of a well-trained model with 72 demonstrations.
title One-Policy-Fits-All: Geometry-Aware Action Latents for Cross-Embodiment Manipulation
topic Robotics
url https://arxiv.org/abs/2603.14522