SpatialDreamer: Incentivizing Spatial Reasoning via Active Mental Imagery

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Cao, Meng, Li, Xingyu, Liu, Xue, Reid, Ian, Liang, Xiaodan
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911308472909824
author Cao, Meng
Li, Xingyu
Liu, Xue
Reid, Ian
Liang, Xiaodan
author_facet Cao, Meng
Li, Xingyu
Liu, Xue
Reid, Ian
Liang, Xiaodan
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2512_07733
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SpatialDreamer: Incentivizing Spatial Reasoning via Active Mental Imagery
Cao, Meng
Li, Xingyu
Liu, Xue
Reid, Ian
Liang, Xiaodan
Computer Vision and Pattern Recognition
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.
title SpatialDreamer: Incentivizing Spatial Reasoning via Active Mental Imagery
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2512.07733