ResWAM-LIBERO (Official LIBERO / LIBERO-Plus Checkpoint)

Official ResWAM multi-task policy checkpoint (OPTML-Group/ResWAM-LIBERO) trained on the standard LIBERO demonstrations (libero_spatial, libero_object, libero_goal, libero_10) without external robot pretraining, achieving 81.1% zero-shot average success rate across all 10,030 perturbed episodes of LIBERO-Plus (Table 1 of the paper).

• Paper: ResWAM: Learning What Changes, Not What Stays, for World-Action Models
• Authors: Chongyu Fan, Gaowen Liu, Ramana Rao Kompella, Jason Corso, Sijia Liu
• Affiliations: Michigan State University, Cisco, University of Michigan, IBM Research
• Project Page: https://chongyu-fan.netlify.app/posts/reswam/
• Code: https://github.com/OPTML-Group/ResWAM
• HF Collection: https://huggingface.co/collections/OPTML-Group/reswam-6af8906dc9b6f21539b30934

Abstract

World-action models (WAMs) augment vision-language-action policies (VLAs) with future prediction so that the policy can anticipate how its actions will affect the physical world. Existing WAMs supervise complete future observations. However, observations typically change little over an action chunk, so the future loss is dominated by copying static background content rather than learning task-relevant robot and object motion, limiting policy learning. Moreover, existing WAMs rely on large video generation models or separate world models, incurring heavy training and inference compute that hinders closed-loop deployment. To improve policy performance, we propose visual residual learning, supervising only the feature change from the current frame to the future frame in a frozen DINOv3 space. Compared with complete future features, this residual learning target yields a higher feature rank with richer and more diverse representations of scene changes, and spatially concentrates on the robot arm and manipulated objects rather than the background. To further improve computational efficiency, we present ResWAM, a residual world-action model with 2.3B parameters in total, including a lightweight 0.5B dual-branch DiT that jointly trains an action branch and a residual branch via conditional flow matching. Without robot pretraining (i.e., pretraining on external robot action trajectories beyond the benchmark demonstrations), ResWAM achieves top average success rates on LIBERO-Plus (81.1%), RoboTwin 2.0 (93.8%), and RoboDojo (9.40% success, 13.41 score) at the lowest training (2.7–4.0 TFLOPs) and inference compute (1.0–1.4 TFLOPs), and leads real-robot Open, Put, and Stack (80.0% average, +10.0% over π_0.5) at 92.7 ms latency.

Checkpoint Summary

Field	Value
Repository	OPTML-Group/ResWAM-LIBERO
Checkpoint file	model.pt
Parameters	2.3B (RynnBrain-1.1-2B 16-layer text backbone + 0.5B dual-branch DiT + frozen DINOv3-B/16)
Action horizon ($H$)	8 (7-D delta EEF + binary gripper, replan every 8 steps)
Compute & Latency	2.7 TFLOPs train / 1.0 TFLOPs infer / 92.2 ms per action chunk (10 Euler steps, RTX A6000)

Zero-Shot LIBERO-Plus Results (10,030 Episodes)

Camera	Robot	Language	Light	Background	Noise	Layout	Avg.
52.3%	71.0%	89.9%	97.7%	98.3%	85.6%	83.2%	81.1%

Files in this Repository

• model.pt: Full model weights (bf16 / fp32).
• codec_statistics.json: Unified 7-D delta-EEF action and 11-D proprioceptive state (xyz + 6D rotation + 2D finger qpos) normalization statistics (ZScoreCodec.from_file("codec_statistics.json")).
• resolved_config.json: Complete resolved model, data, and evaluation configuration.

Download and Evaluation Usage

hf download OPTML-Group/ResWAM-LIBERO --local-dir ./checkpoints/ResWAM-LIBERO

python scripts/eval/eval_libero_plus.py run \
    --checkpoint ./checkpoints/ResWAM-LIBERO/model.pt \
    --codec-statistics ./checkpoints/ResWAM-LIBERO/codec_statistics.json \
    --libero-plus-root ./third_party/LIBERO-plus \
    --output-dir ./outputs/libero/eval_libero_plus \
    --gpus 0,1,2,3

Citation

@article{fan2026reswam,
  title={ResWAM: Learning What Changes, Not What Stays, for World-Action Models},
  author={Fan, Chongyu and Liu, Gaowen and Kompella, Ramana Rao and Corso, Jason and Liu, Sijia},
  journal={arXiv preprint},
  year={2026}
}