arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2505.12705cs.ROcs.AIcs.LG

DreamGen:通过视频世界模型解锁机器人学习中的泛化能力

DreamGen: Unlocking Generalization in Robot Learning through Video World Models

NVIDIA;University of Washington(华盛顿大学);KAIST(韩国科学技术院);UCLA(加州大学洛杉矶分校);UCSD(加州大学圣地亚哥分校);CalTech(加州理工学院);NTU(国立台湾大学);University of Maryland(马里兰大学);UT Austin(得克萨斯大学奥斯汀分校)
Joel Jang, Seonghyeon Ye, Zongyu Lin, Jiannan Xiang, Johan Bjorck, Yu Fang, Fengyuan Hu, Spencer Huang, Kaushil Kundalia, Yen-Chen Lin, Loic Magne, Ajay Mandlek… 展开作者

Joel Jang, Seonghyeon Ye, Zongyu Lin, Jiannan Xiang, Johan Bjorck, Yu Fang, Fengyuan Hu, Spencer Huang, Kaushil Kundalia, Yen-Chen Lin, Loic Magne, Ajay Mandlekar, Avnish Narayan, You Liang Tan, Guanzhi Wang, Jing Wang, Qi Wang, Yinzhen Xu, Xiaohui Zeng, Kaiyuan Zheng, Ruijie Zheng, Ming-Yu Liu, Luke Zettlemoyer, Dieter Fox, Jan Kautz, Scott Reed, Yuke Zhu, Linxi Fan

AI总结:

DreamGen提出一种4阶段流水线,利用视频世界模型生成合成数据训练机器人策略,实现人形机器人22种新行为的环境与行为泛化,并引入DreamGen Bench验证性能。

摘要:

我们引入了DreamGen,这是一个简单但高效的4阶段流水线,用于训练机器人策略,使其通过神经轨迹(从视频世界模型生成的合成机器人数据)在不同行为和环境中实现泛化。DreamGen利用最先进的图像到视频生成模型,将其适配到目标机器人本体,以生成在不同环境中熟悉或新颖任务的照片级逼真合成视频。由于这些模型仅生成视频,我们使用潜在动作模型或逆动力学模型(IDM)恢复伪动作序列。尽管其简单,DreamGen实现了强大的行为和环境泛化:一个人形机器人能够在已见和未见的环境中执行22种新行为,而仅需在一个环境中单个拾取-放置任务的遥操作数据。为了系统评估该流水线,我们引入了DreamGen Bench,一个视频生成基准,显示基准性能与下游策略成功之间的强相关性。我们的工作为扩展机器人学习超越手动数据收集建立了一个有前景的新方向。代码可在https://github.com/NVIDIA/GR00T-Dreams获取。

英文摘要:

We introduce DreamGen, a simple yet highly effective 4-stage pipeline for training robot policies that generalize across behaviors and environments through neural trajectories - synthetic robot data generated from video world models. DreamGen leverages state-of-the-art image-to-video generative models, adapting them to the target robot embodiment to produce photorealistic synthetic videos of familiar or novel tasks in diverse environments. Since these models generate only videos, we recover pseudo-action sequences using either a latent action model or an inverse-dynamics model (IDM). Despite its simplicity, DreamGen unlocks strong behavior and environment generalization: a humanoid robot can perform 22 new behaviors in both seen and unseen environments, while requiring teleoperation data from only a single pick-and-place task in one environment. To evaluate the pipeline systematically, we introduce DreamGen Bench, a video generation benchmark that shows a strong correlation between benchmark performance and downstream policy success. Our work establishes a promising new axis for scaling robot learning well beyond manual data collection. Code available at https://github.com/NVIDIA/GR00T-Dreams.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑