Scaling Cross-Embodiment World Models for Dexterous Manipulation
Zihao He, Bo Ai, Tongzhou Mu, Yulin Liu, Weikang Wan, Jiawei Fu, Yilun Du, Henrik I. Christensen, Hao Su
TL;DR
This paper tackles the challenge of cross-embodiment dexterous manipulation by positing that environment dynamics are embodiment-invariant and can be captured with a unified world model. It adopts a particle-based state and action representation, where hands and objects are sets of 3D particles and actions are particle displacements, and trains a graph-based dynamics model to predict future states. The model is learned from diverse simulated robot hands and real human hands and is deployed via model-based planning that maps joint actions to the particle space through forward kinematics. Key findings show that increasing the number of training embodiments improves generalization to unseen morphologies, co-training simulated and real data yields benefits beyond either alone, and the learned models can control hands with varied degrees of freedom, including deformable-object manipulation; collectively, the work presents world models as a promising interface for cross-embodiment dexterous manipulation.
Abstract
Cross-embodiment learning seeks to build generalist robots that operate across diverse morphologies, but differences in action spaces and kinematics hinder data sharing and policy transfer. This raises a central question: Is there any invariance that allows actions to transfer across embodiments? We conjecture that environment dynamics are embodiment-invariant, and that world models capturing these dynamics can provide a unified interface across embodiments. To learn such a unified world model, the crucial step is to design state and action representations that abstract away embodiment-specific details while preserving control relevance. To this end, we represent different embodiments (e.g., human hands and robot hands) as sets of 3D particles and define actions as particle displacements, creating a shared representation for heterogeneous data and control problems. A graph-based world model is then trained on exploration data from diverse simulated robot hands and real human hands, and integrated with model-based planning for deployment on novel hardware. Experiments on rigid and deformable manipulation tasks reveal three findings: (i) scaling to more training embodiments improves generalization to unseen ones, (ii) co-training on both simulated and real data outperforms training on either alone, and (iii) the learned models enable effective control on robots with varied degrees of freedom. These results establish world models as a promising interface for cross-embodiment dexterous manipulation.
