RIDE: R EWARDING I MPACT -D RIVEN E XPLORATION FOR P ROCEDURALLY -G ENERATED E NVIRONMENTS
Anonymous authors
Paper under double-blind review
A
BSTRACT
Exploration in sparse reward environments remains one of the key challenges of model-free reinforcement learning (RL).
Instead of solely relying on extrinsic rewards provided by the environment, many state-of-the-art methods use intrinsic rewards to encourage the agent to explore the environment.
However, we show that existing methods fall short in procedurally-generated environments where an agent is unlikely to ever visit the same state more than once.
We propose a novel type of intrinsic exploration bonus which rewards the agent for actions that change the agent's learned state representation.
We evaluate our method on multiple challenging procedurally-generated tasks in MiniGrid, as well as on tasks used in prior curiosity-driven exploration work.
Our experiments demonstrate that our approach is more sample efﬁcient than existing exploration methods, particularly for procedurally-generated MiniGrid environments.
Furthermore, we analyze the learned behavior as well as the intrinsic reward received by our agent.
In contrast to previous approaches, our intrinsic reward does not diminish during the course of training and it rewards the agent substantially more for interacting with objects that it can control.
I NTRODUCTION
Deep reinforcement learning (RL) is one of the most popular frameworks for developing agents that can solve a wide range of complex tasks (Mnih et al., 2016; Silver et al., 2016; 2017).
RL agents learn to act in new environments through trial and error, in an attempt to maximize their cumulative reward.
However, many environments of interest, particularly those closer to real-world problems, do not provide a steady stream of rewards for agents to learn from.
In such settings, agents require many episodes to come across any reward, often rendering standard RL methods inapplicable.
Inspired by human learning, the use of intrinsic motivation has been proposed to encourage agents to learn about their environments even when extrinsic feedback is rarely provided (Schmidhuber, 1991b; 2010; Oudeyer et al., 2007; Oudeyer & Kaplan, 2009).
This type of exploration bonus emboldens the agent to visit new states (Bellemare et al., 2016; Burda et al., 2019b; Ecoffet et al., 2019) or to improve its knowledge and forward prediction of the world dynamics (Pathak et al., 2017; Burda et al., 2019a), and can be highly effective for learning in hard exploration games such as Montezuma's Revenge (Mnih et al., 2016).
However, most hard exploration environments used in previous work have either a limited state space or an easy way to measure similarity between states (Ecoffet et al., 2019) and generally use the same “ singleton ” environment for training and evaluation (Mnih et al., 2016; Burda et al., 2019a).
Deep RL agents trained in this way are prone to overﬁtting to a speciﬁc environment and often struggle to generalize to even slightly different environments (Rajeswaran et al., 2017; Zhang et al., 2018a;b).
As a ﬁrst step towards addressing this problem, a number of procedurally-generated environments have been recently released, for example DeepMind Lab
(Beattie et al., 2016), Sokoban (Racani`ere et al., 2017), Malm¨o (Johnson et al., 2016), CraftAssist (Jernite et al., 2019), Sonic (Nichol et al., 2018), CoinRun (Cobbe et al., 2019), Obstacle Tower (Juliani et al., 2019), or Capture the Flag (Jaderberg et al., 2019).
In this paper, we investigate exploration in procedurally-generated sparse-reward environments.
Throughout the paper, we will refer to the general problem that needs to be solved as a task ( e.g. ﬁnd a goal inside a maze) and to the particular instantiation of this task as environment ( e.g. maze layout, colors, textures, locations of the objects, environment dynamics etc.).
The environment can s t a t ˆ a t s t + φ t φ t + embed embed forward inverse ˆ φ t +
R IDE
L fw
L inv
R cur ψ t embed a t π be singleton or procedurally-generated .
Singleton environments are those in which the agent has to solve the same task in the same environment in every episode, i.e. ., the environment does not change between episodes.
A popular example of a hard exploration environment that falls into that category is Montezuma's Revenge.
In procedurally-generated environments, the agent needs to solve the same task, but in every episode the environment is initialized differently, making it unlikely to ever visit the same state twice.
Thus, agents in such environments have to learn policies that generalize well across a very large state space.
We demonstrate that current exploration methods fall short in such environments as they (i) make strong assumptions about the environment (deterministic or resettable to previous states)
(Ecoffet et al., 2019; Aytar et al., 2018), (ii) make strong assumptions about the state space (small number of different states or easy to determine if two states are similar)
(Ecoffet et al., 2019; Burda et al., 2019b; Bellemare et al., 2016;
Ostrovski et al., 2017; Machado et al., 2018a), or (iii) provide intrinsic rewards that can diminish quickly during training (Pathak et al., 2017; Burda et al., 2019a)..
To overcome these limitations, we propose Rewarding Impact-Driven Exploration (RIDE), a novel intrinsic reward for exploration in RL that encourges the agent to take actions which result in impactful changes to its representation of the environment state (see Figure 1 for an illustration).
We compare against state-of-the-art intrinsic reward methods on singleton environments with visual inputs, as well as on hard-exploration tasks in procedurally-generated grid-world environments.
Our experiments show that RIDE outperforms state-of-the-art exploration methods particularly in procedurally-generated environments.
Furthermore, we present a qualitative analysis demonstrating that RIDE, in contrast to prior work, does not suffer from diminishing intrinsic rewards during training and encourages the agent substantially more to interact with objects that it can control (relative to other state-action pairs).
R ELATED W ORK
The problem of exploration in reinforcement learning has been extensively studied.
Exploration methods encourage RL agents to visit novel states in various ways, for example by rewarding surprise (Schmidhuber, 1991b;a; 2010; 2006; Achiam & Sastry, 2017), information gain (Little & Sommer, 2013; Still & Precup, 2012; Houthooft et al., 2016), curiosity (Pathak et al., 2017; Burda et al., 2019b), empowerment (Klyubin et al., 2005; Rezende & Mohamed, 2015; Gregor et al., 2017), diversity (Eysenbach et al., 2019), feature control (Jaderberg et al., 2017; Dilokthanakul et al., 2019), or decision states (Goyal et al., 2019; Modhe et al., 2019).
Another class of exploration methods apply the Thompson sampling heurisitc (Osband et al., 2016; Ostrovski et al., 2017; O’Donoghue et al., 2018; Tang et al., 2017).
Osband et al.
(2016) use a family of randomized Q-functions trained on bootstrapped data to select actions, while Fortunato et al.
(2018) add noise in parameter space to encourage exploration.
Here, we focus on intrinsic motivation methods, which are widely-used and have proven effective for various hard-exploration tasks (Mnih et al., 2016; Pathak et al., 2017;
Bellemare et al., 2016; Burda et al., 2019b).
Intrinsic motivation can be useful in guiding the exploration of RL agents, particularly in environments where the extrinsic feedback is sparse or missing altogether (Oudeyer et al., 2007; 2008;
Oudeyer & Kaplan, 2009; Schmidhuber, 1991b; 2010).
The most popular and effective kinds of intrinsic motivation can be split into two broad classes: count-based methods that encourage the agent to visit new states and curiosity-based methods that encourage the agent to learn about the environment dynamics.
Count-Based Exploration.
Strehl & Littman (2008) proposed the use of state visitation counts as an exploration bonus in tabular settings.
More recently, such methods were extended to highdimensional state spaces (Bellemare et al., 2016; Ostrovski et al., 2017; Martin et al., 2017; Tang et al., 2017; Machado et al., 2018a).
Bellemare et al.
(2016) use a Context-Tree Switching (CTS) density model to derive a state pseudo-count, while Ostrovski et al.
(2017) use PixelCNN as a state density estimator.
Burda et al.
(2019b) employ the prediction error of a random network as exploration bonus with the aim of rewarding novel states more than previously seen ones.
However, one can expect count-based exploration methods to be less effective in procedurally-generated environments with sparse reward.
In these settings, the agent is likely to characterize two states as being different even when they only differ by features that are irrelevant for the task ( e.g. the texture of the walls).
If the agent considers most states to be “novel”, the feedback signal will not be distinctive or varied enough to guide the agent.
Curiosity-Driven Exploration.
Curiosity-based bonuses encourage the agent to explore the environment to learn about its dynamics.
Curiosity can be formulated as the error or uncertainty in predicting the consequences of the agent's actions (Stadie et al., 2015; Pathak et al., 2017; Burda et al., 2019b).
For example, Pathak et al.
(2017) learn a latent representation of the state and design an intrinsic reward based on the error of predicting the next state in the learned latent space.
While we use a similar mechanism for learning state embeddings, our exploration bonus is very different and builds upon the difference between the latent representations of two consecutive states.
As we will see in the following sections, one problem with their approach is that the intrinsic reward can vanish during training, leaving the agent with no incentive to further explore the environment and reducing its feedback to extrinsic reward only.
Generalization in Deep RL.
Most of the existing exploration methods that have achieved impressive results on difﬁcult tasks (Ecoffet et al., 2019; Pathak et al., 2017; Burda et al., 2019b; Bellemare et al., 2016; Choi et al., 2019; Aytar et al., 2018), have been trained and tested on the same environment and thus do not generalize to new instances.
Several recent papers (Rajeswaran et al., 2017;
Zhang et al., 2018a;b; Machado et al., 2018b; Foley et al., 2018) demonstrate that deep RL is susceptible to severe overﬁtting.
As a result, a number of benchmarks have been recently released for testing generalization in RL (Beattie et al., 2016; Cobbe et al., 2019; Packer et al., 2018;
Justesen et al., 2018; Leike et al., 2017; Nichol et al., 2018; Juliani et al., 2019).
Here, we make another step towards developing exploration methods that can generalize to unseen scenarios by evaluating them on procedurally-generated environments.
We opted for MiniGrid (Chevalier-Boisvert et al., 2018) because it is fast to run, provides a standard set of tasks with varied difﬁculty levels, focuses on single-agent, and does not use visual inputs, which allows us to better isolate the exploration problem.
More closely related to our work are the papers of Marino et al.
and Zhang et al.
Marino et al.
(2019) use a reward that encourages changing the values of the non-proprioceptive features for training low-level policies on locomotion tasks.
Their work assumes that the agent has access to a decomposition of the observation state into internal and external parts, an assumption which may not hold in many cases and may not be trivial to obtain even if it exists.
Zhang et al.
(2019) use the difference between the successor features of consecutive states as intrinsic reward.
In this framework, a state is characterized through the features of all its successor states.
While both of these papers use ﬁxed ( i.e. not learned) state representations to deﬁne the intrinsic reward, we use forward and inverse dynamics models to learn a state representation constrained to only capture elements in the environment that can be inﬂuenced by the agent.
Lesort et al.
(2018) emphasize the beneﬁts of using a learned state representation for control as opposed to a ﬁxed one, which may not contain information relevant for acting in the environment.
In the case of Zhang et al.
(2019), constructing a temporally extended state representation for aiding exploration is not trivial.
Such a feature space may add extra noise to the intrinsic reward due to the uncertainty in the states that might be visited in the future.
This is particularly problematic when the agent often encounters states that have never been visited before (as it is the case in procedurally-generated environments) or the environment is highly stochastic.
B ACKGROUND : C URIOSITY -D RIVEN E XPLORATION
We use the standard formalism of a single agent Markov Decision Process (MDP) deﬁned by a set of states S , a set of actions A , and a transition function T : S ×
A → P ( S ) which gives the probability distribution of the next state given a current state and action.
The agent chooses actions by sampling from a stochastic policy π : S → P ( A ) , and receives reward r : S ×
A → R at every time step.
The agent's goal is to learn a policy which maximizes its discounted expected return
R t =
E (cid:104)(cid:80) Tk =
γ k r t + k +
(cid:105) where r t is the sum of the intrinsic and extrinsic reward received by the agent at time t , γ ∈ [0 , 1] is the discount factor, and the expectation is taken with respect to both the policy and the environment.
Here, we consider the case of episodic RL in which the agent maximizes the reward received within a ﬁnite time horizon.
In this paper we consider that, along with the extrinsic reward r et , the agent also receives some intrinsic reward r it , which can be computed for any ( s t − 1 , a t − 1 , s t ) tuple.
Consequently, the agent tries to maximize the weighted sum of the intrinsic and extrinsic reward: r t = r
et + ω
ir r it where ω ir is a hyper-parameter to weight the importance of both rewards.
We built upon the work of Pathak et al.
(2017) who note that some parts of the observation may have no inﬂuence on the agent's state.
Thus, Pathak et al.
propose learning a state representation that disregards those parts of the observation and instead only models (i) the elements that the agent can control, as well as (ii) those that can affect the agent, even if the agent cannot have an effect on them.
Concretely, Pathak et al.
learn a state representations φ ( s ) =
f emb ( s ; θ emb ) of a state s using an inverse and a forward dynamics model (see Figure 1).
The forward dynamics model is a neural network parametrized by θ fw that takes as inputs φ ( s t ) and a t , predicts the next state representation: ˆ φ ( s t +1 )
=
f fw ( φ t , a t ; θ fw ) , and is trained to minimize L fw ( θ fw , θ emb ) =
(cid:107) ˆ φ ( s t +1 )
− φ ( s t +1 ) (cid:107) 22 .
The inverse dynamics model is also a neural network parametrized by θ inv that takes as inputs φ ( s t ) and φ ( s t +1 ) , predicts the agent's action: ˆ a t = f inv ( φ t , φ t
+1 ; θ inv ) , and is trained to minimize L inv ( θ inv , θ emb ) =
CrossEntropy (ˆ a t , a t ) when the action space is discrete.
Pathak et al.'s curiosity-based intrinsic reward is then deﬁned as the Euclidean distance between the actual embedding of the next state φ ( s t +1 ) and the one predicted by the forward model ˆ φ ( s t +1 ) .
I MPACT -D RIVEN E XPLORATION
Our main contribution is a novel intrinsic reward based on the change in the state representation produced by the agent's action.
The proposed method encourages the agent to try out actions that have a signiﬁcant impact on the environment and we demonstrate that it can promote effective exploration strategies when the feedback from the environment is sparse.
We train a forward and an inverse dynamics model to learn a latent state representation φ ( s ) as proposed by Pathak et al.
However, instead of using the Euclidean distance between the predicted next state representation and the actual next state representation as intrinsic reward ( R cur in Figure 1), we deﬁne impact-driven reward as the change between consecutive state representations ( R IDE in Figure 1).
Compared to curiosity-driven exploration, impact-driven exploration rewards the agent for very different state-actions, leading to distinct agent behaviors which we analyze in
Section 6.1.1.
Stanton & Clune (2018) categorize exploration into: across-training and intra-life and argue they are complementary.
Popular methods such as count-based exploration (Bellemare et al., 2016) encourage agents to visit new states in relation to all prior training episodes ( i.e. across-training novelty), but they do not consider whether an agent visits new states within some episode ( i.e. intra-life novelty).
As we will see, RIDE combines both types of exploration.
Formally, RIDE is computed as the L 2 -norm (cid:107) φ ( s t +1 ) − φ ( s t ) (cid:107) 2 of the difference in the learned state representation between consecutive states.
However, to ensure that the agent does not go back and forth between a sequence of states (with a large difference in their embeddings) in order to gain intrinsic reward, we discount RIDE by episodic state visitation counts.
Concretely, we divide the impact-driven reward by (cid:112) N ep ( s t +1 ) , where N ep ( s t +1 ) is the number of times that state has been visited during the current episode, which is initialized to 1 in the beginning of the episode.
In highdimensional regimes, one can use episodic pseudo-counts instead (Bellemare et al., 2016; Ostrovski et al., 2017).
Thus, the overall intrinsic reward provided by RIDE is calculated as:
[Equation]
where φ ( s t +1 ) and φ ( s t ) are the learned representations of consecutive states, resulted from the agent transitioning to state s t +1 after taking action a t in state s t .
The state is projected into a latent space using a neural network with parameters θ emb .
The overall optimization problem that is solved for training the agent is
[Equation]
where θ π are the parameters of the policy and value network ( a t ∼ π ( s t ; θ π ) ), and ω π , ω inv and ω fw are scalars that weigh the relative importance of the reinforcement learning (RL) loss to that of the inverse and forward dynamics losses which are used for learning the intrinsic reward signal.
Note that we never update the parameters of the inverse ( θ inv ), forward ( θ fw ), or embedding networks ( θ emb ) using the signal from the intrinsic or extrinsic reward ( i.e. the RL loss); we only use these learned state embeddings for constructing the exploration bonus and never as part of the agent's policy (Figure 1 highlights that the policy learns its own internal representation of the state ψ t , which is only used for control and never for computing the intrinsic reward).
Otherwise, the agent can artiﬁcially maximize its intrinsic reward by constructing state representations with large distances among themselves, without grounding them in observations of the environment.
Note that there is no incentive for the learned state representations to encode features of the environment that cannot be inﬂuenced by the agent's actions.
Thus, our agent will not receive rewards for reaching states that are inherently unpredictable, making exploration robust with respect to distractor objects or other inconsequential sources of variation in the environment.
As we will later show,
RIDE is robust to the well-known noisy-TV problem in which an agent rewarded for errors in the prediction of its forward model (such as the one proposed in Pathak et al.
(2017)) gets attracted to local sources of entropy in the environment.
Furthermore, the difference of consecutive state representations is unlikely to go to zero during learning as they are representations of actual states visited by the agent and constrained by the forward and inverse model.
This is in contrast to Pathak et al.
(2017) and Burda et al. (2019b) where the intrinsic reward goes to zero as soon as the forward model becomes sufﬁciently accurate or the agent's policy only explores well known parts of the state space.
E XPERIMENTS
We evaluate RIDE on procedurally-generated environments from MiniGrid, as well as two existing singleton environments used in prior work, and compare it against both standard RL and three commonly used intrinsic reward methods for exploration.
For all our experiments, we show the mean and standard deviation of the average return across 5 different seeds for each model.
The average return is computed as the rolling mean over the past 100 episodes.
5. E NVIRONMENTS
The ﬁrst set of environments are procedurally-generated gridworlds in MiniGrid (Chevalier-Boisvert et al., 2018).
We consider three types of hard exploration tasks: MultiRoomNXSY ,
KeyCorridorS3R3 , and ObstructedMaze2Dlh .
In MiniGrid, the world is a partially observable grid of size
NxN. Each tile in the grid contains at most one of the following objects: wall, door, key, ball, box and goal.
The agent can take one of seven actions: turn left or right, move forward, pick up or drop an object, toggle or done.
More details about the MiniGrid environment and tasks can be found in A.3.
For the sole purpose to compare in a fair way to the curiosity-driven exploration work by Pathak et al.
(2017), we ran a one-off experiment on their Mario (singleton) environment.
We train our model with and without extrinsic reward on the ﬁrst level of the game.
The last (singleton) environment we evaluate on is VizDoom (Kempka et al., 2016).
Details about the environment can be found in A.4.
5. B ASELINES
For all our experiments, we use IMPALA (Espeholt et al., 2018) as the base RL algorithm, and RMSProp (Tieleman & Hinton, 2012) for optimization.
All models use the same basic RL algorithm and network architecture for the policy and value functions (see Appendix A.2 and Appendix A. for details regarding the hyperparameters and network architectures), differing only in how intrinsic rewards are deﬁned.
In our experiments we compare with the following baselines:
Count : CountBased Exploration by Bellemare et al.
(2016) which uses state visitation counts to give higher rewards for new or rarely seen states.
RND : Random Network Distillation Exploration by Burda et al.
(2019b) which uses the prediction error of a random network as exploration bonus with the aim of rewarding novel states more than previously encountered ones.
ICM :
Intrinsic Curiosity Module by
Pathak et al.
(see Section 3).
IMPALA : Standard RL approach by Espeholt et al.
(2018) that uses only extrinsic reward and encourages random exploration by adding the entropy of the policy to the objective function.
R ESULTS AND D ISCUSSION
We present results of RIDE in comparison to popular exploration methods, as well as an analysis of the learned policies and properties of the intrinsic reward generated by different methods.
6. M INI G RID
Figure 3 summarizes our results on various hard MiniGrid tasks.
Note that the standard RL approach
IMPALA (purple) is not able to learn in any of the environments since extrinsic reward is too sparse.
Furthermore, our results reveal that RIDE is more sample efﬁcient compared to all the other exploration methods across all MiniGrid tasks considered here.
While other exploration bonuses seem effective on easier tasks and are able to learn optimal policies where IMPALA fails, the gap between our approach and the others is increasing with the difﬁculty of the task.
Furthermore, RIDE manages to solve some very challenging tasks on which the other methods fail to get any reward even after training on over 100M frames (Figure 3). [Figure]
[Table]
In addition to existing MiniGrid tasks, we also tested the ability of the models to deal with stochasticity in the environment by adding a “noisy TV” in the MiniGridN7S4 task, resulting in the new
MiniGirdN7S4NoisyTV task (left-center plot in the top row of Figure 3).
The noisy TV is implemented as a ball that changes its color to a randomly picked one whenever the agent takes a particular action.
As expected, the performance of ICM drops as the agent becomes attracted to the ball as it gets rewarded for not being able to predict the next color.
The Count model also needs more time to train, likely caused by the increasing number of rare and novel states (due to the changing color of the ball).
We include results for ablations to our model in Appendix A.5, highlighting the importance of combining impact-driven exploration with episodic state visitation discounting.
6.1. A NALYSIS OF THE I NTRINSIC R EWARD
To better understand the effectiveness of different exploration methods, we investigate the intrinsic reward an agent receives for certain trajectories in the environment.
Figure 4 shows a heatmap of the intrinsic reward received by RND, ICM, and RIDE on a sampled environment after having been trained on procedurally-generated environments from the MultiRoomN7S4 task.
While all three methods can solve this task, the intrinsic rewards received are different.
Speciﬁcally, the RIDE agent is rewarded in a much more structured manner for opening doors, entering new rooms and turning at decision points.
Table 1 provides quantitative numbers for this phenomenon.
We record the intrinsic rewards received for each type of action, averaged over 100 episodes.
We found that the RIDE is putting more emphasis on actions interacting with the door than for moving forward or turning left or right, while the other methods reward actions more uniformly. [Figure]
Figure 12 and Table 3 in A.6.2 show a similar pattern in the intrinsic rewards for agents trained on the MultiRoomN12S10 task, while Figure 13 and Table 4 in A.6.3 contain the equivalent analysis for agents trained on ObstructedMaze2Dlh.
As emphasized there, RIDE is rewarded more for interactions with objects relative to actions for moving around in the maze, a characteristic which is not as prevalent in the other models.
Figure 5 shows the mean intrinsic reward of all models while training on the MultiRoomN12S10 task.
While the ICM, RND, and Count intrinsic reward converges to very low values quite early in the training process, the RIDE bonus keeps changing and has a higher value even after training on 100m frames.
Hence, RIDE constantly encourages the agent to take actions that change the local environment.
In contrast, Count, RND, and
Curiosity may not consider certain states to be “novel” or “surprising” after longer periods of training as they have seen similar states in the past or learned to almost perfectly predict the next state in a subset of the environment states.
Consequently, their intrinsic rewards diminish during training, so the agent will have a difﬁcult time distinguishing between actions that lead to novel or surprising states and those that do not and it may get trapped in some parts of the state space (see Figure 12).
6.1. S INGLETON VERSUS P ROCEDURALLY -G ENERATED E NVIRONMENTS
It is important to understand and quantify how much harder it is to train existing deep RL exploration methods on tasks in procedurally-generated environments compared to a singleton environment.
To investigate this dependency, we trained the models on a singleton environment of the the ObstructedMaze2Dlh task so that at the beginning of every episode, the agent is spawned in exactly the same maze with all objects located in the same positions.
In this setting, we see that Count, RND, and IMPALA are also able to solve the task (see Figure 7 and compare with the center-right plot in the bottom row of Figure 3 for procedurally-generated environments of the same task).
As expected, this emphasizes that training an agent in procedurally-generated environments creates signiﬁcant challenges over training on a singleton environment for the same task.
Moreover, it highlights the importance of training on a variety of environments to avoid overﬁtting to the idiosyncrasies of a particular environment.
6.1. N
[Figure]
E XTRINSIC R EWARD
To analyze the way different methods explore environments without depending on the chance of running into extrinsic reward (which can dramatically change the agent's policy), we analyze agents that are trained without any extrinsic reward on both singleton and procedurally-generated environments.
The top row of Figure 6 shows state visitation heatmaps for all the models in a singleton environment on the MultiRoomN10S6 task, after training all of them for 50m frames with intrinsic reward only.
The agents are allowed to takesteps in every episode.
The ﬁgure indicates that all models have effective exploration strategies when trained on a singleton maze, the 10th, 9th and 6th rooms are reached by RIDE,
Count/RND, and ICM, respectively.
The Random policy fully explores the ﬁrst room but does not get to the second room within the time limit.
When trained on procedurally-generated mazes, existing models are exploring much less efﬁciently as can be seen in the bottom row of Figure 6.
Here, Count, RND, and ICM only make it to the 4th, 3rd and 2nd rooms respectively within an episode, while RIDE is able to explore all rooms.
This further supports that RIDE learns a state representation that allows generalization across different mazes and is not as distracted by less important details that change from one procedurally-generated environment to another.
6. M ARIO AND V IZDOOM
Our results on the ﬁrst level of the Mario environment in order to compare to Pathak et al.
(see Figure 8a and b) suggest that this environment may not be as challenging as previously believed, given that all the methods evaluated here, including vanilla IMPALA, can learn similarly good policies after training on only 1m frames even without any intrinsic reward (left ﬁgure).
Note that we are able to reproduce the results mentioned in the original ICM paper (Pathak et al., 2017).
However, when training with both intrinsic and extrinsic reward (center ﬁgure), the curiosity-based exploration bonus (ICM) hurts learning, converging later and to a lower value than the other methods evaluated here.
For VizDoom (see Figure 8c) we observe that RIDE performs as well as ICM, while all the other baselines fail to learn an effective policy given the same amount of training.
Note that our ICM implementation can reproduce the results in the original paper on this task, achieving a 100% success rate after training on approximately 60m frames (Pathak et al., 2017).
C ONCLUSION AND F UTURE W ORK
In this work, we propose rewarding impact-driven exploration (RIDE), an intrinsic reward bonus that encourages the agent to explore actions that substantially change the state of the environment, as measured in a learned latent space.
RIDE has a number of desirable properties: it attracts the agent to states where it can affect the environment, it provides a signal to the agent even after training for a long time, and it is conceptually simple as well as compatible with other intrinsic or extrinsic rewards and any deep RL algorithm.
Our approach is particularly effective in procedurally-generated sparse-reward environments where it signiﬁcantly outperforms IMPALA (Espeholt et al., 2018), as well as some of the most popular exploration methods: Count (Bellemare et al., 2016), RND (Burda et al., 2019b),
ICM (Pathak et al., 2017).
Furthermore, RIDE explores procedurally-generated environments more efﬁciently than the other exploration methods.
However, there are still many ways to improve upon RIDE.
For example, one can make use of symbolic information to measure or characterize the agent's impact, consider longer-term effects of the agent's actions, or promote diversity among the kinds of changes the agent makes to the environment.
Another interesting avenue for future research is to develop algorithms that can distinguish between desirable and undesirable types of impact the agent can have in the environment, thus constraining the agent to act safely and avoid distractions ( i.e. actions that lead to large changes in the environment but that are not useful for a given task).
Such different kinds of impact might correspond to distinctive skills or low-level policies that a hierarchical controller can use to learn more complex policies or better exploration strategies.
A A PPENDIX A. N ETWORK A RCHITECTURES
All our models use the same network architecture for the policy and value networks.
The input is passed through a sequence of three (for MiniGrid) or four (environments used by Pathak et al.
(2017)) convolutional layers with 32 ﬁlters each, kernel size of 3x3, stride of 2 and padding of 1.
An exponential linear unit (ELU; (Clevert et al.
(2016))) is used after each convolution layer.
The output of the last convolution layer is fed into a LSTM with 256 units.
Two separate fully connected layers are used to predict the value function and the action from the LSTM feature representation.
For the singleton environments used in prior work, the agents are trained using visual inputs that are pre-processed similarly to Mnih et al.
The RGB images are converted into gray-scale and re-sized to 42 42.
The input given to both the policy and the state representation networks consists of the current frame concatenated with the previous three frames.
In order to reduce overﬁtting, during training, we use action repeat of four.
At inference time, we sample the policy without any action repeats.
A. H YPERPARAMETERS
We ran grid searches over the learning rate ∈ [0 . 0001 , 0 . 0005 , 0 . 001] , batch size ∈ [8 , 32] and unroll length ∈ [20 , 40 , 100 , 200] .
The best values for all models can be found in Table 2.
[Table]
The learning rate is linearly annealed to 0 in all experiments.
We also ran grid searches over the intrinsic reward coefﬁcient ∈ [1 . 0 , 0 . 5 , 0 .
001] for all the models and found that the best values were 0.1 for RIDE, ICM and RND, and 0.05 for Count.
We also ran grid searches over the entropy coefﬁcient ∈ [0 . 01 , 0 .
for all the models and found that the best values were 0.0005 for RIDE and 0.0001 for Count, RND, and ICM.
In all experiments presented here, we use the best values found for each model.
A. M INI G RID E NVIRONMENT
In MiniGrid, the world is a partially observable grid of size NxN. Each tile in the grid contains exactly zero or one object.
The possible object types are wall, door, key, ball, box and goal.
Each object in MiniGrid has an associated discrete color, which can be one of red, green, blue, purple, yellow and grey.
By default, walls are always grey and goal squares are always green.
Rewards are sparse for all MiniGrid environments.
There are seven actions in MiniGrid: turn left, turn right, move forward, pick up an object, drop an object, toggle and done.
The agent can use the turn left and turn right action to rotate and face one of 4 possible directions (north, south, east, west).
The move forward action makes the agent move from its current tile onto the tile in the direction it is currently facing, provided there is nothing on that tile, or that the tile contains an open door.
The agent can open doors if they are right in front of it by using the toggle action.
Observations in MiniGrid are partial and egocentric.
By default, the agent sees a square of 7x7 tiles in the direction it is facing.
These include the tile the agent is standing on.
The agent cannot see through walls or closed doors.
The observations are provided as a tensor of shape 7x7x3.
However, note that these are not RGB images.
Each tile is encoded using 3 integer values: one describing the type of object contained in the cell, one describing its color, and a ﬂag indicating whether doors are open or closed.
This compact encoding was chosen for space efﬁciency and to enable faster training.
For all tasks, , the agent gets an egocentric view of its surroundings, consisting of 33 pixels.
A neural network parameterized as a CNN is used to process the visual observation.
The MultiRoomNXSY environment consists of X rooms, with size at most Y , connected in random orientations.
The agent is placed in the ﬁrst room and must navigate to a green goal square in the most distant room from the agent.
The agent receives an egocentric view of its surrounding, consisting of 33 pixels.
The task increases in difﬁculty with X and Y. Episodes ﬁnish with a positive reward when the agent reaches the green goal square.
Otherwise, episodes are terminated with zero reward after a maximum of 20xN steps.
In the KeyCorridorS3R3 environment, the agent has to pick up an object which is behind a locked door.
The key is hidden in another room, and the agent has to explore the environment to ﬁnd it.
Episodes ﬁnish with a positive reward when the agent picks up the ball behind the locked door or after a maximum of 270 steps.
In the ObstructedMaze2Dlh environment, the agent has to pick up a box which is placed in a corner of a 3x3 maze.
The doors are locked, the keys are hidden in boxes and doors are obstructed by balls.
Episodes ﬁnish with a positive reward when the agent picks up the ball behind the locked door or after a maximum of 576 steps.
In the DynamicObstacles environment, the agent has to navigate to a ﬁxed goal location while avoiding moving obstacles.
In our experiments, the agent is randomly initialized to a location in the grid.
If the agent collides with an obstacles, it receives a penalty of -1 and the episode ends.
A. V IZ D OOM E NVIRONMENT
We consider the Doom 3-D navigation task where the action space of the agent consists of four discrete actions: move forward, move left, move right and no-action.
Our testing setup in all the experiments is the DoomMyWayHome-v0 environments which is available as part of OpenAI Gym (Brockman et al., 2016).
Episodes are terminated either when the agent ﬁnds the vest or if the agent exceeds a maximum of 2100 time steps.
The map consists of 9 rooms connected by corridors and the agent is tasked to reach some ﬁxed goal location from its spawning location.
The agent is always spawned in Room-13 which is 270 steps away from the goal under an optimal policy.
A long sequence of action sis required to reach the goals from these rooms, making this setting a hardexploration problem.
The agent is only provided a sparse terminal reward of +1 if it ﬁnds the vest and 0 otherwise.
While this environment has sparse reward, it is not procedurally generated, so the agent ﬁnds itself in exactly the same environment in each episode and does not need to generalize to different environment instantiations.
This environment is identical to the ”sparse” setting used in
Pathak et al.
A. A BLATIONS
In this section, we aim to better understand the effect of using episodic discounting as part of the intrinsic reward, as well as that of using entropy regularization as part of the IMPALA loss.
Figure 9 compares the performance of our model on different MiniGrid tasks with that of three ablations.
The ﬁrst one only uses episodic state counts as exploration bonus without multiplying it by the impact-driven intrinsic reward ( OnlyEpisodicCounts ), the second one only uses the impact-driven exploration bonus without multiplying it by the episodic state count term ( NoEpisodicCounts ), while the third one is the NoEpisodicCounts model without the entropy regularization term in the IMPALA loss ( NoEntropyNoEpisodicCounts ).
OnlyEpisodicCounts does not solve any of the tasks. NoEntropyNoEpisodicCounts either converges to a suboptimal policy or completely fails. In contrast, NoEpisodicCounts can solve the easier tasks but it requires more interactions than RIDE and fails to learn on the hardest domain. During training, NoEpisodicCounts can get stuck cycling between two states (with a large distance in the embedding states) but due to entropy regularization, it can sometimes escape such local optima
(unlike NoEntropyNoEpisodicCounts ) if it ﬁnds extrinsic reward. However, when the reward is too sparse, NoEpisodicCounts is insufﬁcient while RIDE still succeeds, indicating the effectiveness of augmenting the impact-driven intrinsic reward with the episodic count term.
Figure 10 shows the average number of states visited during an episode of MultiRoomN12S10, measured at different training stages for our full RIDE model and the NoEpisodicCounts ablation.
While the NoEpisodicCounts ablation always visits a low number of different states each episode ( (cid:18) 10 ), RIDE visits an increasing number of states throughout training (converging to ∼ 100 for an optimal policy). Hence, it can be inferred that NoEpisodicCounts revisits some of the states. This claim can be further veriﬁed by visualizing the agents’ behaviors.
After training, NoEpisodicCounts goes back and forth between two states, while RIDE visits each state once on its path to the goal.
Consistent with our intuition, discounting the intrinsic reward by the episodic state-count term does help to avoid this failure mode.
A. A NALYSIS A.6. S TATE V ISITATION IN M ULTI R OOM N12S
In this section, we analyze the behavior learned by the agents.
Figure 11 shows the state visitation heatmaps for all models trained on 100m frames of MultiRoomN12S10, which has a very sparse reward.
Note that while our model has already reached the goal in the farthest room of the maze,
Count has explored about half of the maze, while RND and ICM are still in the ﬁrst two rooms
. A.6. I NTRINSIC R EWARD IN M ULTI R OOM N12S
Figure 12 shows a heatmap of the intrinsic reward received by RIDE, RND, and ICM agents trained on the procedurally-generated MultiRoomN12S10 environment.
Table 3 shows the corresponding intrinsic rewards received for each type of action, averaged over 100 episodes, for the trained models.
This environment is very challenging since the chance of randomly stumbling upon extrinsic reward [Figure] [Figure] [Table]
[Figure]
[Table]
[Figure] [Table] is extremely low. Thus, we see that while the intrinsic reward provided by RIDE is still effective at exploring the maze and ﬁnding extrinsic reward
[Figure]
the exploration bonuses used by RND and ICM are less useful, leading to agents that do not go beyond the second room, even after training on 100m frames.
[Table]
Open
[Table]
A.6. I NTRINSIC R EWARD IN O BSTRUCTED M AZE 2D LH
In order to understand how various interactions with objects are rewarded by the different exploration methods
[Table]
we also looked at the intrinsic reward in the ObstructedMaze2Dlh environment which contains multiple objects .
However, the rooms are connected by locked doors and the keys for unlocking the doors are hidden inside boxes.
The agent does not know in which room the ball is located and it needs the color of the key to match that of the door in order to open it.
Moreover, the agent cannot hold more than one object so it needs to drop one in order to pick up another.
Figure 13 and Table 4 indicate that RIDE rewards the agent signiﬁcantly for interacting with various objects (e.g. opening the box, picking up the key, opening the door, dropping the key, picking up the ball) relative to other actions such as moving forward or turning left and right.
In contrast, RND again rewards all actions much more uniformly and often times, within an episode, it rewards the interactions with objects less than the ones for moving around inside the maze. A. D YNAMIC O BSTACLES E NVIRONMENT
One potential limitation of RIDE is that it may be drawn to take actions that signiﬁcantly change the environment, even when those actions are undesirable.
In order to test the limits of RIDE, we ran experiments on the DynamicObstacles environment in MiniGrid.
As seen in Figure 14, RIDE learns to solve the task of avoiding the moving obstacles in the environment, even if chasing them provides large intrinsic rewards.
Hence, RIDE is still able to learn effectively in certain scenarios in which high-impact actions are detrimental to solving the task.
A. G ENERALIZATION TO U NSEEN C OLORS
We also ran experiments on a version of MultiRoomN7S4 in which the colors of the walls and the goal change at each episode.
The models are trained on a set of 4 colors and tested on a held-out set of 2 colors.
As seen in Figure 15 and Table 5, RIDE and Count learn to solve this task and can generalize to unseen colors at test time without any extra ﬁne-tuning.
RND and ICM perform slightly worse on the test environments, and only one out of ﬁve seeds of ICM converges to the optimal policy on the train environments.
The best seed for each model was used to evaluate on the test set.
A. O THER P RACTICAL I NSIGHTS
While developing this work, we also experimented with a few other variations of RIDE that did not work.
First, we tried to use observations instead of learned state embeddings for computing the
RIDE reward, but this was not able to solve any of the tasks.
Using a common state representation for both the policy and the embeddings also fell short.
[Table]