what_they_did
The authors first train Astra-WM, a Bagel-based world simulator conditioned on context images and language-describing camera motions, with view consistency tuning to improve pose and content consistency across generated views. They then train Astra-VL, a VLM policy, using a two-phase RL curriculum: Phase I stabilizes tool-use exploration by rewarding diverse invocations; Phase II refines the policy to invoke the simulator only when imagined observations improve over direct answering. The system is evaluated on two spatial reasoning benchmarks: MMSI-Bench (multi-view scene understanding) and MindCube (mental rotation and perspective taking).
key_findings
- Astra-WM improves simulator-augmented Gemini-3-Flash on MMSI-Bench from 45.1 to 49.5 (a 4.4-point gain).
- Astra-VL improves its Qwen3-VL backbone from 29.8 to 38.8 on MMSI-Bench (a 9.0-point gain) and from 36.8 to 42.7 on MindCube (a 5.9-point gain).
- Ablation studies confirm that both the world simulator and the RL-trained policy are necessary; removing either component reduces performance significantly.
- The RL curriculum is critical: the two-phase training helps the model learn to invoke the simulator selectively, avoiding unnecessary queries.
why_it_matters
This work demonstrates that visual agents can transcend passive observation by actively generating and reasoning over imagined visual evidence from alternative viewpoints. The framework offers a principled way to integrate generative world models into VLM reasoning, enabling cross-view consistency and spatial reasoning about unobserved layouts—capabilities that are essential for embodied AI, robotics, and navigation.
caveats
The method is evaluated only on synthetic benchmarks (MMSI-Bench, MindCube), which may not capture the complexity of real-world spatial reasoning. The world simulator requires paired training data (context images + target novel views) that may be expensive to acquire in new environments. The RL curriculum introduces additional training complexity and hyperparameter sensitivity. Generalization to open-ended tasks with many possible viewpoints is not yet tested.