DreamControl-v2: Simpler and Scalable Autonomous Humanoid Skills via Trainable Guided Diffusion Priors

Abstract

Developing robust autonomous loco-manipulation skills for humanoids remains an open problem in robotics. While RL has been applied successfully to legged locomotion, applying it to complex, interaction-rich manipulation tasks is harder given long-horizon planning challenges for manipulation. A recent approach along these lines is DreamControl, which addresses these issues by leveraging off-the-shelf human motion diffusion models as a generative prior to guide RL policies during training. In this paper, we investigate the impact of DreamControl’s motion prior and propose an improved framework that trains a guided diffusion model directly in the humanoid robot’s motion space, aggregating diverse human and robot datasets into a unified embodiment space. We demonstrate that our approach captures a wider range of skills due to the larger training data mixture and establishes a more automated pipeline by removing the need for manual filtering interventions. Furthermore, we show that scaling the generation of reference trajectories is important for achieving robust downstream RL policies. We validate our approach through extensive experiments in simulation and on a real Unitree-G1.

Autonomous Real World Deployment

Squatting

Opening a drawer

Deep squat and pick

Picking an object

Picking an object

Pouring

Punching

Wiping surface

Generated Samples from DreamControl-v2

SpatialTemporal Conditioned Generation

A person walks forward

a person lifts a box from the ground with both hands

a person opens a drawer

a person punches with their right hand

a person punches with their right hand

a person Runs on the spot

Text only Conditioned generation

a person takes a forward step

a person jumps

a person does jumping jacks

a person balances on one leg

a person moves side to side

a person does a squat

a person steps back

a person stands in T-pose

Diverse Samples across tasks

a person opens a drawer

a person punches with their right hand

a person does jumping jacks

a person balances on one leg

a person raises both hands

a person moves side to side

a person raises a single hand

a person does a squat

a person walks forward

Method

Method Overview