Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ProSR: Process-Shaped Spatial Reasoning for Reliable Chain-of-Thought in VLMs

About

Reliable spatial reasoning remains a core bottleneck for vision-language models (VLMs). Existing mainstream training paradigms for spatial reasoning largely rely on outcome alignment or process imitation, lacking explicit constraints on the reasoning process, and therefore struggle to ensure genuine visual dependence and stable reasoning trajectories. In this paper, we construct a high-quality CoT dataset covering diverse spatial phenomena and diagnose the model's reasoning process, revealing two typical types of process degradation during reinforcement learning optimization: Spurious Grounding, which bypasses visual evidence, and Tail Instability, where uncertainty abnormally rises in the later stage of reasoning. To address these issues, we propose ProSR, a process-shaping optimization framework for spatial reasoning. Through a Counterfactual Invariance Penalty and a Tail Drift Penalty, ProSR extends the optimization objective from single answer correctness to two process-level dimensions: visual dependence and trajectory stability. Experiments on multiple complex and out-of-distribution spatial reasoning benchmarks show that ProSR improves answer accuracy while generating reasoning trajectories that are more stable and more dependent on visual evidence.

Jiangyang Li, Cong Wan, Changjie Wu, Songlin Dong, Lingjun Zhang, Linzhe Shi, Xu Wang, Zhiheng Ma, Hang Zhang, Mu Xu, Yihong Gong• 2026

Related benchmarks

TaskDatasetResultRank
Spatial ReasoningViewspatial
Accuracy51.4
125
Spatial ReasoningEmbSpatial
Overall Accuracy79.8
103
Spatial ReasoningMindCube (tiny)
Accuracy88.6
65
Spatial Reasoning3DSRBench
Overall Accuracy62.4
60
Spatial ReasoningSPAR-Bench--
45
Showing 5 of 5 rows

Other info

Follow for update