用游戏数据训练视觉世界模型,结合蒙特卡洛树搜索实现机器人长序列规划。
WorldPlanner: Monte Carlo Tree Search and MPC with Action-Conditioned Visual World Models
- 基于少量玩耍数据学习动作条件视觉模型与扩散采样器
- 在真实机器人上完成3项任务,规划性能显著优于行为克隆基线
- 适合需要复杂推理的机器人任务,尤其擅长减少模型幻觉
机器人需从原始感官输入理解环境,并推理行动后果以解决复杂任务。行为克隆(BC)利用特定任务的人类示范学习端到端策略,但难以迁移且数据生成困难。本文采用模型驱动方法,仅需数小时无需刻意设计的玩耍数据,即可学习动作条件视觉世界模型、基于扩散的行动采样器,以及可选的奖励模型。该世界模型结合行动采样器与奖励模型,通过蒙特卡洛树搜索(MCTS)优化长序列行动。规划结果由零阶模型预测控制(MPC)执行。实验验证了该方法在3个真实机器人任务中的有效性,涵盖不同规划与建模复杂度。结果显示,规划策略在标准操作测试环境中显著优于行为克隆基线。
原文摘要 · Abstract (English)
Robots must understand their environment from raw sensory inputs and reason about the consequences of their actions in it to solve complex tasks. Behavior Cloning (BC) leverages task-specific human demonstrations to learn this knowledge as end-to-end policies. However, these policies are difficult to transfer to new tasks, and generating training data is challenging because it requires careful demonstrations and frequent environment resets. In contrast to such policy-based view, in this paper we take a model-based approach where we collect a few hours of unstructured easy-to-collect play data to learn an action-conditioned visual world model, a diffusion-based action sampler, and optionally a reward model. The world model -- in combination with the action sampler and a reward model -- is then used to optimize long sequences of actions with a Monte Carlo Tree Search (MCTS) planner. The resulting plans are executed on the robot via a zeroth-order Model Predictive Controller (MPC). We show that the action sampler mitigates hallucinations of the world model during planning and validate our approach on 3 real-world robotic tasks with varying levels of planning and modeling complexity. Our experiments support the hypothesis that planning leads to a significant improvement over BC baselines on a standard manipulation test environment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。