SpatialVLM: Endowing Vision-Language Models with Spatial Reasoning Capabilities

Boyuan Chen, Zhuo Xu, Sean Kirmani, Brian Ichter, Danny Driess, Pete Florence, Dorsa Sadigh, Leonidas J. Guibas, Fei Xia

arXiv:2401.12168 · 2026-07-27 공개 · arXiv · PDF

vision-language robotics spatial-reasoning vqa dataset-generation metric-space training-pipeline internet-scale

Abstract

Understanding and reasoning about spatial relationships is a fundamental capability for Visual Question Answering (VQA) and robotics. While Vision Language Models (VLM) have demonstrated remarkable performance in certain VQA benchmarks, they still lack capabilities in 3D spatial reasoning, such as recognizing quantitative relationships of physical objects like distances or size difference. We hypothesize that VLMs' limited spatial reasoning capability is due to the lack of 3D spatial knowledge in training data and aim to solve this problem by training VLMs with Internet-scale spatial reasoning data. To this end, we present a system to facilitate this approach. We first develop an automatic 3D spatial VQA data generation framework that scales up to 2 billion VQA examples on 10 million real-world images. We then investigate various factors in training recipe including data quality, training pipeline and VLM architecture. Our workfeatures the first Internet-scale 3D spatial reasoning dataset in metric space. By training a VLM on such data, we significantly enhance its ability on both qual-itative and quantitative spatial VQA. Finally, we demonstrate that this VLM unlocks novel downstream applications in chain-of thought spatial reasoning and robotics due to its quantitative estimation capability. Website: https://spatial-vlm.github.iol

한국어 요약

한 줄 요약

SpatialVLM은 20억 개의 3D 공간 추론 VQA 데이터로 학습한 VLM을 통해 공간 추론 능력을 향상시킨다.

핵심 기여도

핵심 아이디어

기존 VLM은 이미지-캡션 쌍으로 훈련되어 공간 정보가 부족한 데이터에 의존하므로 3D 공간 추론 능력이 제한된다. 이에 저자들은 오프-더-쉘프 비전 모델을 활용해 2D 이미지에서 3D 공간 정보를 추출하는 자동 라벨링 프레임워크를 제안한다. 구체적으로, 오픈-바이어보러리 디텍션, 메트릭 깊이 추정, 세멘틱 세그멘테이션, 객체 중심 캡셔닝 모델을 결합하여 실제 데이터를 대규모로 라벨링한다. 이는 기존의 합성 이미지 기반 데이터 생성 방식과 달리, 현실 세계의 복잡성과 다양성을 반영한다. 저자들은 이를 통해 VLM이 공간적 관계를 정량적으로 이해할 수 있도록 훈련시킨다.

기술적 접근법

주요 결과

의의 및 한계

이 연구는 VLM이 인간처럼 3D 공간을 이해할 수 있도록 하는 기초를 마련했다. 특히, 인터넷 규모의 실제 이미지 기반 3D 공간 데이터셋은 기존 학습 데이터의 한계를 극복하는 데 기여한다. 또한, 공간 추론 능력을 갖춘 VLM은 로봇공학, 증강현실 등 다양한 분야에서 활용 가능하다. 그러나 본 연구의 공간 질문 템플릿은 유한하며, 더 복잡한 기하학적 원리에 대한 연구가 필요하다는 한계가 있다.

실용적 활용

SpatialVLM은 로봇 작업의 보상 함수로 사용될 수 있으며, 체인-오브-사고 추론을 통해 복잡한 공간 문제 해결에 활용될 수 있다. 또한, 객체 크기 추정과 같은 정량적 추론 능력은 물체 재배치, AR/VR 등에서 실용적 가치를 가진다.