ArXiv

Learning Action Priors for Cross-embodiment Robot Manipulation

Authors
Dong Jing, Tianqi Zhang, Jiaqi Liu...
Categories
cs.RO, cs.AI, cs.CV
arXiv
https://arxiv.org/abs/2606.26095v1
PDF
https://arxiv.org/pdf/2606.26095v1

Brief

Action-prior pretraining for cross-embodiment robot manipulation introduces a two-stage framework: a flow-matching encoder–decoder learns unconditioned temporal motion priors (Stage 1), then these priors are transferred into Vision-Language-Action training via decoder reuse and latent distillation (Stage 2). The encoder also compresses histories into a single context token. Experiments across 13 simulated and real tasks show faster convergence, higher success rates, and stronger real-world few-shot performance; scaling Stage 1 data further improves generalization.

Why it matters

Proposes a two-stage action-prior pretraining for Vision-Language-Action (VLA) models: Stage 1 uses a lightweight flow-matching encoder–decoder to learn temporal motion structure from unconditioned action trajectories; Stage 2 transfers the prior via decoder reuse and early-stage latent distillation while allowing end-to-end policy refinement. The trained encoder also compresses state-action history into a single temporal context token.

Key details

  • Evaluated on 13 diverse cross-embodiment tasks (simulated and real-world), the method yields faster convergence, higher success rates, and substantially better performance on data-scarce real-world tasks; increasing the amount of Stage 1 action data produces a more generalizable prior that improves downstream VLA performance (paper published 2026-06-24).
Source evidence

Abstract

Most Vision-Language-Action (VLA) models build on a Vision-Language Model (VLM) backbone by attaching an action module and optimizing the full policy jointly. This design inherits strong visual and linguistic priors from the VLM, but leaves the action module to learn physical motion almost from scratch. As a result, the policy lacks an explicit motion prior, forcing early optimization to simultaneously discover temporal action dynamics and cross-modal alignment, a challenge further amplified in cross-embodiment settings. In this work, we propose to pretrain the action module with motion priors before cross-modal VLA alignment. Specifically, we introduce a two-stage training framework that equips the action module with cross-embodiment temporal motion structure before VLA training begins. In Stage~1, a lightweight flow-matching-based encoder-decoder action module efficiently learns temporal motion structure solely from unconditioned action trajectories, without processing visual or language tokens. In Stage~2, this learned prior is transferred to VLA training through decoder reuse and early-stage latent distillation, aligning visual-language features with the action embedding space while still allowing end-to-end policy refinement. In addition, the trained encoder serves as a compact history compressor, summarizing state-action histories into a single temporal context token for history-aware modeling at negligible cost. Extensive experiments across 13 diverse cross-embodiment tasks on both simulated and real-world platforms validate the effectiveness of our approach. Compared with VLA training without action priors, our model achieves faster convergence, higher success rates, and substantially stronger performance on data-scarce real-world tasks. Moreover, scaling up the action data in Stage~1 yields a more generalizable action prior that directly improves downstream VLA performance.