arXiv:2412.15109cs.RO2024-12ICLR被引 179

用视觉预测反推动作,让机器人更聪明地学习抓取。

Predictive Inverse Dynamics Models are Scalable Learners for Robotic Manipulation

  • 用视觉状态反推动作,实现视觉与动作的闭环学习。
  • 在真实场景中提升43%性能,新模型在多个基准上创纪录。
  • 适合想用少量数据快速适配新环境的机器人研发者。

当前机器人操作的学习方法主要分为两类:一类聚焦于‘动作’,通过大量机器人数据进行行为克隆;另一类强调‘视觉’,利用大规模视觉数据预训练表示或生成模型(即世界模型)以增强泛化能力。本文提出一种端到端范式——预测性逆动力学模型(PIDM),通过基于机器人预测视觉状态来反推动作。该方法在视觉与动作之间形成闭环,使模型成为更优的可扩展动作学习器。我们采用Transformer架构处理视觉状态和动作,命名为Seer。Seer在大规模机器人数据集(如DROID)上预训练,并仅需少量微调数据即可适配现实场景。得益于大规模端到端训练及视觉与动作的协同作用,Seer在仿真与真实实验中均显著优于以往方法,在LIBERO-LONG基准上提升13%,CALVIN ABC-D提升21%,真实任务提升43%。尤其在CALVIN ABC-D上达到平均长度4.28的新纪录,对新物体、光照变化和强干扰环境均有出色泛化能力。代码与模型已开源。

原文摘要 · Abstract (English)

Current efforts to learn scalable policies in robotic manipulation primarily fall into two categories: one focuses on "action," which involves behavior cloning from extensive collections of robotic data, while the other emphasizes "vision," enhancing model generalization by pre-training representations or generative models, also referred to as world models, using large-scale visual datasets. This paper presents an end-to-end paradigm that predicts actions using inverse dynamics models conditioned on the robot's forecasted visual states, named Predictive Inverse Dynamics Models (PIDM). By closing the loop between vision and action, the end-to-end PIDM can be a better scalable action learner. In practice, we use Transformers to process both visual states and actions, naming the model Seer. It is initially pre-trained on large-scale robotic datasets, such as DROID, and can be adapted to realworld scenarios with a little fine-tuning data. Thanks to large-scale, end-to-end training and the synergy between vision and action, Seer significantly outperforms previous methods across both simulation and real-world experiments. It achieves improvements of 13% on the LIBERO-LONG benchmark, 21% on CALVIN ABC-D, and 43% in real-world tasks. Notably, Seer sets a new state-of-the-art on CALVIN ABC-D benchmark, achieving an average length of 4.28, and exhibits superior generalization for novel objects, lighting conditions, and environments under high-intensity disturbances on real-world scenarios. Code and models are publicly available at https://github.com/OpenRobotLab/Seer/.

机器人操作视觉动作端到端学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。