解耦视频世界模型加速机器人规划,质量不减

解耦视频世界模型加速机器人规划,质量不减

Ziyu Shan, Zhenyu Wu, Xiaofeng Wang, Zheng Zhu, Ziwei Wang

1 分钟阅读•2026年7月1日

南洋理工大学的研究人员开发了DVG-WM,这是一种世界模型,通过巧妙的双阶段方法生成机器人操作任务的未来视频帧:快速低分辨率动态模型加上轻量级细化步骤。该设计相比现有基于视频的世界模型,将规划时间降低了一个数量级,使实时视频预测更接近在实体机器人上的实际应用。

目录

研究人员构建了什么

DVG-WM(解耦视频生成世界模型)是一个神经网络,给定当前摄像头图像和机器人动作指令,预测机器人手臂在操作任务中接下来会看到什么。与以往每步从头生成全分辨率帧的视频预测模型不同——这需要数十次昂贵的去噪步骤——DVG-WM将任务拆分为两个专门的模块。

第一个模块处理低分辨率视频帧,对场景的物理动态建模:物体移动方向、夹爪如何旋转、发生什么碰撞。它运行50步去噪,但分辨率大幅降低(例如64×36而非256×144),因此速度极快。第二个模块将这些粗略的动态预测“细化”为清晰的高分辨率图像,仅需4步额外的去噪。由于视觉细节与物理动态被解耦,模型可以各自以最优计算成本处理。

结果是一个世界模型,运行速度比之前的最先进方法(如LVP潜在视频预测)快数倍,同时保持相当或更好的预测质量。DVG-WM在合成模拟数据集和一个包含9000多条抓取、倾倒、堆叠轨迹的真实世界机器人操作数据集上进行了训练。

视觉对比图,展示DVG-WM预测与真实抓取序列

关键成果

DVG-WM在CALVIN基准(标准模拟机器人操作环境)和真实数据集(使用Franka Emika Panda机械臂)上进行了评估。关键指标是预测质量(通过FID和LPIPS分数衡量)和计算时间(每帧毫秒计算)。

与强基线LVP相比,DVG-WM实现了显著加速——作者报告从50步全分辨率去噪减少到仅4步,而50步被移至低分辨率。这使得推理时间大幅降低,足以在单个GPU上实现迭代规划。在CALVIN基准上,DVG-WM在视觉保真度(FID和LPIPS)上与LVP相当或略优,同时每个规划步的实际时间快5倍以上。

当作为模型预测控制的世界模型使用时,DVG-WM在多个操作场景中提高了任务成功率——特别是需要精确交互的如插销任务——因为快速推理允许在相同时间预算内评估更多候选动作序列。在真实数据集上,该模型泛化到未见过的物体排列和光照条件,展现出超越训练分布的鲁棒性。

工作原理

DVG-WM基于扩散视频生成骨干网络,但带有一个关键的架构创新:动态与视觉合成的解耦训练。

训练期间,低分辨率动态模块(“速度”路径)学习预测下采样视频序列在机器人动作条件下的演化。细化模块(“质量”路径)学习将这些低分辨率预测映射到高分辨率图像,专注于纹理和边缘细节,忽略场景动态。

在规划推理期间(例如在MPC中),机器人控制器采样多个动作序列,对每个序列展开世界模型,并对结果评分。由于DVG-WM运行足够快,能在每个规划周期内生成数十条未来轨迹,控制器可以选择更可能成功的动作。

用于真实世界评估的数据集预处理:所有视频调整到统一分辨率(可能是256×144),舍弃少于49帧的轨迹以确保一致的预测范围。训练集包含9000多条真实世界操作轨迹,涵盖抓取、倾倒、堆叠和钻孔任务。

组件分辨率去噪步数目的
动态模块64×36(低)50对机器人、物体与环境之间的物理交互建模
细化模块256×144(全)4添加视觉纹理、锐化和全分辨率细节

解耦设计是效率的关键:昂贵的50步扩散过程在一个小图像网格(64×36仅包含256×144图像约9%的像素)上运行,计算成本低廉。细化模块只需4步,因为粗略的动态输出已经包含正确的物体位置和轮廓——只需添加高频视觉信息(纹理、阴影)。

架构图,展示两阶段流程:低分辨率动态后接细化

为何对机器人技术重要

生成视觉预测的世界模型是机器人规划、模拟和强化学习的强大工具。但直到现在,它们对于实际硬件的闭环控制来说太慢了——每帧视频预测可能需要数秒。DVG-WM将延迟降低到适合在单个GPU上进行迭代MPC的范围,这是在工业机器人生产中部署视频预测世界模型的重要一步。

对于需要适应变化环境(如不可预测的货架布局、光照变化)的仓库机器人,快速世界模型使得无需预先模拟即可在线重新规划。同样,协作机器人在装配线上可以使用DVG-WM在碰撞风险或抓取失败发生前进行预测,从而提高安全性并减少停机时间。

解耦原则——将动态与视觉质量分离——也是其他需要速度和视觉丰富度的机器人感知任务的蓝图。随着边缘GPU性能增强,像DVG-WM这样的模型可以在机器人控制器上本地运行,实现无需依赖云的实时视觉预见。

局限性与开放问题

DVG-WM在较短范围的任务(最多49帧,约3-4秒视频)上进行了评估。对于更长任务或完整回合,重复预测的误差累积可能会降低质量。该模型还假设静态背景,不模拟可变光照动态或人类交互——真实工厂可能更混乱。

另一个开放问题:解耦设计如何迁移到不同摄像头视角或机械臂配置?细化模块在特定分辨率比上训练;适应不同传感器分辨率可能需要重新训练。最后,速度提升依赖于固定的降采样比——速度与质量之间的最优权衡可能因任务和硬件而异。

常见问题

DVG-WM解决了什么问题? 现有的视频预测世界模型对于实时机器人规划太慢,因为它们从头生成全分辨率帧。DVG-WM将过程拆分为低分辨率下的快速动态模块和轻量级细化步骤,实现了显著加速。

DVG-WM比之前的方法快多少? 在CALVIN基准上,DVG-WM运行速度比LVP基线快5倍以上,同时保持相当的视觉预测质量,这得益于将大部分去噪步骤转移到小尺寸低分辨率图像上。

DVG-WM能在真实机器人上工作吗? 是的。该模型在使用Franka Emika Panda机械臂收集的9000多条真实世界操作轨迹数据集上进行了训练和评估,涵盖抓取、倾倒、堆叠和钻孔任务。

速度背后的关键创新是什么? 不是用一个昂贵的扩散模型处理所有事情,DVG-WM将动态与视觉解耦。它在64×36分辨率下运行50步去噪来模拟运动,然后仅在全分辨率下运行4步来增加清晰度。

结论

DVG-WM证明,通过将问题分离为动态预测和视觉细化,可以实现快速、高质量的视频世界模型用于机器人操作。报告的速度提升使视频驱动的规划更接近在实体机器人上的实际部署,且不牺牲预测精度。未来的工作可能会将方法扩展到更长范围、动态背景和多摄像头设置。

如果DVG-WM能在机器人的板载GPU上运行,你最希望它首先应用于哪个操作任务?