Generative world models, spatial reasoning, and action
Jiatao Gu
3D in the Era of World Models · ECCV 2026 Workshop · September 9
Image generation improved before coherent video generation did.

Build 3D into the generator.
Learn from 2D images; control the camera.

Use an image prior to optimize 3D.
A frozen 2D diffusion model guides a NeRF.

Couple diffusion across views.
NeRF guides jointly refined virtual views.
Use image-generation knowledge to produce a consistent world.
Make spatial structure part of what the generator learns and predicts.
SSDNeRF · ICCV 2023

WVD · RGB–XYZ prediction · CVPR 2025


XYZ


The goal: turn image-level generative capability into
3D-consistent generation for world modeling.
Learn spatial and temporal correlations directly in the generative model.


Expressive generative models + sufficient data and compute
3D structure can be learned implicitly.


Controls specify a candidate motion.

Generative models predict how the world changes under a chosen motion.

The next view depends on what the user does.

If the deliverable is pixels or actions, why prescribe a 3D representation?


A spatial experience can be generated without first delivering a complete 3D scene.



Compelling spatial experiences can emerge through RGB generation.
How can these predictions help an agent decide what to do?
How do these predictions help an agent choose what to do?

Relative positions, distances, and occlusions.
How objects and observations change under an action.
Reachability, clearance, and collision.
Does explicit 3D help us answer these questions?

Past frames, actions, and the current question
Infer spatial relationships from the available evidence
Location, visibility, or a possible next move
Spatial understanding can stay implicit in the model and its context.
As history grows: keep more context, compress it, or retrieve what matters.
Much of our reasoning is in service of acting in the physical world.
How far does the hand need to move?
Do the hand and object actually meet?
Can the motion be executed in the world?
3D gives reasoning an explicit interface to the space where actions have consequences.
Reasoning over time needs memory.
The observation history keeps growing.
The model learns what persists and what matters from data.
More history also requires a way to manage memory.
Keep the same spatial identities.
Update their state as new evidence arrives.
3D provides a structural prior for persistence.
The same world need not be stored again at every timestep.
Stable structure, evolving objects.

Reuse geometry in world coordinates.
Update changing objects over time.
Query the state when the observer returns.
The world state can persist beyond the current context window.
Reuse spatial reasoning across time.

A 3D map carries geometry across observations.
New depth revises the current map.
Check distance and collision for each new plan.
The time changes; the meaning of a spatial query stays the same.
A persistent memory layer for world prediction and action prediction.

New evidence
State · features · uncertainty
Persist · update · reuse
Views and future states
Motion and constraints
Learn over persistent state, then reuse the model across time.
Next: spatial motion · spatial constraints · persistent memory
I’ll focus on PointAction and OmniGuide.
PointAction

Connect predicted 3D motion to robot control.
Video prediction → 3D point motion → robot actions
OmniGuide

Use spatial constraints to guide a pretrained VLA.
VLA + 3D fields → guided actions
A spatial representation the model can predict and reason with.
Given an observation and an instruction, imagine a plan the robot can follow.
WAM
World Action
Model
Predict a visual future
Controls aligned with
the predicted future
WPAM
World Point
Action Model
PointAction
Jointly predict
RGB + XYZ
Explicit robot point motion
Decode point motion
with the robot state
Make the world model’s plan explicit in 3D before translating it into action.
PointAction · from imagined motion to robot control

The action decoder still has to infer…
How far should the gripper move in 3D?
Where does the gripper meet the object?
Which commands realize that motion?
Grounding RGB in control requires robot-specific action supervision.
Make the predicted motion explicit before translating it into controls.


Expose spatial displacement over time.
Select the moving robot surface for control.
Represent motion without arm-specific action channels.
Predict how the robot should move; learn how this robot executes it.
Separate learning what should move from learning a robot’s control space.

Reconstruction and depth estimation provide point supervision without action labels.
A small decoder translates point motion and robot state into that arm’s actions.
The bridge is 3D point motion: learnable from videos, usable for control.

XYZ gives the 3D position associated with a pixel.
A sequence of pointmaps exposes spatial change over time.
SAM 3 segments the predicted RGB with the prompt “robot”.
The mask comes from a separate segmentation pass.
01
DROID + BridgeData V2
02
Stereo / monocular depth
03
Joint future prediction
04
Post-train model + decoder
No paired action labels in pretraining
20–50 per task
$o$: RGB · $u$: XYZ + robot mask · $a$: actions · $s_t$: robot state · $l$: instruction
A shared representation with target-arm adaptation of the predictive model and decoder.


Target-arm data supports RGB–XYZ post-training and decoder training; transfer is not zero-shot.
Simulation decoder: 6 Transformer blocks · hidden width 256 · 4 attention heads
RoboCasa365 sim, avg success (%), 100 rollouts per task.
| Decoder input | ID | OOD-Env |
|---|---|---|
| RGB only | 25.1 | 20.3 |
| RGB + XYZ, robot only | 37.2 | 30.9 |
| XYZ, full scene | 27.1 | 19.4 |
| XYZ, robot + scene | 40.3 | 33.7 |
| Post-hoc DA3 XYZ, robot only | 28.4 | 21.7 |
| Jointly predicted XYZ, robot only | 47.7 | 44.1 |
In this setting, which spatial variables we expose changes the control result.




Generated RGB and corresponding dynamic XYZ pointmaps
DROID + BridgeData V2 · 300 held-out trajectories per dataset
| Method | PSNR ↑ | SSIM ↑ | FVD ↓ | Chamfer L1 ↓ |
|---|---|---|---|---|
| TesserAct | 12.23 | 0.487 | 746 | 0.389 |
| 4DNeX | 13.86 | 0.542 | 818 | 0.370 |
| LVP | 19.61 | 0.816 | 330 | — |
| Wan 2.1 14B | 14.53 | 0.674 | 671 | — |
| Ours | 19.63 | 0.821 | 320 | 0.122 |
RoboCasa365 · success (%) · 100 rollouts per task
| Setting | GR00T N1.7 | $\pi_{0.5}$ | VPP | Cosmos | Ours |
|---|---|---|---|---|---|
| ID (10 seen) | 44.5 | 39.8 | 34.5 | 45.2 | 47.7 |
| OOD-Env | 37.6 | 35.2 | 32.2 | 42.9 | 44.1 |
| OOD-Task | 8.6 | 6.9 | 7.4 | 14.0 | 17.0 |
Spatial grounding helps control. Task generalization remains a separate challenge.
YAM and xArm7 — neither seen during 4D-video pretraining. Target-arm adaptation uses 20–50 expert trajectories per task.






(a) xArm7 — 100 rollouts / task
| % | P&P | St.Cu | St.Cup | Avg. |
|---|---|---|---|---|
| GR00T N1.7 | 30.0 | 7.0 | 7.0 | 14.7 |
| $\pi_{0.5}$ | 42.0 | 12.0 | 14.0 | 22.7 |
| Ours | 67.0 | 28.0 | 34.0 | 43.0 |
(b) YAM — 20 evals / task
| % | St.Cu | PickPen | InsertCup |
|---|---|---|---|
| GR00T N1.5 | 0 | 20 | 15 |
| $\pi_0$ | 0 | 10 | 15 |
| Ours | 20 | 60 | 50 |
xArm7: 43.0% vs. 22.7% average success; both new arms require target-arm adaptation.
POINTACTION
Update model weights.
Train / fine-tune RGB–XYZ prediction
and the point-to-action decoder.
3D is a learned intermediate representation.
OMNIGUIDE
Update action samples.
Keep the pretrained VLA fixed;
steer its sampling with spatial gradients.
3D is an inference-time constraint.
3D can shape what a model learns and how a trained model acts.
A VLA proposes actions; explicit spatial reasoning helps refine them.
VLA = Vision–Language–Action model · e.g., $\pi_{0.5}$ and GR00T N1.6
1 · PROPOSE
Images, instruction and robot state
condition a motion proposal.
2 · CHECK IN 3D
Distance to the target?
Clearance from obstacles?
Alignment with a demonstration?
3 · REFINE
Use spatial gradients to steer
the VLA’s action sampling.
Repeat while denoising · policy weights stay fixed
3D makes the constraints explicit and directly computable.
Yunzhou Song, Long Le, Yong-Hyun Park, Jie Wang, Junyao Shi,
Lingjie Liu, Jiatao Gu, Eric Eaton, Dinesh Jayaraman, Kostas Daniilidis

$\mathcal L_y(X)$ scores a candidate Cartesian trajectory.
Attract toward a goal.
Repel away from obstacles.
The field lives in Cartesian space and guides generation in action space.
One action chunk describes several future robot steps.
$A^0\sim\mathcal N(0,I)$ becomes clean $A^1$.
OmniGuide convention: $\tau=0$ noise, $\tau=1$ clean · $o$ collects the current policy context
| Source | Energy, up to constants | Geometric effect |
|---|---|---|
| Scene geometry | $\mathcal L_C(x)=-\log\mathrm{SDF}(x)$ Within $0<\mathrm{SDF}(x)\le d$ | Repel robot probes from nearby obstacles. |
| Semantic target | $\mathcal L_S(x)=\frac{\|x-x^*\|^2}{2\sigma_S^2}$ | VLM pointing + depth gives a metric 3D goal $x^*$. |
| Human motion | $\mathcal L_H(X)=\frac{1}{2\sigma_H^2}\sum_{(i,k)\in M}\|x_i-h_k\|^2$ | Attract matched robot poses to a reference wrist trajectory. |
All three score a clean Cartesian trajectory $X$.
Collision point clouds exclude the robot body and task-relevant objects. Human matching progresses forward in the demonstration.

The gradient changes the action sample; policy weights stay fixed.

Action sample → clean estimate → Cartesian poses → energy.
Pass the spatial gradient through the robot mapping and the policy’s input.
Differentiate the action sample. The pretrained weights stay fixed.
Before full denoising, compare a few short previews of possible plans.
Next: compare selection, refinement, and both.
Compare seed selection, denoising refinement, and their combination.
No spatial guidance
Initialization guidance
Denoising guidance
Select, then refine

Refinement gives the larger gains; combining both performs best in this comparison.
RoboCasa + GR00T N1.6-3B · 50 trials per setting · ground-truth depth and object poses


Combined guidance gives the best task success here. Collision-only guidance retains slightly higher safety.
One 3D trajectory, with both spatial objectives guiding its generation.

| Aggregate metric | Base | Guided |
|---|---|---|
| Success | 24.2% | 92.4% |
| Collision avoidance | 7.0% | 93.5% |
Published cross-setting aggregates, not the success rate of this one video.
Local spatial objectives still depend on perception, calibration, and a useful policy prior.
FUTURE DIRECTIONS
An agent that anticipates change, compares action consequences, and learns from new evidence.
World prediction → 3D motion → control
3D constraints → gradients → actions
Can spatial reasoning become a persistent part of the model?
An agent’s ability to predict how the world changes and what its actions will cause.
Preserve what matters for predicting change.
Anticipate consequences before acting.
Actions condition predictions; predictions guide actions.
FUTURE DIRECTIONS
Pixels, latents, objects, geometry, or a hybrid?
What should the state preserve?
Autoregression, diffusion, flows, or structured dynamics?
How do errors accumulate over time?
Planning, policy learning, or inference-time guidance?
Which predictions matter for decisions?
Where can 3D make these choices more effective or efficient?
FUTURE DIRECTIONS · 1. STATE / MEMORY

Encode observations into compact, spatially meaningful latents.
Preserve object identity, geometry, relations, and motion.
Then learn what should persist, what should change, and what remains uncertain.
The geometry visualizes the representation. The state itself can stay latent.
FUTURE DIRECTIONS · 1. STATE / MEMORY
Keep stable structure, represent uncertainty, and revise the state when new evidence arrives.
Stable spatial structure
What is no longer observed
What new evidence changes
Research questions: memory growth · update cost · consistency over long interactions
FUTURE DIRECTIONS · 2. LEARNING TO PREDICT
$z_t$
Geometry · objects · uncertainty
$p_\theta(z_{t+1}\mid z_t,a_t)$
Action-dependent possible futures
$z_{t+1}$
For planning and control
Here $z_t$ summarizes the available history; this is a proposed modeling interface.
Represent several plausible outcomes.
Reuse the predictor as state evolves.
Evaluate candidate actions repeatedly.
Next: what the STARFlow family offers as generative foundations.
FUTURE DIRECTIONS · GENERATIVE FOUNDATIONS
A foundation for learning distributions over compact representations.

FUTURE DIRECTIONS · GENERATIVE FOUNDATIONS

ONGOING
Use an action-conditioned rollout model to compare possible futures.

Video: open-loop prediction under dataset actions. The loop above describes the planning direction.
FUTURE DIRECTION · CONNECTING MEMORY AND PREDICTION
Read a persistent spatial state, predict action consequences, then update it with new evidence.
Align spatial state with model latents.
Predict object and relation updates.
Track uncertainty and revise errors.
An architecture to explore: learn the read / update interface and test long-term consistency, control, and cost.
3D as internal reasoning for world models and action.
PointAction · OmniGuide (CoRL 2026) · STARFlow / STARFlow-V

Predict the velocity $\epsilon-\tilde z$.
Randomize the observed prefix length $m$ and its noise level $\tau'$.
With 50% probability, set $\tau'=0$ so the history is clean.
PointAction convention: $\tau=0$ clean, $\tau=1$ noise · frozen VAE and text encoder, LoRA adaptation