SpatialBoost: Enhancing Visual Representation through Language-Guided Reasoning
Organizations: KAIST · RLWRLD · NAVER Cloud
Abstract
Despite the remarkable success of large-scale pre-trained image representation models (i.e., vision encoders) across various vision tasks, they are predominantly trained on 2D image data and therefore often fail to capture 3D spatial relationships between objects and backgrounds in the real world, constraining their effectiveness in many downstream applications. To address this, we propose SpatialBoost, a scalable framework that enhances the spatial awareness of existing pre-trained vision encoders by injecting 3D spatial knowledge expressed in linguistic descriptions. The core idea involves converting dense 3D spatial information from 2D images into linguistic expressions, which is then used to inject such spatial knowledge into vision encoders through a Large Language Model (LLM). To this end, we adopt a multi-turn Chain-of-Thought (CoT) reasoning process that progressively incorporates dense spatial knowledge and builds hierarchical spatial understanding. To validate effectiveness, we adapt SpatialBoost to state-of-the-art vision encoders such as DINOv3, and evaluate its performance gains on a wide range of benchmarks requiring both 3D perception and general vision abilities. For instance, SpatialBoost improves DINOv3 performance from 55.9 to 59.7 mIoU on ADE20K, achieving state-of-the-art performance with 3.8% gain over the pre-trained DINOv3.
Figures & tables
| VLR | VG | GU | 3D SU | ||||
| Method | ScanQA | SQA3D | ScanRefer (%) | [email protected] (%) | RTE (m) | Acc | mIoU |
| Vision-only trained encoder | |||||||
| V-JEPAv2 [ 4 ] | 37.7 | 49.0 | 55.5 | 95.4 | 0.08 | 69.5 | 33.2 |
| Vision-Language trained encoder | |||||||
| AIMv2 [ 29 ] | 37.4 | 48.1 | 50.9 | 47.8 | 0.31 | 53.9 | 11.8 |
| dino.txt [ 46 ] | 39.8 | 50.4 | 52.5 | 81.7 | 0.16 | 85.7 | 59.4 |
| Image classification | Image retrieval | |||||
| Method | ImageNet (kNN) | ImageNet (lin.) | Oxford-H | Paris-H | Met (GAP) | AmsterTime |
| Vision-only trained encoder | ||||||
| V-JEPAv2 [ 4 ] | 82.1 | 84.0 | 40.6 | 76.1 | 38.2 | 31.9 |
| Vision-Language trained encoder | ||||||
| AIMv2 [ 29 ] | 85.7 | 88.3 | 60.8 | 86.4 | 52.0 | 50.3 |
| dino.txt [ 46 ] | 80.0 | 81.4 | 49.7 | 78.5 | 41.8 | 33.1 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| system prompt [ |
| “ You are a helpful multimodal assistant. |
| Generate question-answer pairs for given two images. |
| Both images are came from same scene. |
| When referring to the image, please call it the first image or the second image. " |
| ] |
| general vqa prompt [ |
| Label type | Labels |
|---|---|
| Positive labels | “ an iPhone photo of an indoor scene " |
| “ an iphone photo of an outdoor scene " | |
| “ a DSLR photo of an indoor scene " | |
| “ a DSLR of an outdoor scene " | |
| Negative labels | “ a close up shot of a single object " |
| “ a product displayed in front of a white background " |
| single point questions [ |
| “ What is the depth value at pixel point [A]? " |
| “ How far away is point [A]? " |
| “ Tell me the depth of point [A]. " |
| ] |
| single point answers |
| “ [X] away. " |
| bounding cube questions [ |
| “ Identify [A] and [B] " |
| “ What is the center of the 3d bounding box coordinate for [A]? " |
| ] |
| bounding cube answers [ |
| “ [X] " |
| “ Center: [X] " |
| distance questions [ |
| “ What is the distance between the [A] and the [B]? " |
| “ How far is the [A] from the [B]? " |
| “ How distant is the [A] from the [B]? " |
| “ Measure the distance from the [A] to the [B]. " |
| ] |
| distance answers [ |
| Depth | Segmentation | Classification | ||||||||
| Method | OpenCLIP | DINOv2 | OpenCLIP | DINOv2 | OpenCLIP | DINOv2 | ||||
| lin. | DPT | lin. | DPT | lin. | +ms | lin. | +ms | |||
| Pre-trained | 0.56 | 0.41 | 0.38 | 0.29 | 39.1 | 45.7 | 47.7 | 53.1 | 83.9 | 86.3 |
| Pix | 0.52 | 0.40 | 0.34 | 0.29 | 39.6 | 46.3 | 48.2 | 53.4 | 84.0 | 86.6 |
| Obj | 0.53 | 0.41 | 0.37 | 0.30 | 39.4 | 46.3 | 48.0 | 53.3 | 84.7 | 87.2 |
| Scene | 0.53 | 0.41 | 0.38 | 0.32 | 39.2 | 45.9 | 47.7 | 53.3 | 84.5 | 87.1 |
| Method | Classification | Segmentation | Depth estimation |
|---|---|---|---|
| DINOv2 (Pre-trained) | 86.3 | 47.7 | 0.38 |
| Full Fine-tuning | 79.5 | 49.4 | 0.31 |
| LoRA | 81.7 | 49.0 | 0.32 |
| Dual-Channel Attn. | 87.6 | 49.2 | 0.32 |
| Model | Sample size | Classification | Segmentation | Depth estimation |
|---|---|---|---|---|
| SigLIPv2 | Pre-trained | 89.1 | 42.8 | 0.51 |
| +SpatialBoost | 50K | 89.5 | 43.2 | 0.44 |
| 100K | 89.7 | 44.5 | 0.42 | |
| 300K | 90.0 | 45.1 | 0.39 | |
| DINOv3 | Pre-trained | 88.4 | 55.9 | 0.31 |
| +SpatialBoost | 50K | 88.6 | 56.8 | 0.29 |
Visual Representation Learning
Visual representation learning develops image and video features that capture useful structure for recognition, retrieval, spatial reasoning, and action.
See all 224 papers