arXiv:2601.12626cs.CV2026-01被引 9

发现视觉语言模型用线性绑定空间ID进行时空推理

Linear Mechanisms for Spatiotemporal Reasoning in Vision Language Models

  • 通过线性绑定空间ID到文本激活,实现位置信息编码
  • 空间ID在模型中间层可系统性影响推理结果
  • 适用于分析模型缺陷与改进设计,适合对齐研究者

时空推理是视觉语言模型(VLMs)的突出能力,但其内在机制仍不清晰。我们假设视觉/几何和文本的空间结构表示会在模型计算中某处融合。通过搜索这一融合点,我们发现视觉语言模型通过线性绑定空间ID到文本激活来编码物体位置,并通过语言令牌进行推理。通过严格的因果干预实验,我们证明这些普遍存在于模型中的空间ID能系统性地中介中间层的模型信念。此外,空间ID可作为诊断现有VLM局限性的工具,并提供有价值的训练信号。我们将分析扩展至视频VLM,发现类似的线性时间ID机制。通过刻画所提出的时空ID机制,我们揭示了视觉语言模型中此前未被充分探索的内部推理过程,为提升可解释性及更对齐、更强大的模型设计奠定基础。代码已开源:https://github.com/Raphoo/linear-mech-vlms。

原文摘要 · Abstract (English)

Spatio-temporal reasoning is a remarkable capability of Vision Language Models (VLMs), but the underlying mechanisms of such abilities remain largely opaque. We postulate that visual/geometrical and textual representations of spatial structure must be combined at some point in VLM computations. We search for such confluence, and ask whether the identified representation can causally explain aspects of input-output model behavior through a linear model. We show empirically that VLMs encode object locations by linearly binding \textit{spatial IDs} to textual activations, then perform reasoning via language tokens. Through rigorous causal interventions we demonstrate that these IDs, which are ubiquitous across the model, can systematically mediate model beliefs at intermediate VLM layers. Additionally, we find that spatial IDs serve as a diagnostic tool for identifying limitations in existing VLMs, and as a valuable learning signal. We extend our analysis to video VLMs and identify an analogous linear temporal ID mechanism. By characterizing our proposed spatiotemporal ID mechanism, we elucidate a previously underexplored internal reasoning process in VLMs, toward improved interpretability and the principled design of more aligned and capable models. We release our code for reproducibility: https://github.com/Raphoo/linear-mech-vlms.

视觉语言模型时空推理可解释性线性机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。