用多视角关系蒸馏提升视觉语言模型的空间推理能力
Multi-View Relational Distillation for Spatial Reasoning with Vision-Language Models

- 蒸馏教师模型的跨视角特征相似性,而非直接复制特征
- 在保持语言对齐的同时,空间推理准确率显著提升
- 适合需要轻量级空间理解的机器人与自动驾驶场景
视觉语言模型(VLMs)在图像和视频理解上表现强劲,但其视觉空间表征仍易受几何干扰,导致在具身智能、机器人和自动驾驶中空间推理失败。现有方法或微调VLM进行空间问答,可能强化错误视觉表征;或融合大尺寸几何对齐视觉模型特征,显著增加推理开销。知识蒸馏是替代方案,但直接匹配多视角教师特征会破坏视觉-文本预训练对齐,损害对象与语义理解。本文提出多视角关系蒸馏(MVRD),不直接蒸馏特征,而是蒸馏跨视图的块级余弦相似性。此类关系编码了足够的几何对应信息以支持空间理解,同时使学生模型表征保持未定状态,更贴近原始视觉-语言空间。在多个代表性VLM上,MVRD优于监督微调与特征蒸馏,接近特征融合方法,但参数增加少、延迟低。实验表明,MVRD使视觉表征更具几何性且保留语言对齐,还可泛化至3D场景理解任务,如物体定位、密集描述生成和问答。
原文摘要 · Abstract (English)
Vision-language models (VLMs) have achieved strong image and video understanding, yet their visual-spatial representations remain geometrically fragile, leading to failures in spatial reasoning needed for embodied AI, robotics, and autonomous driving. Prior approaches to geometry grounding either fine-tune VLMs on spatial question answering, which can perpetuate spurious visual representations, or fuse features from large geometry-grounded vision models, which substantially increases model size at inference. Knowledge distillation from geometry-grounded vision models offers an alternative, but directly matching multi-view teacher features can disrupt the pretrained alignment between visual and textual representations, degrading object- and language-semantic capabilities. We propose multi-view relational distillation (MVRD), which distills patch-wise cosine similarities across views instead of the teacher features themselves. These relations encode geometric correspondences adequate for spatial understanding, while leaving the student representation underdetermined, allowing it to remain close to its pretrained vision- language space. Across representative VLMs, MVRD improves visual-spatial reasoning, outperforming supervised fine-tuning and feature distillation while approaching feature fusion methods with considerably fewer added parameters and lower latency. We show that MVRD makes visual representations more geometric while retaining language alignment, and generalizes to 3D scene understanding tasks such as object grounding, dense captioning, and question answering.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。