Model Imitation for Model-Based Reinforcement Learning

  • 2020-03-16 05:47:10
  • Yueh-Hua Wu, Ting-Han Fan, Peter J. Ramadge, Hao Su
  • 0


Model-based reinforcement learning (MBRL) aims to learn a dynamic model toreduce the number of interactions with real-world environments. However, due toestimation error, rollouts in the learned model, especially those of longhorizons, fail to match the ones in real-world environments. This mismatchinghas seriously impacted the sample complexity of MBRL. The phenomenon can beattributed to the fact that previous works employ supervised learning to learnthe one-step transition models, which has inherent difficulty ensuring thematching of distributions from multi-step rollouts. Based on the claim, wepropose to learn the transition model by matching the distributions ofmulti-step rollouts sampled from the transition model and the real ones viaWGAN. We theoretically show that matching the two can minimize the differenceof cumulative rewards between the real transition and the learned one. Ourexperiments also show that the proposed Model Imitation method can compete oroutperform the state-of-the-art in terms of sample complexity and averagereturn.


Quick Read (beta)

This feature is not avaialbe for this paper.