让大模型精准理解物体空间关系,提升机器人导航能力
Spatial-LLaVA: Enhancing Large Language Models with Spatial Referring Expressions for Visual Understanding
- 用新数据集SUN-Spot v2.0训练,强化图像与描述中物体的空间对应
- 在零样本空间推理任务上领先3.15%,显著提升空间理解能力
- 适合自动驾驶、人机交互等需精确定位的现实场景
多模态大语言模型在理解图文输入方面表现优异,但面对空间关系判断或特征相似物体定位等专业任务时,因网络数据稀缺而表现不佳。为此,我们构建了包含90,000张图像-标题对的SUN-Spot v2.0数据集,并为地标物体添加额外标注。每对图像-标题采用Set-of-Marks提示,将图像中的每个地标对象与标题中提及的对象精确映射。我们提出Spatial-LLaVA,基于该数据集和先进语言模型生成的对话数据进行训练,确保图像对象与标题指代间的强对齐,使模型在不依赖物体语义信息的情况下学习空间指代表达。Spatial-LLaVA在零样本视觉空间推理基准测试中比先前方法提升3.15%。该模型专精于空间指代理解,适用于自动驾驶、交互式机器人等需要高精度目标识别的现实应用。
原文摘要 · Abstract (English)
Multimodal large language models (MLLMs) have demonstrated remarkable abilities in comprehending visual input alongside text input. Typically, these models are trained on extensive data sourced from the internet, which are sufficient for general tasks such as scene understanding and question answering. However, they often underperform on specialized tasks where online data is scarce, such as determining spatial relationships between objects or localizing unique target objects within a group of objects sharing similar features. In response to this challenge, we introduce the SUN-Spot v2.0 dataset1, now comprising a total of 90k image-caption pairs and additional annotations on the landmark objects. Each image-caption pair utilizes Set-of-Marks prompting as an additional indicator, mapping each landmark object in the image to the corresponding object mentioned in the caption. Furthermore, we present Spatial-LLaVA, an MLLM trained on conversational data generated by a state-of-the-art language model using the SUNSpot v2.0 dataset. Our approach ensures a robust alignment between the objects in the images and their corresponding object mentions in the captions, enabling our model to learn spatial referring expressions without bias from the semantic information of the objects. Spatial-LLaVA outperforms previous methods by 3.15% on the zero-shot Visual Spatial Reasoning benchmark dataset. Spatial-LLaVA is specifically designed to precisely understand spatial referring expressions, making it highly applicable for tasks in real-world scenarios such as autonomous navigation and interactive robotics, where precise object recognition is critical.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。