Загрузка видео...
Не удалось загрузить видео
Introducing 𝐃𝐫𝐞𝐚𝐦𝐆𝐞𝐧! We got humanoid robots to perform totally new 𝑣𝑒𝑟𝑏𝑠 in new environments through video world models. We believe video world models will solve the data problem in robotics. Bringing the paradigm of scaling human hours to GPU hours. Quick 🧵
117,493 просмотров • 1 год назад •via X (Twitter)
Комментарии: 11

Currently, robot data scaling is done through human labor. Recent work showed some potential signs of robots doing useful things in unseen homes (i.e., open-world generalization), but this required taking the physical robots and collecting data in 100+ homes.

How about new 𝑣𝑒𝑟𝑏𝑠? When prompted, current robots stand still or perform tasks they were trained on (e.g. pick-and-place). There is currently no work in literature that can enable robots to perform new verbs outside of the teleoperation data for visuomotor robot policies.

We introduce 𝐃𝐫𝐞𝐚𝐦𝐆𝐞𝐧, an embarrassingly simple 4-step pipeline that generates synthetic robot training data using video world models. (1) Fine-tune, (2) Prompt, (3) Extract actions, (4) Train. Simple as that.

Through DreamGen, we generate “Dreams” or 𝑁𝑒𝑢𝑟𝑎𝑙 𝑇𝑟𝑎𝑗𝑒𝑐𝑡𝑜𝑟𝑖𝑒𝑠 of 22 new verbs in 10 unseen environments, and train robots to perform these tasks "zero-shot".

DreamGen can also augment seen, contact-rich tasks that are hard to simulate (e.g. folding, scooping m&ms) for different robot systems (Franka & SO-100) and different robot policies (Diffusion Policy, π₀, GR00T N1), all resulting in non-trivial gains.

We also introduce 𝘋𝑟𝘦𝑎𝘮𝐺𝘦𝑛 𝐵𝘦𝑛𝘤ℎ, a video generative benchmark for robotics that shows a positive correlation with downstream robot policies, so that video model researchers can help enable robotics without actually having to set up their own physical robot systems.

This was an 8-month research effort, jointly led by @SeonghyeonYe @zy27962986 @szxiangjn, and advised by @scott_e_reed @yukez @DrJimFan , and with amazing collaborators from Nvidia GEAR Lab, Nvidia Cosmos Team, and UW. Check out more videos in the blog post & details in the paper! 🌐 Blog post: 📝 Paper: We will also be releasing the code & an API to try out DreamGen (prompt -> lerobot dataset) in the upcoming days! Do reach out if you are interested in joining our GR00T Dreams Team. We are scaling up the data, GPUs, and the team!

Discover the future of AI investing. AIS delivers exposure to the companies driving the next wave of innovation—semiconductors, data centers, and AI applications. Explore the supercycle today.

@threadreaderapp unroll

So cool to see this! Congrats 🔥

Congrats Joel! Doing big things👀
