MHER: Model-based Hindsight Experience Replay

Rui Yang, Meng Fang, Lei Han, Yali Du, Feng Luo, Xiu Li

Keywords: Multi-goal RL, model-based RL, hindsight experience replay, goal-conditioned supervised learning

TL;DR: We proposes MHER, a novel algorithm leveraging model to generate virtual achieved goals for both goal relabeling and policy improvement

Abstract: Solving multi-goal reinforcement learning (RL) problems with sparse rewards is generally challenging. Existing approaches have utilized goal relabeling on collected experiences to alleviate issues raised from sparse rewards. However, these methods are still limited in efficiency and cannot make full use of experiences. In this paper, we propose Model-based Hindsight Experience Replay (MHER), which exploits experiences more efficiently by leveraging environmental dynamics to generate virtual achieved goals. Replacing original goals with virtual goals generated from interaction with a trained dynamics model leads to a novel relabeling method, model-based relabeling (MBR). Based on MBR, MHER performs both reinforcement learning and supervised learning for efficient policy improvement. Theoretically, we also prove the supervised part in MHER, i.e., goal-conditioned supervised learning with MBR data, optimizes a lower bound on the multi-goal RL objective. Experimental results in several point-based tasks and simulated robotics environments show that MHER achieves significantly higher sample efficiency than previous model-free and model-based multi-goal methods.

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/arxiv:2107.00306/code)