Perceptio: Perception Enhanced Vision Language Models via Spatial Token Generation
Organizations: Amazon
Abstract
Large Vision Language Models (LVLMs) excel at semantic understanding but struggle with fine grained spatial grounding, as the model must implicitly infer complex geometry without ever producing a spatial interpretation. We present Perceptio, a perception enhanced LVLM with 2D and 3D spatial reasoning abilities, enabled via explicit semantic segmentation tokens and depth tokens generated directly within the autoregressive sequence. Concretely, we (i) distill a VQVAE depth codebook from a strong monocular teacher to tokenize dense depth into compact sequences, and (ii) integrate SAM2 based semantic segmentation tokens and VQ-VAE depth tokens inside the LLM so the model first emits spatial tokens and then answers. To stabilize depth token generation, we introduce novel composite depth-token objectives (marker, token, and count losses) and a soft-merging technique for differentiable reconstruction. We adopt a multi-task co-training strategy across diverse datasets, letting the model learn perception tokens to tackle multiple downstream tasks. Building on InternVL, Perceptio achieves state-of-the-art performance across benchmarks: improving referring expression segmentation by +0.8/+1.4/+1.1 cIoU on RefCOCO/+/g HardBLINK spatial understanding accuracy by 10.3%, and MMBench accuracy by 1.0%, demonstrating that explicit spatial chain-of-thought materially strengthens spatial grounding in LVLMs.
Figures & tables
| Method | Image Segmentation | Image Chat / Understanding | Diagram / Science QA | ||||||||
| RefCOCO [ 11 ] | RefCOCO+ [ 11 ] | RefCOCOg [ 41 ] | MME-P [ 9 ] | MME-C [ 9 ] | MME-T | MMBench [ 21 ] | SEED-Bench [ 15 ] | AI2D [ 12 ] | MMStar [ 6 ] | SQA test [ 22 ] | |
| LLAVA-1.5-13B [ 18 ] | – | – | – | – | – | 1531 | 68.8 | 70.1 | – | – | – |
| Video-LLaVA-7B [ 17 ] | – | – | – | – | – | – | 60.9 | – | – | – | – |
| mPLUG-Owl3-8B [ 40 ] | – | – | – | – | – | – | 77.6 | – | – | – | – |
| InternVL2-8B [ 8 ] | – | – | – | – | – | – | 81.7 | 76.2 | 83.8 | – | – |
| PixelLM-7B [ 28 ] | 73.0 | 66.3 | 69.3 | 309 | 135 | 444 | 17.4 | – | 0.0 | – | – |
| Model | HardBLINK | |||
|---|---|---|---|---|
| 3 Points | 4 Points | 5 Points | Average | |
| Sa2VA-8B [ 42 ] | 21.0 | 17.7 | 25.0 | 21.2 |
| InternVL2.5-26B [ 7 ] | 41.1 | 31.5 | 26.6 | 33.1 |
| LLaVA 1.5 13B [ 18 ] | 35.5 | 37.9 | 29.0 | 34.1 |
| Fine-tuned LLaVA [ 3 ] | 58.9 | 52.4 | 41.1 | 50.8 |
| LLaVA-Aurora [ 3 ] | 66.9 | 60.5 | 54.8 | 60.7 |
| Model | 3 Points | 4 Points | 5 Points | Avg. |
|---|---|---|---|---|
| P | 75.8 | 71.0 | 66.1 | 71.0 |
| P ( Depth) | 48.4 | 48.4 | 38.7 | 45.2 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
Large Vision-Language Models
Large vision-language models (LVLMs) connect image and video perception with language generation, reasoning, and interactive tasks.
See all 404 papers