ๆญฃๅจๅ ่ฝฝ่ง้ข...
่ง้ขๅ ่ฝฝๅคฑ่ดฅ
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,574 ๆฌก่ง็ โข 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๐
