The Dreamer world model
Danijar Hafner, Timothy Lillicrap, Jimmy Ba and Mohammad Norouzi presented the "Dreamer" algorithm at ICLR 2020. This RL agent is able to plan future actions based on those performed via latent imagination. Recently, the algorithm outperformed pure reinforcement learning in video games from brief interactions in a world model. Researchers at UC Berkeley wanted to know if Dreamer could facilitate faster learning on physical robots.Learning without a simulator
Danijar Hafner, who presented Dreamer in 2020, is also part of the UC Berkeley team. An AI PhD student in Toronto, he is currently in the lab of Pieter Abbeel, another team member, as well as Philipp Wu, Alejandro Escontrela and Ken Goldberg. The study uses recent developments in the Dreamer world model to train a range of robots through online reinforcement learning in the real world, without a simulator or demonstration.
The figure above illustrates how Dreamer builds a world model from a replay buffer of previous experiences, learns behaviors from imagined deployments in the latent space of the world model, and continuously interacts with the environment to explore and refine its behaviors.
The team's goal is to push the boundaries of real-world robot learning and provide a robust platform for future research that will demonstrate the benefits of world models for such learning.
The researchers applied Dreamer to 4 robots, demonstrating that real-world learning was possible without introducing new algorithms. The tasks were varied, taking place in different action spaces, sensory modalities and reward structures:
- They trained a quadruped to roll on its back, stand up and walk in just one hour. They then pushed the robot and found that Dreamer adapts within 10 minutes to resist or quickly roll over and stand up.
- Visual Pick and Place (above): On two different robotic arms, Dreamer learns to pick and place multiple objects directly from camera images and scattered rewards, approaching human performance.
- On a wheeled robot, Dreamer learns to navigate to a goal solely from camera images, automatically resolving ambiguity about the robot's orientation.
Our articles will then appear first in Google Top Stories.