In plain words: It learns a summary of images that makes the robot's motion and costs easy to predict, so a classic linear control trick can plan from pixels. On robot tasks, including a real arm, it beat model-based methods while learning more efficiently than model-free ones.
Abstract · SOLAR: Deep Structured Representations for Model-Based Reinforcement Learning
Model-based reinforcement learning (RL) has proven to be a data efficient approach for learning control tasks but is difficult to utilize in domains with complex observations such as images. In this paper, we present a method for learning representations that are suitable for iterative model-based policy improvement, even when the underlying dynamical system has complex dynamics and image observations, in that these representations are optimized for inferring simple dynamics and cost models given data from the current policy. This enables a model-based RL method based on the linear-quadratic regulator (LQR) to be used for systems with image observations. We evaluate our approach on a range of robotics tasks, including manipulation with a real-world robotic arm directly from images. We find that our method produces substantially better final performance than other model-based RL methods while being significantly more efficient than model-free RL.
Marvin Zhang, Sharad Vikram, Laura Smith, Pieter Abbeel, Matthew J. Johnson, Sergey Levine
arXiv:1808.09105 · cs.LG, cs.RO, stat.ML · submitted Aug 28, 2018 · updated Jun 22, 2019
abstract · pdf · html · ICML 2019. Project website: https://sites.google.com/view/icml19solar