Visual Spatial Tuning (VST)
收藏资源简介:
Visual Spatial Tuning (VST) 是一个全面的框架,旨在培养视觉语言模型(VLMs)具备类人的视觉空间能力——从空间感知到高级推理。它包括VST-Perception (VST-P),包含4.1M样本,覆盖19个技能,涉及单图像、多图像场景和视频,以增强VLMs的空间感知;以及VST-Reasoning (VST-R),包含135K精选样本,用于教导模型进行空间推理,包括逐步推理和基于规则的数据,以通过强化学习增强推理能力。
Visual Spatial Tuning (VST) is a comprehensive framework designed to equip Vision-Language Models (VLMs) with human-like visuospatial capabilities, spanning from spatial perception to advanced reasoning. It comprises two components: VST-Perception (VST-P), which contains 4.1 million samples across 19 skills, covering single-image, multi-image scenarios and videos to enhance the spatial perception of VLMs; and VST-Reasoning (VST-R), which includes 135,000 curated samples for teaching models to perform spatial reasoning, including step-by-step reasoning and rule-based data, to strengthen reasoning abilities via reinforcement learning.
Visual Spatial Tuning (VST) 数据集概述
框架介绍
Visual Spatial Tuning (VST) 是一个综合性框架,旨在培养视觉语言模型具备类人空间视觉能力——从空间感知到高级推理。
数据集组成
VST-Perception (VST-P)
- 样本规模:410万样本
- 任务数量:19个任务
- 覆盖场景:
- 单图像
- 多图像
- 视频
- 性能提升:
- CVBench-3D:提升约20%
- BLINK:提升约5%
- VSIBench:提升约16%
VST-Reasoning (VST-R)
- 样本规模:13.5万样本
- 数据划分:
- 推理步骤数据 (CoT):教授模型如何进行空间推理
- 规则可检查数据:用于在线强化学习以增强推理技能
- 性能提升:
- MMSI-Bench:提升8.9%
训练流程
采用渐进式训练管道:
- 监督微调:建立基础空间知识
- 强化学习:增强空间推理能力
模型性能
空间与通用基准测试
| 模型 | CV | 3DSR | MMSI | BLINK | VSI | MMStar | MMB | RealworldQA | MMMU | OCRB | AI2D |
|---|---|---|---|---|---|---|---|---|---|---|---|
| VST-3B-SFT | 84.4 | 54.1 | 30.2 | 59.1 | 57.9 | 58.0 | 80.9 | 68.4 | 45.2 | 83.7 | 82.5 |
| VST-3B-RL | 84.2 | 56.5 | 31.3 | 57.2 | 57.7 | 58.9 | 80.5 | 68.5 | 49.8 | 80.9 | 82.4 |
| VST-7B-SFT | 85.5 | 54.6 | 32.0 | 62.1 | 60.6 | 63.1 | 83.3 | 72.2 | 50.6 | 85.5 | 84.9 |
| VST-7B-RL | 86.5 | 60.1 | 34.8 | 62.6 | 61.2 | 63.5 | 83.0 | 68.5 | 49.4 | 86.1 | 83.5 |
VSIBench 详细性能
| 方法 | 平均 | 物体计数 | 绝对距离 | 物体大小 | 房间大小 | 相对距离 | 相对方向 | 路径规划 | 接近顺序 |
|---|---|---|---|---|---|---|---|---|---|
| VST-3B-SFT | 57.9 | 69.3 | 45.4 | 71.8 | 62.4 | 59.0 | 46.0 | 38.7 | 70.2 |
| VST-3B-RL | 57.7 | 66.6 | 45.0 | 72.8 | 60.9 | 59.9 | 47.6 | 40.7 | 68.3 |
| VST-7B-SFT | 60.6 | 72.0 | 44.4 | 74.3 | 68.3 | 59.7 | 55.8 | 44.9 | 65.2 |
| VST-7B-RL | 61.2 | 71.6 | 43.8 | 75.5 | 69.2 | 60.0 | 55.6 | 44.3 | 69.2 |
SUN RGBD 3D 物体检测
| 方法 | AP@15 |
|---|---|
| Seed1.5-VL | 33.5 |
| Gemini-2.0-Pro | 32.5 |
| Gemini Robotics-ER | 48.3 |
| VST-3B-SFT | 37.3 |
| VST-3B-RL | 40.1 |
| VST-7B-SFT | 41.6 |
| VST-7B-RL | 44.2 |
模型可用性
| 模型名称 | HuggingFace 地址 |
|---|---|
| VST-3B-SFT | https://huggingface.co/rayruiyang/VST-3B-SFT |
| VST-3B-RL | https://huggingface.co/rayruiyang/VST-3B-RL |
| VST-7B-SFT | https://huggingface.co/rayruiyang/VST-7B-SFT |
| VST-7B-RL | https://huggingface.co/rayruiyang/VST-7B-RL |
应用领域
- 视觉语言模型空间能力增强
- 机器人学习强化
- 物理基础人工智能发展




