ImageWAM: Do World Action Models Really Need Video Generation, or Just Image Editing?
TLDR
ImageWAM repurposes image editing models for robot action prediction, outperforming video-based world action models with lower cost and latency.
Reasoning
The paper presents a novel approach that replaces video generation with image editing for world action models, achieving superior performance and efficiency in both simulator and real-world experiments. However, it lacks explicit discussion of limitations and potential dependencies on pretrained models.
Read-first score
Read-first score 56.8, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 36.
Field roles
Rank sensitivity
Stability: volatile; rank range: 423.
Keyword Scores
Deep Analysis
Innovations
- Proposes ImageWAM, a framework that repurposes pretrained image editing models for robot action prediction instead of video generation.
- Uses KV caches from image-editing denoising as a compact world-action context, without decoding the target frame at inference.
- Demonstrates that image editing provides a better-matched prior for world-action modeling: target-frame transformation, action-relevant visual differences, and localized visual changes through edit pretraining.
Methodology
ImageWAM repurposes a pretrained image editing model to produce KV caches during denoising, which condition a flow-matching action expert. It does not decode the target frame at inference time. The model is evaluated on simulator and real-world experiments, compared to standard VLA baselines and competitive WAMs.
Key Results
ImageWAM outperforms standard VLA baselines and matching competitive WAMs without additional policy pretraining across simulator and real-world experiments. It reduces FLOPs to 1/6 and latency to 1/4 of video-based WAMs.