用视觉语言模型构建符号化世界模型,实现零样本泛化决策。
From Pixels to Predicates: Learning Symbolic World Models via Pretrained Vision-Language Models
- 利用预训练视觉语言模型生成候选符号谓词并直接从图像评估
- 在模拟与真实场景中实现跨物体类型、布局、数量的强泛化能力
- 适合需要长时序规划与新目标适应的机器人任务
本文旨在解决复杂机器人领域中,仅依赖少量低层技能和短时程演示(含图像序列)时的长期决策问题。核心是学习抽象符号化世界模型,通过规划实现对新目标的零样本泛化。关键在于定义对象属性与关系的符号谓词集合。本文利用预训练视觉语言模型(VLMs)提出大量可能相关的视觉谓词,并直接从摄像头图像中评估其有效性。训练时,将候选谓词与示范输入优化驱动的模型学习算法,获得由精简谓词子集定义的符号化世界模型。测试时,使用VLM构建当前世界状态的符号描述,再通过搜索式规划算法找到实现目标的低层技能序列。实验表明,该方法在仿真与真实世界中均能大幅泛化:适用于多种物体类型、排列方式、物体数量及视觉背景,可应对训练中未见的新目标和远超训练时长的决策序列。
原文摘要 · Abstract (English)
Our aim is to learn to solve long-horizon decision-making problems in complex robotics domains given low-level skills and a handful of short-horizon demonstrations containing sequences of images. To this end, we focus on learning abstract symbolic world models that facilitate zero-shot generalization to novel goals via planning. A critical component of such models is the set of symbolic predicates that define properties of and relationships between objects. In this work, we leverage pretrained vision-language models (VLMs) to propose a large set of visual predicates potentially relevant for decision-making, and to evaluate those predicates directly from camera images. At training time, we pass the proposed predicates and demonstrations into an optimization-based model-learning algorithm to obtain an abstract symbolic world model that is defined in terms of a compact subset of the proposed predicates. At test time, given a novel goal in a novel setting, we use the VLM to construct a symbolic description of the current world state, and then use a search-based planning algorithm to find a sequence of low-level skills that achieves the goal. We demonstrate empirically across experiments in both simulation and the real world that our method can generalize aggressively, applying its learned world model to solve problems with a wide variety of object types, arrangements, numbers of objects, and visual backgrounds, as well as novel goals and much longer horizons than those seen at training time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。