SuperMimic: Scaling Humanoid Whole-Body Control with One Differentiable World Model
AuthorsQiayuan Liao, Yiyang Shao, Junfeng Long, Kevin Zakka, Roberto Horowitz, Pieter Abbeel, Koushil Sreenath
AffiliationsUC Berkeley
SuperMimic uses a differentiable world model to train humanoid motion controllers far faster, and demonstrates the result on a real robot.
Key results
SuperMimic uses more than this many times fewer GPU-hours to surpass SONIC’s released checkpoint.
Shared benchmark used to compare motion-tracking failures.
Failures on the benchmark, compared with 179 for SONIC.
Fewer imagined steps needed by SuperMimic’s analytic-gradient updates.
What the paper found
SuperMimic tackles the compute cost of teaching humanoid robots to reproduce large libraries of human motion. Instead of relying on model-free reinforcement learning, it learns one differentiable world model from simulator transitions, then uses backpropagation through time over short imagined rollouts to give the tracking policy analytic gradients. The model and policy improve together, sharing the humanoid’s dynamics across motions. In 3 hours on one NVIDIA H100 GPU, SuperMimic surpasses the tracking return of SONIC, a baseline trained for 7 days on 128 GPUs, while using more than 1000× fewer GPU-hours. On a benchmark of 129,514 motions, it fails on 25 after 3 hours, compared with 179 failures for SONIC; after 9.8 hours on two GPUs, it has zero failures. The method also reaches PPO’s best return on world-model rollouts with about 64× fewer imagined steps, and its controller transfers to a physical Unitree G1 for motions including side rolls and flips. The results suggest a shared learned dynamics model can make scaling whole-body control far less dependent on large compute clusters.
Original abstract
Tracking large human-motion datasets has become the dominant route to general, expressive humanoid whole-body control, but scaling across many motions is expensive. Current pipelines use model-free reinforcement learning, either merging per-motion specialists through lossy distillation, or brute-forcing a single policy with cluster-scale compute. SuperMimic amortizes learning across all motions with one differentiable world model of their shared dynamics, fit from logged transitions of a standard rigid-body simulator. The world model and the tracking policy train together in one loop, and backpropagation-through-time over short imagined rollouts gives the policy analytic gradients. Fast imagined rollouts and analytic gradients make policy learning efficient. In 3 hours on a single GPU, SuperMimic surpasses the return of a state-of-the-art motion tracker baseline trained for 7 days on 128 GPUs. It reduces GPU-hours by more than $1000\times$ and fails on only 25 of 129,514 benchmark motions, versus 179 failures for the baseline. After 9.8 hours on two GPUs, SuperMimic achieves zero failures. From 1 to 264 hours of motion, the world model's prediction error stays stable, and GPU-hours to a target return grow slowly. Our policy transfers to a physical Unitree G1 across diverse and acrobatic motions. SuperMimic removes compute as the bottleneck to scaling humanoid whole-body control, making it practical to measure motion-tracking scaling laws.
Read the original paperMore in Embodied AI
Browse all 57 papers →Scaling Online Exploration for Robot Agents
Runhan Huang, Xiangcheng Zhang, Haoyu Chen, Yitong Li, Yen-Jen Wang, Bohan Lyu, Haotian Ye, Chuye Hong, Yixuan Wang, Chloe Su, Haoyu Xiong, Yangzhen Wu, Enze Xie, Travis Manderson, Stefano Ermon, Pieter Abbeel, Antonio Torralba, Dawn Song, Weirui Ye, Yilun Du
RAX lets coding agents explore, learn, and compose robot skills across different embodiments without relying on human demonstrations.
Self-Improving Video-to-Action Models
Hyunwoo Ryu, Eric Ming Chen, Max Wilcoxson, Vincent Sitzmann
Mirror helps embodied agents turn visual plans into actions by learning from their own attempts in the environment.
In-context Robot Learning Made Simple: A Democratized Recipe for Manipulation Tasks
Minxing Li, Minghao Han, Weizhi Zhao, Hanwen Wang, Xiangshuo Liu, Shuyao Shang, Jingxiang Zhou, Mingchao Sun, Hongyu Pan, Mu Xu, Yu Liu, Lue Fan, Zhaoxiang Zhang
SimpleICL presents an accessible recipe for teaching robots manipulation tasks from visual demonstrations without massive pretraining or specialized infrastructure.