用检索增强的视觉语言模型分析机器人策略泛化所需类型。
Grounding Robot Generalization in Training Data via Retrieval-Augmented VLMs
- 通过通用策略嵌入检索相关训练数据,定位关键样本。
- 视觉语言模型可准确判断任务与训练数据的差异类型。
- 适用于评估机器人在真实场景中的泛化能力,适合研究者使用。
近期机器人操作研究推动了策略在新场景下的泛化能力。然而,如何准确界定不同评估设置是否真正代表从训练分布出发的泛化仍具挑战。为实现更精确的泛化评估,我们提出RADAR——一个可扩展的框架,通过直接对比测试时的任务与策略训练数据,确定所需的泛化类型。RADAR采用两阶段流程:首先利用通用策略嵌入进行检索,找出与特定评估任务相关的训练样本;随后,视觉语言模型(VLMs)分析该任务与检索到的数据在多个维度上的差异,输出可解释的分析结果,并对所需泛化类型进行分类。受控实验表明,VLMs能有效分析泛化数据,且检索步骤显著提升分类准确性。我们将RADAR扩展至大规模数据集,结果与先前人工定义的基准条件高度一致。演示见 radar-analysis.github.io。
原文摘要 · Abstract (English)
Recent work on robot manipulation has advanced policy generalization to novel scenarios. However, it is often difficult to characterize how different evaluation settings actually represent generalization from the training distribution of a given policy. To work towards more precise evaluation of generalization in robotics, we propose RADAR, a scalable framework for directly comparing test-time evaluation tasks to policy training data, to determine what form of policy generalization is required. RADAR consists of a two-stage pipeline: first, retrieval using generalist policy embeddings identifies which training examples are relevant for a given evaluation task. Next, vision-language models (VLMs) analyze the evaluation task against the retrieved data, outputting interpretable analysis on how they compare along a variety of axes, and an overall classification of what type of policy generalization is required. Through controlled experiments, we demonstrate that VLMs are effective at analyzing data for generalization, and that our retrieval step effectively identifies examples needed to make accurate classifications with respect to the training data. Furthermore, we scale RADAR to large-scale datasets, where we observe agreement with human-defined benchmark conditions from prior work. We provide demonstrations at radar-analysis.github.io.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。