atlas

Research map / Vision-language and multimodal / Multimodal LLMs

SpatialVLM: Endowing Vision-Language Models with Spatial Reasoning Capabilities

CVPR 2024 · Vision-language and multimodal › Multimodal LLMs · approach: Spatial reasoning in VLMs · 234 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Generates 2B spatial VQA pairs from 10M internet images using off-the-shelf vision experts (CLIP filtering, FlexCap captions, ZoeDepth metric depth) and trains a PaLM 2-E-based VLM on them to gain quantitative and qualitative 3D spatial reasoning.

Problem

VLMs struggle with 3D spatial reasoning such as qualitative spatial relations and metric distances/sizes, likely because internet-scale image-caption training data contains little spatial information.

Approach

Automatically lift internet-scale 2D images into metric 3D point clouds using off-the-shelf expert models (open-vocabulary captioning, segmentation, metric depth, horizontal-plane canonicalization), then template-generate millions of direct spatial VQA pairs and co-train a VLM on them alongside PaLM-E data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add synthetic spatial VQA data (co-training) to PaLM 2-E training mixPaLM 2-E based VLM · Qualitative spatial VQA benchmarkAccuracy: 50.4% to 75.2%Helped
Add synthetic spatial VQA data (co-training) to PaLM 2-E training mixPaLM 2-E based VLM · Quantitative spatial VQA benchmarkIn range [50,200]%: 33.9% to 37.2%Helped
Add synthetic spatial VQA data (co-training) to PaLM 2-E training mixPaLM 2-E based VLM · VQA v2 test-devAccuracy: 76.6% to 79.0% (+2.4)Helped
Add synthetic spatial VQA data (co-training) to PaLM 2-E training mixPaLM 2-E based VLM · OKVQAAccuracy: 61.4% to 61.0% (-0.4)No clear effect
Unfreeze the ViT encoder during finetuning on spatial data (ablation)SpatialVLM (PaLM 2-S) · Manipulation quantitative VQAIn range [50,200]%: 34.9% to 37.2% (+2.3)Helped
Unfreeze the ViT encoder during finetuning on spatial data (ablation)SpatialVLM (PaLM 2-S) · Manipulation quantitative VQAIn range [90,110]%: 5.6% to 8.4% (+2.8)Helped
Add Gaussian label noise (std 0 to 0.3) to quantitative answers (ablation)SpatialVLM manipulation quantitative training · Manipulation quantitative VQAIn range [50,200]%: 59.0% (std 0) vs 61.1% (std 0.3)No clear effect
Use LLM (GPT-4) orchestrating SpatialVLM for chain-of-thought spatial reasoningSpatialVLM · Complex spatial reasoning examplesqualitative demonstrationNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: WebLI VQAv2 PaLM-E training mix robotic manipulation dataset

Benchmarks: Spatial VQA human-annotated benchmark (qualitative and quantitative) OKVQA VQA v2

Metrics: accuracy output numbers rate percentage in half-to-twice range of ground truth MSE

Baselines: GPT-4V LLaVA-1.5 InstructBLIP PaLI PaLM-E PaLM 2-E

Builds on: PaLM-E PaLM 2 ZoeDepth FlexCap Socratic Models

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy