用预训练模型和提示词实现机器人状态识别,无需重训练
Robotic State Recognition with Image-to-Text Retrieval Task of Pre-Trained Vision-Language Model and Black-Box Optimization
- 用图像-文本检索任务匹配预设提示词进行状态判断
- 通过黑箱优化调整提示词权重,提升识别准确率
- 支持透明门、水龙头、厨房清洁度等复杂状态识别
环境与物体状态识别(如门的开闭、灯的开关)对执行日常辅助与安保任务的机器人至关重要。传统方法依赖人工标注训练神经网络、定制传感器或手动编程提取点云或原始图像特征。本文提出一种基于预训练视觉语言模型的机器人状态识别方法,该模型具备图像到文本检索(ITR)能力。预先准备多种语言提示词,通过ITR计算当前图像与各提示词的相似度,结合黑箱优化确定最优提示词权重,实现状态识别。实验表明,仅需设计多个提示词即可完成多种状态识别,无需重新训练网络或手动编程。由于每个识别器只需配置提示词及其权重,无需部署多套模型,有利于资源管理。可成功识别透明门开闭状态、水龙头是否流水,以及厨房清洁程度等以往难以处理的定性状态。
原文摘要 · Abstract (English)
State recognition of the environment and objects, such as the open/closed state of doors and the on/off of lights, is indispensable for robots that perform daily life support and security tasks. Until now, state recognition methods have been based on training neural networks from manual annotations, preparing special sensors for the recognition, or manually programming to extract features from point clouds or raw images. In contrast, we propose a robotic state recognition method using a pre-trained vision-language model, which is capable of Image-to-Text Retrieval (ITR) tasks. We prepare several kinds of language prompts in advance, calculate the similarity between these prompts and the current image by ITR, and perform state recognition. By applying the optimal weighting to each prompt using black-box optimization, state recognition can be performed with higher accuracy. Experiments show that this theory enables a variety of state recognitions by simply preparing multiple prompts without retraining neural networks or manual programming. In addition, since only prompts and their weights need to be prepared for each recognizer, there is no need to prepare multiple models, which facilitates resource management. It is possible to recognize the open/closed state of transparent doors, the state of whether water is running or not from a faucet, and even the qualitative state of whether a kitchen is clean or not, which have been challenging so far, through language.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。