arXiv:2503.10370cs.ROcs.CV2025-03ICRA被引 18

用语言控制机器人在模拟中练技能,零样本迁移到真实世界。

LUMOS: Language-Conditioned Imitation Learning with World Models

  • 在世界模型潜空间中在线学习,避免策略导致的分布偏移。
  • 仅需不到1%的后见语言标注,在长任务上表现优于现有方法。
  • 适合想用少标注实现多任务语言控制的机器人研究者。

我们提出LUMOS,一种语言条件的多任务模仿学习框架,用于机器人控制。LUMOS通过在学习到的世界模型潜空间中进行大量长时程推演来习得技能,并将这些技能零样本迁移至真实机器人。通过在世界模型潜空间中在线学习,该算法缓解了多数离线模仿学习方法面临的策略诱导分布偏移问题。LUMOS从结构化程度低的玩耍数据中学习,仅需少于1%的后见语言标注,但在测试时可通过语言指令进行引导。我们通过结合图像与语言的后见目标重标注,以及在世界模型潜空间中优化跨多时间步的内在奖励,实现了连贯的长时程性能。在具有挑战性的长时程CALVIN基准测试中,LUMOS在链式多任务评估中超越了先前基于学习的方法,且达到相当水平。据我们所知,这是首个在离线世界模型内实现真实机器人语言条件连续视觉运动控制的工作。视频、数据集和代码详见http://lumos.cs.uni-freiburg.de。

原文摘要 · Abstract (English)

We introduce LUMOS, a language-conditioned multi-task imitation learning framework for robotics. LUMOS learns skills by practicing them over many long-horizon rollouts in the latent space of a learned world model and transfers these skills zero-shot to a real robot. By learning on-policy in the latent space of the learned world model, our algorithm mitigates policy-induced distribution shift which most offline imitation learning methods suffer from. LUMOS learns from unstructured play data with fewer than 1% hindsight language annotations but is steerable with language commands at test time. We achieve this coherent long-horizon performance by combining latent planning with both image- and language-based hindsight goal relabeling during training, and by optimizing an intrinsic reward defined in the latent space of the world model over multiple time steps, effectively reducing covariate shift. In experiments on the difficult long-horizon CALVIN benchmark, LUMOS outperforms prior learning-based methods with comparable approaches on chained multi-task evaluations. To the best of our knowledge, we are the first to learn a language-conditioned continuous visuomotor control for a real-world robot within an offline world model. Videos, dataset and code are available at http://lumos.cs.uni-freiburg.de.

机器人控制语言条件世界模型模仿学习

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