让视觉语言模型学会识别物体状态,提升机器人抓取决策能力
StateVLM: A State-Aware Vision-Language Model for Robotic Affordance Reasoning

- 用辅助回归损失训练模型,实现物体位置与状态的精准定位
- 在自建数据集OSAR上,性能比无该机制模型高5.2%
- 特别适合需要理解物体状态的机器人操作任务
视觉语言模型(VLMs)在机器人任务中表现优异,能感知视觉信息并理解自然语言指令。然而,由于大语言模型固有的局限性,它们在数值推理方面表现不佳,尤其在物体检测和物体状态定位任务中。为此,本文提出一种新训练策略,将数值推理作为回归任务引入VLM。该方法利用框解码器输出计算辅助回归损失(ARL),在微调阶段提升模型对物体位置与状态的感知能力,同时保持推理时的标准序列预测。基于此策略,我们构建了StateVLM(状态感知视觉语言模型),可精细感知物体位置、状态及可抓取区域。由于缺乏相关基准,我们发布了开源数据集OSAR,包含1172个场景、7746个物体及其边界框。在RefCOCO系列数据集上的对比实验显示,加入ARL后模型平均提升1.6%;在OSAR上,StateVLM+ARL相较基线平均提升5.2%,尤其在复杂抓取推理任务中显著增强输出一致性。
原文摘要 · Abstract (English)
Vision-language models (VLMs) have shown remarkable performance in various robotic tasks, as they can perceive visual information and understand natural language instructions. However, when applied to robotics, VLMs remain subject to a fundamental limitation inherent in large language models (LLMs): they struggle with numerical reasoning, particularly in object detection and object-state localization. To explore numerical reasoning as a regression task in VLMs, we propose a novel training strategy to adapt VLMs for object detection and object-state localization. This approach leverages box decoder outputs to compute an Auxiliary Regression Loss (ARL) during fine-tuning, while preserving standard sequence prediction at inference. We leverage this training strategy to develop StateVLM (State-aware Vision-Language Model), a novel model designed to perceive and learn fine-grained object representations, including precise localization of objects and their states, as well as graspable regions. Due to the lack of a benchmark for object-state affordance reasoning, we introduce an open-source benchmark, Object State Affordance Reasoning (OSAR), which contains 1172 scenes with 7746 individual objects and corresponding bounding boxes. Comparative experiments on adapted benchmarks (RefCOCO, RefCOCO+, and RefCOCOg) demonstrate that ARL improves model performance by an average of 1.6% compared to models without ARL. Experiments on the OSAR benchmark further support this finding, showing that StateVLM with ARL achieves an average of 5.2% higher performance than models without ARL. In particular, ARL is also important for the complex task of affordance reasoning in OSAR, where it enhances the consistency of model outputs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。