Skeleton-Grounded Latent Action Model
for Robot World Models

1 HIT (Shenzhen)2 Show Lab, NUS3 Ping An Tech

Abstract

Robot world models simulate how environments evolve in response to robot actions. Latent action models enable learning dynamics from videos by encoding visual transitions as proxy actions. However, reconstruction-driven latent actions mix robot motion with object responses and other scene changes, complicating their correspondence with robot controls. We propose SkeLAM, a skeleton-guided latent action model for controllable robot world modeling. SkeLAM decomposes latent actions into a kinematic component grounded by robot skeleton supervision and a residual component retaining complementary visual dynamics. Cross-modal swap reconstruction encourages RGB-derived and skeleton-derived kinematic representations to be interchangeable during video reconstruction. The world model incorporates both components through separate conditioning pathways, while branch dropout reduces its dependence on future-derived residual information. A lightweight adapter maps robot actions to the kinematic space, and subsequent world-model fine-tuning adapts generation to these predicted conditions. At inference, the model generates future observations from the current observation and robot actions. On DROID, SkeLAM reduces end-effector ADE by 19.5% and LPIPS by 11.1% compared with DreamDojo. Motion probing supports kinematic specialization, while qualitative results demonstrate zero-shot human-to-robot motion transfer in generated rollouts. Policy evaluation shows agreement between predicted success rates and those measured in real robot executions.

Method

SkeLAM overview. Cross-modal swaps align skeleton and RGB motion representations. Separate latent pathways condition the world model; an action adapter makes the learned motion representation usable with robot actions.

Skeleton-grounded latent actions. SkeLAM encodes RGB transitions into a 32D kinematic latent and a 32D residual latent. Paired robot skeletons supervise the kinematic component; cross-modal swap reconstruction encourages RGB and skeleton motion representations to be interchangeable.

World-model conditioning. The two components use separate projection pathways in Cosmos-Predict2.5-2B. Branch dropout reduces reliance on residual information derived from future observations.

Action adaptation. A lightweight adapter maps robot actions to the kinematic space. At inference, the model generates future observations from the current RGB image and robot actions, with the residual set to zero.

Experimental Results

Trained on only ~10 hours of robot data with a pretrained world-model backbone, SkeLAM ranks first on seven of eight reported DROID metrics among action-conditioned models. Ctrl-World and OSCAR achieve higher direction F1.

DROID

400 clips · 1 observed + 12 future frames · 480 × 640
DROID · selected visual and motion metrics
MethodControlPSNR ↑dBSSIM ↑LPIPS ↓ADE ↓pixelsDirection F1 ↑
TesserActText reference15.080.74000.256183.760.2950
GE-BaseText reference12.510.45300.586455.300.3006
IRASimAction embedding11.550.41370.599367.110.2616
Ctrl-WorldAction embedding16.340.59100.402942.960.5888
Cosmos-Predict2.5Action embedding11.510.42090.601860.760.2375
Cosmos3Action embedding16.980.76300.218667.230.3278
EnerVerse-ACAction map14.130.51570.413959.630.2433
GE-SimAction map12.530.45500.495055.570.2799
OSCARSkeleton12.570.44710.478843.740.5810
DreamDojo†Robot action18.910.80570.147351.110.4989
CD-LAM†Action → latent18.340.79100.158653.900.4903
SkeLAM (ours)Action → kinematic19.350.83400.131041.150.5465

Compared with DreamDojo: ADE falls from 51.11 to 41.15 pixels; LPIPS falls from 0.1473 to 0.1310. Lower values indicate closer agreement with the reference.

↑ Higher is better · ↓ Lower is better. Text-conditioned models are visual references. † Large-scale interaction pretraining. Selected metrics are shown here; see the paper and appendix ↗ for all metrics, coverage, and protocols.

Qualitative Results

Ground-truth and generated observations at selected time steps. The overlaid trajectories indicate end-effector motion; the images also show changes in gripper geometry and scene appearance.

Action-conditioned video generation. Four scenes comparing ground truth and SkeLAM with DreamDojo, Cosmos-Predict2.5, OSCAR, and Ctrl-World.

Frame comparisons

Main comparison. SkeLAM more closely follows the curved reference trajectory while retaining the tabletop layout. Local gripper errors remain. Markers and curves visualize end-effector positions and motion.
Additional tabletop scene. Ground-truth and generated frames compare end-effector motion, object geometry, and scene appearance across time.
Additional kitchen scene. The sequence shows the initial observation and three later time steps. Final-frame overlays make motion differences easier to inspect.

Click any figure to view it at full resolution.

Zero-Shot Human-to-Robot Motion Transfer

After training on ~10 hours of robot data, SkeLAM transfers human hand motions to generated robot rollouts zero-shot, with no additional fine-tuning. With all parameters frozen, the RGB encoder extracts latent actions from human demonstrations in UniSkill; these latents condition a robot rollout from its own initial observation.

Human-to-robot motion transfer. Five motion prompts, with human demonstrations on the left, SkeLAM in the middle, and DreamDojo on the right.

Selected frames

Two zero-shot examples. A towel-pushing prompt elicits a downward reach; a tissue-pulling prompt elicits upward motion. Markers indicate end-effector positions.

These examples provide qualitative evidence of motion correspondence across embodiments. They do not establish task completion or broad transfer performance.

Open-Loop Policy Evaluation

We evaluate seven policies across 15 RoboArena sessions (105 episodes). Both world models receive the same initial observations and recorded actions. Human annotators label success in the generated rollouts for comparison with real executions.

SkeLAM achieves 63.02% balanced accuracy and 33.33% success recall. Policy-level Spearman correlation is 0.5347, and success-rate MAE decreases from 7.62 to 4.76 percentage points compared with Cosmos-Predict2.5.

The real episodes contain 9 successes and 96 failures. These results concern recorded-action replay; closed-loop policy evaluation remains future work.

Policy-level agreement. Each point represents a policy; the dashed diagonal indicates perfect agreement with real success rates.

BibTeX

@misc{lv2026skelam,
  title  = {Skeleton-Grounded Latent Action Model for Robot World Models},
  author = {Qi Lv and Jiayi Lin and Yuxin Jiang and Jianzong Wang
            and Xiang Deng and {Mike Shou Zheng}},
  year   = {2026}
}

Research figure

Scroll to inspect the full-resolution figure. Press Esc to close.