arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2210.03094cs.ROcs.AIcs.LG
收藏

VIMA:基于多模态提示的通用机器人操作

VIMA: General Robot Manipulation with Multimodal Prompts

Stanford University(斯坦福大学);Macalester College(麦卡莱斯特学院);Allen Institute for AI(艾伦人工智能研究所);NVIDIA(英伟达)

Yunfan Jiang, Agrim Gupta, Zichen Zhang, Guanzhi Wang, Yongqiang Dou, Yanjun Chen, Li Fei-Fei, Anima Anandkumar, Yuke Zhu, Linxi Fan

AI总结:

本文提出VIMA,一种基于Transformer的机器人智能体,通过多模态提示统一表达多种操作任务,并构建大规模仿真基准,在零样本泛化中显著优于现有方法,同时具备高数据效率。

摘要:

基于提示的学习已成为自然语言处理中的成功范式,其中单个通用语言模型可以被指示执行由输入提示指定的任何任务。然而,机器人技术中的任务指定形式多样,例如模仿一次性演示、遵循语言指令以及达到视觉目标。它们通常被视为不同任务,并由专用模型处理。我们展示了广泛的机器人操作任务可以通过交错文本和视觉标记的多模态提示来表达。据此,我们开发了一个新的仿真基准,该基准包含数千个程序化生成的桌面任务及多模态提示、超过60万条用于模仿学习的专家轨迹,以及一个用于系统泛化的四级评估协议。我们设计了一个基于Transformer的机器人智能体VIMA,它处理这些提示并自回归地输出电机动作。VIMA采用了一种配方,实现了强大的模型可扩展性和数据效率。在相同的训练数据下,它在最困难的零样本泛化设置中,任务成功率比替代设计高出最多2.9倍。在训练数据减少10倍的情况下,VIMA的性能仍比最佳竞争变体高出2.7倍。代码和视频演示可在https://vimalabs.github.io/获取。

英文摘要:

Prompt-based learning has emerged as a successful paradigm in natural language processing, where a single general-purpose language model can be instructed to perform any task specified by input prompts. Yet task specification in robotics comes in various forms, such as imitating one-shot demonstrations, following language instructions, and reaching visual goals. They are often considered different tasks and tackled by specialized models. We show that a wide spectrum of robot manipulation tasks can be expressed with multimodal prompts, interleaving textual and visual tokens. Accordingly, we develop a new simulation benchmark that consists of thousands of procedurally-generated tabletop tasks with multimodal prompts, 600K+ expert trajectories for imitation learning, and a four-level evaluation protocol for systematic generalization. We design a transformer-based robot agent, VIMA, that processes these prompts and outputs motor actions autoregressively. VIMA features a recipe that achieves strong model scalability and data efficiency. It outperforms alternative designs in the hardest zero-shot generalization setting by up to $2.9\times$ task success rate given the same training data. With $10\times$ less training data, VIMA still performs $2.7\times$ better than the best competing variant. Code and video demos are available at https://vimalabs.github.io/

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑