WM-VLM:探析交错式视觉-文本推理中的内部世界模型
TLDR
提出WM-VLM,用轻量级世界模型分支生成中间视觉状态,使2D/3D心理旋转推理最高提升39.25个百分点。
评分理由
Strengths include a clear two-stage training procedure, programmatically verifiable spatial tasks, strong gains over supervised fine-tuning, and ablations showing dependence on generated visual states. Weaknesses are that evaluations use synthetic mental rotation tasks rather than real-world benchmarks, and the world model scope is limited to next-state generation for spatial reasoning, not broader dynamics.
Read-first 评分解释
综合优先阅读分 34.4,由主题、引用、图谱、方法、可复现性和近期性等信号加权得到。 原始总分保留为 38。
研究版图角色
排序敏感性
稳定性:volatile;排名波动范围:278。
关键词评分
深度分析
创新点
- 提出WM-VLM,为预训练的VLM配备一个轻量级世界模型分支,用于生成中间视觉状态。
- 采用两阶段训练:先学习生成下一个视觉状态,再学习利用该生成状态进行推理。
- 构造了具有可验证中间视觉状态的程序化空间推理任务,以同时评估生成质量和对视觉状态的依赖程度。
- 证明了与仅使用语言的监督微调VLM相比,生成的视觉状态能提升空间推理表现。
方法
WM-VLM在预训练的VLM上增加一个轻量级世界模型分支,用于生成中间视觉状态。模型采用两阶段训练:首先生成下一个视觉状态,然后利用该状态进行推理。评估使用程序化构造的2D和3D心理旋转任务,这些任务具有可验证的中间视觉状态,并与监督微调的骨干模型进行比较,同时对生成的视觉状态进行消融实验。
关键结果
WM-VLM在2D和3D心理旋转任务上持续优于监督微调的骨干模型,提升最高达39.25个百分点。消融实验显示,移除或破坏生成的视觉状态会使性能急剧下降。