English

SpatialDreamer: Incentivizing Spatial Reasoning via Active Mental Imagery

Computer Vision and Pattern Recognition 2025-12-09 v1

Abstract

Despite advancements in Multi-modal Large Language Models (MLLMs) for scene understanding, their performance on complex spatial reasoning tasks requiring mental simulation remains significantly limited. Current methods often rely on passive observation of spatial data, failing to internalize an active mental imagery process. To bridge this gap, we propose SpatialDreamer, a reinforcement learning framework that enables spatial reasoning through a closedloop process of active exploration, visual imagination via a world model, and evidence-grounded reasoning. To address the lack of fine-grained reward supervision in longhorizontal reasoning tasks, we propose Geometric Policy Optimization (GeoPO), which introduces tree-structured sampling and step-level reward estimation with geometric consistency constraints. Extensive experiments demonstrate that SpatialDreamer delivers highly competitive results across multiple challenging benchmarks, signifying a critical advancement in human-like active spatial mental simulation for MLLMs.

Keywords

Cite

@article{arxiv.2512.07733,
  title  = {SpatialDreamer: Incentivizing Spatial Reasoning via Active Mental Imagery},
  author = {Meng Cao and Xingyu Li and Xue Liu and Ian Reid and Xiaodan Liang},
  journal= {arXiv preprint arXiv:2512.07733},
  year   = {2025}
}
R2 v1 2026-07-01T08:15:12.640Z