How Do LLMs and VLMs Understand Viewpoint Rotation Without Vision? An Interpretability Study
AI 摘要
研究了LLM和VLM在无视觉输入下对视角旋转的理解能力,并进行了可解释性分析。
主要贡献
- 提出了一个用于评估语言模型视角旋转理解(VRU)能力的数据集
- 分析了LLM和VLM在VRU任务中的表现差距,发现模型难以将视角位置与观察结果绑定
- 通过选择性微调关键注意力头,提升了VRU性能且避免了灾难性遗忘
方法论
构建数据集评估LLM和VLM,使用分层探测分析和注意力头因果干预来理解模型内部机制,并选择性微调。
原文摘要
Over the past year, spatial intelligence has drawn increasing attention. Many prior works study it from the perspective of visual-spatial intelligence, where models have access to visuospatial information from visual inputs. However, in the absence of visual information, whether linguistic intelligence alone is sufficient to endow models with spatial intelligence, and how models perform relevant tasks with text-only inputs still remain unexplored. Therefore, in this paper, we focus on a fundamental and critical capability in spatial intelligence from a linguistic perspective: viewpoint rotation understanding (VRU). Specifically, LLMs and VLMs are asked to infer their final viewpoint and predict the corresponding observation in an environment given textual description of viewpoint rotation and observation over multiple steps. We find that both LLMs and VLMs perform poorly on our proposed dataset while human can easily achieve 100% accuracy, indicating a substantial gap between current model capabilities and the requirements of spatial intelligence. To uncover the underlying mechanisms, we conduct a layer-wise probing analysis and head-wise causal intervention. Our findings reveal that although models encode viewpoint information in the hidden states, they appear to struggle to bind the viewpoint position with corresponding observation, resulting in a hallucination in final layers. Finally, we selectively fine-tune the key attention heads identified by causal intervention to improve VRU performance. Experimental results demonstrate that such selective fine-tuning achieves improved VRU performance while avoiding catastrophic forgetting of generic abilities. Our dataset and code will be released at https://github.com/Young-Zhen/VRU_Interpret .