一个统一模型同时预测动作与视频,速度快且精度高。
Unified Video Action Model
- 用联合隐空间建模视频与动作关系,解耦生成过程提升效率。
- 推理时跳过视频生成,动作预测速度更快,准确率接近专用模型。
- 单个模型可完成策略学习、动力学建模等多任务,适合机器人通用场景。
统一视频与动作模型在机器人领域具有重要意义:视频提供场景信息以预测动作,动作则提供动态信息以生成视频。然而,如何有效结合视频生成与动作预测仍具挑战,现有基于视频生成的方法在动作准确率和推理速度上难以媲美直接策略学习。为此,我们提出统一视频动作模型(UVA),通过联合优化视频与动作预测,实现高精度与高效推理。核心在于学习视频-动作联合隐表示,并解耦视频-动作解码过程。联合隐表示连接视觉与动作域,有效建模序列间关系;解耦解码借助两个轻量级扩散头,推理时跳过视频生成,显著加速动作预测。该统一框架支持掩码输入训练,通过选择性遮蔽视频或动作,单一模型可完成策略学习、前向/逆向动力学建模及视频生成等多样化任务。大量实验表明,UVA能作为通用解决方案,覆盖多种机器人任务,性能不逊于专用方法。
原文摘要 · Abstract (English)
A unified video and action model holds significant promise for robotics, where videos provide rich scene information for action prediction, and actions provide dynamics information for video prediction. However, effectively combining video generation and action prediction remains challenging, and current video generation-based methods struggle to match the performance of direct policy learning in action accuracy and inference speed. To bridge this gap, we introduce the Unified Video Action model (UVA), which jointly optimizes video and action predictions to achieve both high accuracy and efficient action inference. The key lies in learning a joint video-action latent representation and decoupling video-action decoding. The joint latent representation bridges the visual and action domains, effectively modeling the relationship between video and action sequences. Meanwhile, the decoupled decoding, powered by two lightweight diffusion heads, enables high-speed action inference by bypassing video generation during inference. Such a unified framework further enables versatile functionality through masked input training. By selectively masking actions or videos, a single model can tackle diverse tasks beyond policy learning, such as forward and inverse dynamics modeling and video generation. Via an extensive set of experiments, we demonstrate that UVA can serve as a general-purpose solution for a wide range of robotics tasks, such as policy learning, forward/inverse dynamics and video observation prediction, without compromising performance compared to methods tailored for specific applications. Results are best viewed on https://unified-video-action-model.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。