用少量带动作标签视频,让闭源视频扩散模型学会预测动作后果。
AVID: Adapting Video Diffusion Models to World Models
- 通过训练可学习掩码,修改预训练模型中间输出实现动作条件生成。
- 在游戏和机器人数据上优于现有适配方法,生成视频更准确。
- 适合缺乏标注数据的机器人决策场景,推动具身智能发展。
大规模生成模型在多个领域取得显著成功。然而,对于机器人等序列决策问题,动作标注数据往往稀缺,因此扩展基础模型用于决策仍具挑战。一种潜在解决方案是利用广泛存在的未标注视频训练世界模型,以模拟动作的后果。若世界模型准确,可用于下游任务优化决策。图像到视频的扩散模型已能生成高度逼真的合成视频,但这些模型不支持动作条件,且最强大的模型为闭源,无法微调。本文提出AVID,无需访问预训练模型参数,即可将预训练视频扩散模型适配为动作条件世界模型。AVID在少量特定领域的动作标注视频数据上训练适配器,利用可学习掩码修改预训练模型的中间输出,生成准确的动作条件视频。我们在视频游戏和真实世界机器人数据上评估AVID,结果表明其在扩散模型适配方面优于现有基线。结果表明,若正确使用,预训练视频模型可成为具身智能的强大工具。
原文摘要 · Abstract (English)
Large-scale generative models have achieved remarkable success in a number of domains. However, for sequential decision-making problems, such as robotics, action-labelled data is often scarce and therefore scaling-up foundation models for decision-making remains a challenge. A potential solution lies in leveraging widely-available unlabelled videos to train world models that simulate the consequences of actions. If the world model is accurate, it can be used to optimize decision-making in downstream tasks. Image-to-video diffusion models are already capable of generating highly realistic synthetic videos. However, these models are not action-conditioned, and the most powerful models are closed-source which means they cannot be finetuned. In this work, we propose to adapt pretrained video diffusion models to action-conditioned world models, without access to the parameters of the pretrained model. Our approach, AVID, trains an adapter on a small domain-specific dataset of action-labelled videos. AVID uses a learned mask to modify the intermediate outputs of the pretrained model and generate accurate action-conditioned videos. We evaluate AVID on video game and real-world robotics data, and show that it outperforms existing baselines for diffusion model adaptation.1 Our results demonstrate that if utilized correctly, pretrained video models have the potential to be powerful tools for embodied AI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。