CausalSpatial
收藏资源简介:
CausalSpatial是首个用于评估对象中心因果空间推理能力的诊断基准数据集。它采用对象中心的表述方式,每个查询都基于图像中目标对象的特定假设运动,探究模型是否能在四个预期任务(碰撞、兼容性、遮挡和轨迹)中模拟实例级动态的后果。
CausalSpatial is the first diagnostic benchmark dataset for evaluating object-centric causal spatial reasoning capabilities. It adopts an object-centric representation framework, where each query is based on the specific hypothesized motion of a target object in an image, to examine whether a model can simulate the consequences of instance-level dynamics across four intended tasks: collision, compatibility, occlusion, and trajectory.
CausalSpatial 数据集概述
数据集基本信息
- 数据集名称:CausalSpatial
- 核心定位:首个用于评估因果空间推理能力的诊断性基准。
- 核心任务:评估模型在对象中心化的假设运动下,进行因果空间推理的能力。
- 论文状态:论文已发布于arXiv(arXiv:2601.13304),数据集已发布。
数据集设计
- 设计理念:采用对象中心化的构建方式。每个查询都基于图像中对目标对象施加的特定假设运动。
- 评估能力:探究模型是否能在四个预测任务中模拟实例级动态的后果。
- 四大任务:
- 碰撞
- 兼容性
- 遮挡
- 轨迹
数据内容与结构
- 数据格式:可通过Hugging Face
datasets库加载。 - 子集划分:数据集包含多个子集,例如
collision。 - 评估工具兼容:提供了用于
VLMEvalKit评估框架的TSV格式文件。
数据获取与加载
-
Hugging Face仓库ID:
Mwxinnn/CausalSpatial -
加载示例: python from datasets import load_dataset dataset = load_dataset("Mwxinnn/CausalSpatial", "collision", split="train")
-
VLMEvalKit文件下载: python from huggingface_hub import hf_hub_download file_path = hf_hub_download(repo_id="Mwxinnn/CausalSpatial", filename="VLMEvalKit/CausalSpatial.tsv", repo_type="dataset")
评估方法
- 支持模型:可评估包括GPT5、Claude、Gemini、Qwen2.5 VL、Qwen3-VL在内的多模态大语言模型。
- 评估流程:支持直接评估,也支持结合轨迹预测与视频生成框架(COW)进行间接评估。
- COW框架:需要额外使用
IWM管道进行轨迹预测和视频生成,以辅助模型推理。
依赖环境与工具
- 关键依赖项目:
- MapAnything:用于从输入回归场景的3D几何。
- ATI:用于视频生成的轨迹运动控制框架。
- 环境配置:需安装指定
requirements.txt,并下载相关模型权重。
引用信息
-
引用格式:
@article{ma2025causalspatial, title={CausalSpatial: A Comprehensive Benchmark for Object-Centric Causal Spatial Reasoning}, author={Ma, Wenxin and Wang, Chenlong and Yuan, Ruisheng and Chen, Hao and Dai, Nanru and Zhou, S. Kevin and Yang, Yijun and Yuille, Alan and Chen, Jieneng}, journal={arXiv preprint arXiv:2601.13304}, year={2026} }




