Abstract
Shifting GUI world modeling from pixel reconstruction to latent action-conditioned predictions improves efficiency and avoids identity shortcuts via contrastive learning.
Pixel-space world models have recently emerged as a promising approach for GUI agents, predicting future screens prior to execution and subsequently re-perceiving them for action selection. However, this two-stage paradigm faces two key challenges in GUI environments: (L1) the high computational cost of pixel reconstruction and (L2) the need to reconstruct large and diverse appearance-level details beyond what is required for action selection. In practice, this screen-generation objective can induce an identity shortcut problem, where future-state prediction largely preserves the current state rather than capturing action-induced changes. In this work, we reformulate GUI world modeling by shifting the prediction target from full-screen reconstruction to action-induced future information in latent space. Concretely, we leverage a large multimodal model as the latent world model, whose shared representations support two dedicated heads for both action selection and predictive transition modeling. The action head directly translates these latent representations into action selection, while the transition head captures action-conditioned future information. To mitigate the identity shortcut, we further introduce an action-aware contrastive objective that encourages the transition representations to capture discriminative action-induced changes. Extensive experiments across offline and online GUI settings show that our method achieves up to 31.9% and 40.6% relative gains in success rate over pixel-space world models, respectively, while substantially reducing inference latency by up to 98.9%. Our code is available at https://github.com/sjyu001/latent2world.
Community
TL;DR: Do GUI agents really need to reconstruct future screenshots to make better decisions?
We introduce a Latent World Action Model that jointly learns action selection and action-conditioned future prediction directly in latent space, without expensive pixel-level reconstruction. Through action-aware contrastive learning, our model captures meaningful GUI transitions and improves task success by up to 31.9% (offline) and 40.6% (online) over pixel-space world models, while reducing world-model inference latency by up to 98.9%.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- DA-WAM: Decision-Aligned Future Latents for Driving World Models (2026)
- V-JEPA Policy: Building Effective World-Action Models on Predictive Visual Latents (2026)
- From World Models to World Action Models: Rethinking Next-State Prediction (2026)
- Reconstructing Is Not Acting: Action-Centric Latent Dynamics Modeling (2026)
- WLA$^3$: World Latent Action Modeling for Semantics, Dynamics, and Kinematics (2026)
- SLIP-VLA: Single-Step Latent Imagination for Policy Learning in Vision-Language-Action Models (2026)
- Drive-HWM: Hierarchical World Models for Dynamic-Latent Guided Autonomous Driving (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2610.14228 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper