Video yükleniyor...
Video Yüklenemedi
🥳Excited to share: Hierarchical World Models as Visual Whole-Body Humanoid Controllers Joint work with Jyothir S V Vlad Sobal Yann LeCun Xiaolong Wang Hao Su Our method, Puppeteer, learns high-dim humanoid policies that look natural, in an entirely data-driven way! 🧵👇(1/n)
252,074 görüntüleme • 2 yıl önce •via X (Twitter)
10 Yorum

For example: our method achieves similar reward to our prev SOTA method TD-MPC2 on these challenging tasks, while producing motions that are broadly preferred by humans. Here's an example of TD-MPC2 learning to "hack" the simulator! (2/n)

We propose a benchmark for visual whole-body control with a 56-dim humanoid, and show that our method consistently matches TD-MPC2 while generating natural behaviors. Similar to findings by @carlo_sferrazza @pabbeel et al, we see that SAC + DreamerV3 struggle with humanoids (3/n)

How our method works: we first pretrain a "tracking" world model (TD-MPC2) on retargeted MoCap data, and then train a visual "puppeteer" (also TD-MPC2) that controls the tracking world model via low-dim 3D joint targets. No reward design, skill primitives, or other tricks! (4/n)

In a user study (n=46) we ask participants to compare the "naturalness" of our method vs. vanilla TD-MPC2. Humans strongly prefer our method that is pretrained with human data vs. a pure RL approach, even when they achieve the same reward! (5/n)

This was a super fun project to work on, and aligns really well with our joint vision of future AI systems: world models that are (1) hierarchical, (2) non-generative, and (3) use planning for decision-making. Check out paper + code + models at ✨ (6/n)

Both method and environment code is available at so if you're interested in RL or humanoids be sure to check it out 🤖 It takes ~4 days to train a hierarchical world model with visual inputs on a single RTX 3090 GPU. (7/n)

@jyothir_s_v @vlad_is_ai @ylecun @xiaolonw @haosu_twitr Awesome results Nicklas!

@jyothir_s_v @vlad_is_ai @ylecun @xiaolonw @haosu_twitr Thank you Dhanush! Appreciate it

@jyothir_s_v @vlad_is_ai @ylecun @xiaolonw @haosu_twitr Very exciting! Hoping to see the paper. But what's the motivation to use such a model-based RL method for this task?

@jyothir_s_v @vlad_is_ai @ylecun @xiaolonw @haosu_twitr Why RL: it is a very flexible framework for control that is not dependent on e.g. expert demos for each of the tasks that we want our robot to do. Why specifically TD-MPC2 for the hierarchy: we tried a few algos and it was the only one that could handle the high dim actions :-)

