VISUAL REPRESENTATIONS FOR ROBOT LEARNING

What Visual Representations
Do World Action Models Need?

What to represent. How to integrate. When to combine.

Kai Liu1,2,*Jiangran Lyu2,3,*Tingyu Deng3Yixin Zheng1,2Jingtian Wan2,4Wenbo Cui1,2Mi Yan2,3Yizhou Zhou5Haoran Li1Dongbin Zhao1,†Zhizheng Zhang2,†He Wang2,3,†
1 Institute of Automation, Chinese Academy of Sciences2 Galbot3 Peking University4 University of Science and Technology of China5 Individual Researcher

* Equal contribution · † Co-corresponding authors

arXiv:2610.13024 · October 2026

Motivation

The gap between
reconstruction and manipulation.

World action models inherit a VAE latent space optimized for visual reconstruction. Yet reconstructing what a scene looks like does not necessarily capture the information needed to act within it. So a natural question is: what visual representations do WAMs need?

We study this question within a common WAM framework, comparing representation sources, feature integration mechanisms, and combinations of sources. The result is a practical Representation Recipe — R²-WAM — validated across WAM formulations, simulation benchmarks, and physical robot tasks.

13pretrained visual encoders
5feature integration mechanisms
3research questions
Paper overview: visual sources, five integration mechanisms, progressive feature composition, and validation across robot settingsView full figure
The study connects representation choice, integration, and composition under a shared training protocol.

THREE QUESTIONS, THREE FINDINGS

A closer look at
what makes a difference.

Explore the evidence behind the recipe.
Success rates (%) and gains in percentage points (pp).

RQ1 · WHICH REPRESENTATIONS HELP?

Start with a strong
general-purpose representation.

Across the evaluated encoders, general-purpose representations achieve a higher average LP success rate than specialized representations: 82.2% versus 80.0%. A general-purpose encoder has the highest observed mean in each of the seven perturbation categories.

+9.3pp

DINOv3 over the VAE-only baseline
84.0% vs. 74.7% overall success

Single-source comparison

Gain over VAE-only (pp) · channel-wise concatenation · means over three evaluation seeds

VAE-only baseline: 74.7%

Bars: gain over baseline · SR: success rate

FROM FINDINGS TO A RECIPE

Introducing R²-WAM.

A compact combination of visual representations,
integrated through channel-wise concatenation.

VISUAL SELF-SUPERVISIONDINOv3
MULTI-OBJECTIVE LEARNINGUniTok
VIDEO PREDICTIVE LEARNINGV-JEPA 2.1
CHANNEL-WISE CONCATENATIONR2-WAM

The three-source recipe reaches 87.3% on LIBERO-Plus. Adding HRP reaches 87.7%; we retain three sources to limit auxiliary encoders and feature-prediction targets.

VALIDATION BEYOND THE MAIN STUDY

The same recipe.
Different settings.

Improvements over corresponding VAE-only baselines
across simulation benchmarks and WAM formulations.

ROBOCASA GR1 TABLETOP

84.7%

+5.9 percentage points

VAE-only: 78.8%
24 tasks · 50 trials per task

ROBOTWIN 2.0

92.8%

+2.2 percentage points

VAE-only: 90.6%
50 tasks · clean and randomized settings

OTHER WAM FORMULATIONS

Direct action prediction

70.8 79.4%

Video-then-action (IDM)

73.6 87.7%

LIBERO-Plus success · VAE-only → + recipe

ON A PHYSICAL ROBOT

From visual features
to everyday actions.

Four manipulation tasks on Galbot G1.
20 evaluation trials per task, per model.

Demonstrations from the paper’s supplementary material 01:34

Real-robot evaluation
TaskMetricVAE-onlyR²-WAM
Air-fryer food preparationComplete-task success50.0%60.0%
Instructed pick-and-placeComplete-task success70.0%80.0%
Tool sweepingMean debris collection rate48.3%66.7%
Cup pushingComplete-task success65.0%75.0%

Sweeping measures the proportion of six debris pieces collected per trial; the other tasks measure complete-task success.

REFERENCE THIS WORK

A recipe for
further research.

BIBTEX
@misc{liu2026r2wam,
  title = {What Visual Representations Do World Action Models Need?},
  author = {Kai Liu and Jiangran Lyu and Tingyu Deng and Yixin Zheng
    and Jingtian Wan and Wenbo Cui and Mi Yan and Yizhou Zhou
    and Haoran Li and Dongbin Zhao and Zhizheng Zhang and He Wang},
  year = {2026},
  eprint = {2610.13024},
  archivePrefix = {arXiv},
  primaryClass = {cs.CV},
  url = {https://arxiv.org/abs/2610.13024}
}

Paper figure

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