让视觉模型真正用上几何信息做空间推理
Make Geometry Matter for Spatial Reasoning
- 通过屏蔽2D视觉特征迫使模型依赖几何令牌
- 在关键区域自适应增强几何信息的贡献度
- 适用于需要精准空间理解的图像与视频任务
尽管大规模训练使视觉语言模型在图像和视频理解上表现强劲,但在静态场景和动态视频中的空间推理能力仍受限。现有方法尝试通过引入预训练3D基础模型的几何令牌来改善,但简单融合后微调常导致几何线索被忽视,模型仍过度依赖2D视觉特征。本文提出GeoSR框架,通过两个核心组件:(1) 几何解禁掩码,在训练中战略性屏蔽部分2D视觉令牌,削弱非几何捷径,强制模型依赖几何令牌进行空间推理;(2) 几何引导融合,一种门控路由机制,自适应放大在几何证据关键区域的几何令牌贡献。实验表明,GeoSR在静态与动态空间推理基准上持续优于先前方法,显著提升几何信息利用效率,达成新最佳性能。
原文摘要 · Abstract (English)
Empowered by large-scale training, vision-language models (VLMs) achieve strong image and video understanding, yet their ability to perform spatial reasoning in both static scenes and dynamic videos remains limited. Recent advances try to handle this limitation by injecting geometry tokens from pretrained 3D foundation models into VLMs. Nevertheless, we observe that naive token fusion followed by standard fine-tuning in this line of work often leaves such geometric cues underutilized for spatial reasoning, as VLMs tend to rely heavily on 2D visual cues. In this paper, we propose GeoSR, a framework designed to make geometry matter by encouraging VLMs to actively reason with geometry tokens. GeoSR introduces two key components: (1) Geometry-Unleashing Masking, which strategically masks portions of 2D vision tokens during training to weaken non-geometric shortcuts and force the model to consult geometry tokens for spatial reasoning; and (2) Geometry-Guided Fusion, a gated routing mechanism that adaptively amplifies geometry token contributions in regions where geometric evidence is critical. Together, these designs unleash the potential of geometry tokens for spatial reasoning tasks. Extensive experiments on both static and dynamic spatial reasoning benchmarks demonstrate that GeoSR consistently outperforms prior methods and establishes new state-of-the-art performance by effectively leveraging geometric information. The project page is available at https://suhzhang.github.io/GeoSR/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。