来源笔记

RoboWorld: Fast and Reliable Neural Simulators for Generalist Robot Policy Evaluation

Robot Policy EvaluationVideo World ModelsVision Language ActionGeneralist Robot PolicyNeural Simulation

RoboWorld 通过在一个快速的动作条件视频世界模型中运行通用机器人策略,并用任务进度 VLM 评分标准评估生成的 rollout,来评估通用机器人策略。它报告称,在不进行实体 rollout 的情况下,评估结果与真实 RoboArena 策略排名高度一致。

  • 通用机器人策略需要在不同任务、物体和环境中进行大量试验,这使真实机器人评估变慢且成本高。
  • 手工构建的模拟器需要资产和环境工程,其 sim-to-real 差距可能改变策略排名。
  • 视频世界模型可以生成策略 rollout,但长时域伪影、缓慢的去噪过程和二元 VLM 评分可能把模型错误变成错误的策略失败判断。
  • 该系统在 DROID 上训练一个自回归视频世界模型,初始化自 Wan2.1-T2V-1.3B,并使用因果注意力、动作交叉注意力和逐帧噪声调度。
  • Step Forcing 使用与推理阶段相同的少步去噪调度来训练模型,因此 4 步 rollout 条件与训练条件更接近。
  • Step Forcing 创建一步自前推先验,让模型在训练中接触自身不完美的上下文,同时不需要运行完整的顺序 rollout。
  • 锚定步骤混入带噪声的真实上下文,使动作-观测动态保持与数据相连,而不是由模型诱导状态主导。
  • RoboWorld 以闭环方式运行策略:策略动作、生成的下一步多视角观测、再次生成策略动作。VLM 裁判默认使用 GPT-4o,给出 0-5 的任务进度分数,并使用腕部视角检测世界模型伪影。
  • 在 RoboArena 上,RoboWorld 用 4,186 个生成的 rollout 评估了 8 个开源策略,并与真实世界排行榜匹配,Pearson r=0.989,Spearman rho=0.970。
  • 完整的 RoboArena 风格神经评估耗时 100 个 H100 GPU 小时,包括策略动作生成和世界模型 rollout。
  • 任务进度 VLM 评分标准相对于 RoboArena 排名达到 Spearman rho=0.970;改用二元成功评分后,rho 降至 0.922。
  • 在 BAIR Robot Pushing 上,使用 4 个去噪步骤的 Step Forcing 达到 SSIM 0.8063 ID 和 0.7374 OOD,LPIPS 0.0525 ID 和 0.0768 OOD。使用 8 个步骤的 Teacher Forcing 达到 SSIM 0.7942 ID 和 0.7118 OOD,LPIPS 0.0554 ID 和 0.1058 OOD。
  • 在 300 帧、20 秒的 RoboArena 视频生成中,自回归 Step Forcing 运行速度为 15.31 FPS;4 步双向基线为 5.70 FPS。论文报告称,该方法取得总体最佳 LPIPS,并在 4 步方法中取得最佳 SSIM 和 FVD。
  • 在 DROID 腕部视角消融中,完整方法的 FVD 为 231.0。移除自前推先验后,FVD 升至 258.5;移除锚定步骤后,FVD 升至 294.0。