Title: EVENT DISCOVERY FOR HISTORY REPRESENTATION IN REINFORCEMENT LEARNING

Abstract: Environments in Reinforcement Learning (RL) are usually only partially observable. To address this problem, a possible solution is to provide the agent with information about past observations. While common methods represent this history using a Recurrent Neural Network (RNN), in this paper we propose an alternative representation which is based on the record of the past events observed in a given episode. Inspired by human memory, these events describe only important changes in the environment and, in our approach, are automatically discovered using self-supervision. We evaluate our history representation method using two challenging RL benchmarks: some games of the Atari-57 suite and the 3D environment Obstacle Tower. Using these benchmarks we show the advantage of our solution with respect to common approaches.

1. INTRODUCTION:

Deep Reinforcement Learning (RL) algorithms have been successfully applied to a range of challenging domains, from computer games (Mnih et al., 2013) to robotic control (OpenAI et al., 2018) . These approaches use a Neural Network (NN) both to represent the current observation of the environment and to learn the agent's optimal policy, used to choose the next action. For instance, the state observation can be the current game frame or an image of the robot camera, and a Convolutional Neural Network (CNN) may be used to obtain a compact feature vector from it.

However, often RL environments are only partially observable and having a significant representation of the past may be crucial for the agent (Kapturowski et al., 2019) . For instance, in the Atari 2600 Pong game, the state can be represented as two consecutive frames. In this way, the agent can determine both the position of the ball and the direction of its movement. More complex partially observable domains require a longer state history input to the agent. For instance, when navigating inside a 1-st person-view 3D labyrinth, the agent obtains little information from the current scene observation and needs several previous frames to localise itself.

A common solution to represent the observation history is based on the use of Recurrent Neural Networks (RNNs), where the RNN hidden-layer activation vector is input to the agent, possibly together with the current state observation (Mnih et al., 2016) . However, RL is characterized by highly nonstationary data, which makes training unstable (Schulman et al., 2016) and this instability is exacerbated when a recurrent network needs to be simultaneously trained to extract the agent's input representation. We show in Sec. 5.2 that in some common cases an RNN-based history representation struggles to improve the agent's results over an input representation composed of only the instantaneous observations. In this paper we propose a different direction, in which the agent's observation history is represented using a set of discrete events, which describe important changes in the state of the world (Orr et al., 2018) . Environment-specific events are automatically discovered during training by clustering past observations and are then used as landmarks in our history representation. Intuitively, this is inspired by the common human behaviour: when making decisions, humans do not keep detailed visual information of the previous steps. For instance, while navigating through the halls of a building, it is sufficient to recall a few significant landmarks seen during the walk, e.g. specific doors or furniture.

Following this idea, in this paper we propose an Event Discovery History Representation (EDHR) approach, composed of 3 iterative stages: experience collection, policy optimisation and event discovery. We discover events by clustering past state observations. In more detail, we maximize the Mutual Information (MI) between the latent representations of temporally-close frames to cluster the frame semantics. This is a form of self-supervision, in which no additional annotation is needed for the event discovery: the higher the predictability of one frame with respect to the other, the larger the semantic information shared by the two frames (van den Oord et al., 2018; Hjelm et al., 2019; Anand et al., 2019; Ji et al., 2019) .

Once clusters have been formed, the probability distribution of a given frame F t for the set of current clusters is used as the semantic representation of the state observation at time t and is recorded in a longer history. Finally, the history is input to the agent together with an instantaneous observation representation, obtained, following (Mnih et al., 2013) , as a stack of the last 4 frames. This information is now used for policy optimisation. Note that our proposed history representation is independent of the specific RL approach. However, in all our experiments we use the PPO algorithm (Schulman et al., 2017) for the policy and the value function optimisation. The 3 stages are iterated during training, and thus past clusters can be modified and adapted to address new observations, while the agent is progressing through the environment.

We summarize the contribution of this paper below. First, we use a modified version of the Invariant Information Clustering (IIC) algorithm (Ji et al., 2019) to discover significant events in the agent's past observations. Second, we propose to replace common RNN-based history representations with a time-dependent probability distribution of the observations for the events so far discovered. We evaluate our history representation (EDHR) using the PPO algorithm on several environments of the Atari-57 benchmark (Bellemare et al., 2012) and on the 3D environment Obstacle Tower (Juliani et al., 2019) , showing that it provides a significant boost with respect to the most common input representation methods. Specifically, we show that EDHR outperforms plain PPO in cases, where history can increase observability, and it outperforms RNN-based methods in several common cases, while simultaneously requiring 2× less wall-clock time for training.

The source code of the method and all the experiments is publicly available at the anonymous link: https://github.com/iclr2020anon/EDHR and will be published after acceptance.

2. RELATED WORK:

Partially Observable Markov Decision Processes. We consider a setting in which an agent interacts with an environment to maximize a cumulative reward signal. The interaction process is defined as a Partially Observable Markov Decision Process (POMDP) < S, A, T, R, Ω, O, γ > (Monahan, 1982; Kaelbling et al., 1998) , where S is a finite state space, A is a finite action space, R(s, a) is the reward function, T (•|s, a) is the transition function, Ω is a set of possible observations, O is a function which maps states to probability distributions over observations and γ ∈ [0, 1) is a discount factor used to compute the cumulative discounted reward. Within this framework, at each step t, the environment is in some unobserved state s ∈ S. When the agent takes an action a, the environment transits to the new state s with probability T (s |s, a), providing the agent with a new observation o ∈ Ω with probability O(o |s ) and reward r ∼ R(s, a).

POMDPs are an active research area and several approaches have been recently proposed to address partial observability. DQN (Mnih et al., 2013) is one of the earliest deep RL models directly trained using the high-dimensional observation o t as input (i.e., the raw pixels of the current frame). To deal with partial observability, the input of the model is composed of a stack of the 4 last grayscale frames, processed as 4 channels using a 2D CNN. We also use this observation representation which we call instantaneous to highlight the difference from the past state history. Hausknecht & Stone (2015) replace the fully-connected layer in DQN with an LSTM (Hochreiter & Schmidhuber, 1997) layer. At each step, the model receives only one frame of the environment together with the LSTM hidden-layer value which represents the past. The idea was extended in (Igl et al., 2018) by inferring latent belief states. To simulate reduced observability and demonstrate the robustness of the method, Hausknecht & Stone (2015) introduce the flickering Atari games, obscuring the observation with a certain probability at each step. These methods focus on shortterm partial observability, such as noisy observations, and approximate the state with an implicitly inferred continuous history. In contrast, our method explicitly represents each step of the past using a dictionary of discovered events, and concatenates a sequence of steps in an event-based history.

Our final goal is to augment the current state representation with important past changes of the environment.

Recurrent layers (e.g. LSTM or GRU (Cho et al., 2014) ), are a common choice for modern RL architectures. The A3C algorithm (Mnih et al., 2016) showed a high mean performance with an LSTM layer. R2D2 (Kapturowski et al., 2019) is specifically focused on processing long-history sequences (80 steps with additional 40 burn-in steps) using an LSTM. However, despite its widespread, RNNs have disadvantages within an RL framework because they increase the training instability (Sec. 1). In this paper we propose an alternative history representation and we empirically show in Sec. 5.2 that our proposal outperforms RNN-based solutions in several common cases.

Unsupervised and Self-supervised Learning. Unsupervised and self-supervised learning aim to extract useful representations from data (e.g. images, videos) without annotation and reward signal. While common approaches are covered in recent surveys (Jing & Tian, 2019; Kolesnikov et al., 2019) , below we analyse the most relevant methods to this work.

A first class of unsupervised feature-extraction methods is based on a generative encoding-decoding scheme. The methods typically minimize the reconstruction error in the pixel space, thus the representation is encouraged to capture pixel-level details, which may differ from semantics. Ha & Schmidhuber (2018) combine Variational Autoencoders (Kingma & Welling, 2014) (VAEs) with RNNs in their World Models. A VAE is trained to reconstruct the current frame and simultaneously produce a latent, compressed frame representation which is input to an RNN predicting the next latent feature vector. The input of the RL agent consists of the latent vector of the current observation and the hidden state of the RNN.

An alternative to a reconstruction loss in the input space is clustering in the feature space. Deep-Cluster (Caron et al., 2018) iteratively alternates between k-means clustering of the CNN features and updating the network's parameters with the pseudo-labels obtained by means of clustering.

Another important self-supervised paradigm is based on maximizing the Mutual Information (MI) between latent representations. Typically these methods rely on collecting pairs of positive and negative samples. Two positive samples should share the same semantics, while negatives should be perceptually different. A discriminator is trained to distinguish between positive and negative pairs, so computing a lower-bound on the MI of the learned latent representations. In CPC (van den Oord et al., 2018) and ST-DIM (Anand et al., 2019) positives are temporally close frames, while in IIC (Ji et al., 2019) simple data-augmentation techniques are applied on still images. In DIM (Hjelm et al., 2019) this idea is extended including locality, and MI is computed also between global and local representations of the same image. DIM was applied to several games of the Atari 2600 suite (Bellemare et al., 2012) in ST-DIM, however, the reported evaluation is based only the disentanglement degree of the obtained representations and does not consider the possible benefit for an RL task.

Our event discovery stage (Sec. 3) is based on IIC (Ji et al., 2019 ), but we use as positives temporallyclose frames extracted from an agent trajectory instead of data-augmentation. Note that the main difference between our proposal and IIC is that our goal is not clustering, but clusters are used as a set of discrete events to represent the agent's history, in contrast to RNN and other continuous history representations.

3. EVENT DISCOVERY AND HISTORY REPRESENTATION:

Event discovery stage. As mentioned in Sec. 1-2, we use IIC (Ji et al., 2019) for our unsupervised event discovery. While we refer the reader to the Appendix C and the original paper for the details, we show below how we use this clustering method to obtain a set of discrete events and simultaneously learn to encode our observations. Let o t be the observation of the environment at time step t, presented as an image. We want to learn a representation mapping Φ : Ω → [0, 1] C , where Y = {1, ..., C} is a set of C events and Φ(o t ) is the probability distribution of a discrete random variable z which (soft-)assigns o t to the elements in Y and can be defined as: Φ c (o t ) = P (z = c|o t ) (Ji et al., 2019) . Φ(•) is our unsupervised encoder and it is based on a CNN followed by a fully-connected layer and a softmax layer (Fig. 1 ), parametrized by φ. The clustering process in IIC is based on pairs of images randomly transformed which contain History Representation and Policy Optimisation. Once φ is updated (see Sec. 4), we use Φ(o t ; φ) to represent an observation at time t and update the agent's policy. In more detail, we represent the agent with an Actor-Critic architecture approximated with a NN π parametrised by θ. We use the PPO algorithm to optimise the policy and the value function. At time step t, the agent receives two types of data as input, an instantaneous representation I(t) consisting of 4 concatenated raw-pixel, grayscale frames We empirically observed that the history matrix H(t) is very sparse (Sec. 5.3), being most of the elements in Φ(o t ) close to 0 (Φ(o t ) is usually very peaked). Thus, an even more compact history representation may be produced. Anyway, with a small value of S (S = 32 in all our experiments) and a small number of clusters C (we use C = 16), H(t) is a low-dimensional, sparse matrix which represents significant information about the recent agent's past trajectory. Importantly, in contrast with an RNN-based history representation, which may be prone to forgetting past information, in our proposed history representation H(t), all the past S events are stored, together with their timedependent evolution.

I(t) = {o t , o t-1 , o t-2 , o t-

I(t) is input to a second CNN (Ψ) which extracts a vector v t representing the instantaneous visual information (v t is obtained by flattening the last CNN feature map). We call the parameters of Ψ θ I , to differentiate them from the agent's proper parameters θ A which control the policy and the value function (θ = [θ I ; θ A ]). Finally, v t is concatenated with the (flattened) matrix H(t) and input to the agent. Fig. 2 shows the full architecture of our agent. We highlight that the CNN of the unsupervised encoder (Φ) is different from Ψ, and the two CNNs do not share parameters (φ = θ I ). Moreover, while the encoder is trained in an unsupervised fashion, Ψ, used to extract visual information (v t ) from I(t), is trained using the RL reward. This emphasizes the complementary between the knowledge of the two networks, being the former based on clustering visual similarities in data which do not depend on the specific RL task, while the latter being trained as in a standard RL framework.

4. TRAINING:

We run the agent in several parallel environments for a predefined number of steps (rollout size r).

The actions are chosen according to the current policy and encoder (observation collection stage).

Next, using the collected experience, we update θ for n P epochs (policy optimisation stage) and φ for n E epochs (event discovery stage). The second and the third stages (described in Sec. 3) are each other independent and can be performed in parallel, which removes the event discovery runtime overhead. These three stages are iterated, so to adapt Φ(•) (and Y ) to the observation distribution change. The full algorithm is described in Alg. 1.

4.1. IMPLEMENTATION DETAILS:

The weights φ and θ are initialized with a (semi-) orthogonal matrix, as described in (Saxe et al., 2014) , apart from the head of Φ (i.e., the fully-connected layer on top of the CNN, see Fig. 1 ), which is initialized randomly. All the biases are initialized with zeros.

Following (Ji et al., 2019) , to improve robustness, the head of Φ is duplicated 5 times with different initial random parameters. Moreover, Φ includes an auxiliary overclustering head with C × 5 clusters. Each head produces a different representation with a potentially different degree of utility for the agent. A possible solution is the simple concatenation of the 5 different event probability distributions Φ 1 (o), ...Φ 5 (o). However, this solution would increase the dimensionality of H(t).

We instead select the "best" head using the MI value as the selection criterion as follows. We split training in two phases. In the first phase (N * p pretrain iterations of the outmost loop in Alg. 1) the agent receives only I(t) as input and θ and φ (and each head in Φ) are updated as usual (lines 8 and 9 in Alg. 1). Then, the head corresponding to the highest MI value (averaged over the last iterations) is selected and kept fixed during the second phase, in which the agent receives the full input (H(t) included).

In all our experiments we use the same set of hyperparameters and the same network architecture. We adopt the training procedure and the hyperparameters in (Schulman et al., 2017) , including the number of time steps equal to 10M. Our agent network architecture is similar to (Schulman et al., 2017) , except from the additional history input (H(t)). Φ is composed of a CNN and a fullyconnected layer (Sec. 3). Specifically, its CNN configuration is similar to Ψ, except the first layer, which takes only one channel as input. In Appendix A we list all other hyperparameters.

5. EXPERIMENTAL EVALUATION:

We evaluate EDHR on two challenging deep RL benchmarks: the Arcade Learning Environment (Bellemare et al., 2012) (ALE) and the Obstacle Tower (Juliani et al., 2019) . Moreover, we compare the results of EDHR with PPO and PPO-RNN, where the latter is described below.

PPO-RNN. Ψ is the same as in EDHR and in PPO (Schulman et al., 2017) . However, past observations are represented as follows. The encoder Φ is replaced with a similar network (Υ) but without the softmax layer and having 256 neurons as output of the fully-connected layer instead of C. For each t ∈ {t, ..., t -S + 1}, ω t = Υ(o t ) is input to a GRU (Cho et al., 2014) with a hidden layer of size 512 (re-initiated with zeros each S steps). Finally, the hidden layer activation vector represents the history and replaces H(t) in Fig. 2 , keeping all the rest of the agent unchanged. Note that no self-supervision is used here to train the network components.

5.1. ENVIRONMENTS:

In ALE we use 7 classic Atari 2600 video games: MsPacman, SpaceInvaders, Breakout, Gravitar, Qbert, Seaquest and Enduro. Empirically, balanced complexity and diverse game dynamics make this subset of the benchmark representative for comparison. In all these environments, o t is an RGB image. The action space A is discrete and the number of actions varies in each game from 4 to 18. Following closely the experimental protocol in (Schulman et al., 2017) , we apply the same observation pre-processing (frame downscaling, etc.).

Obstacle Tower is a 3D 3-rd person game environment. It consists of a number of floors, each floor being composed of several rooms. The layout is procedurally generated based on an environment seed. The rooms can contain puzzles to solve, the complexity of which increases with the floor number, creating a natural curriculum learning for the agent. The environment has high visual fidelity, realistic lighting, a variety of textures and geometry, physics-driven interactions. Possible actions include movements in four directions, camera rotations and jumps. The agent receives a reward of +1.0 after passing each floor and an auxiliary reward of +0.1 after solving an intermediate puzzle.

An observation o t is an 84 × 84 RGB image, converted to grayscale. The action space is discrete, with 54 actions, which are reduced to 12 after removing redundancy. This environment is especially challenging because of several reasons. Unlike the Atari games with simple sprite-based graphics, this environment has diverse and realistic observations. The sparse reward signal complicates training and favours a low-dimensional agent's input. The agent is trained using environment seeds 0-7 and it is evaluated with seeds 50-57, which shows the generalisation abilities of the algorithm (see the Train and Test rows in Tab. 2, respectively).

5.2. RESULTS:

In all the experiments we report the final scores averaged over 100 episodes. Each experiment is performed 3 times with a different random seed. We report the average value and the standard deviation. Tab. 1-2 show the results. "PPO (Sch.)" refers to the results reported in (Schulman et al., 2017) , while the results "PPO" are based on our reproduction of that algorithm. We compare the results with our reproduction to remove the impact of minor technical implementation details, but in Tab. 1 we also report the original results "PPO (Sch.)" as a reference. "PPO-RNN" is described above.

These results show the benefit of our history representation which is able to significantly boost the scores in most of the games, although more reactive games, like Breakout, do not show any benefit. PPO-RNN underperforms our EDHR on ALE benchmark. Note that, in most of the games, PPO-RNN has a similar or even worse performance of plain PPO. We believe this is due to the difficulty of training an RNN simultaneously with the agent in an RL framework, especially when the training budget is limited. Specifically, the training budget, adopted in all our experiments and uniformly with all the tested methods, consists of 10M of time steps (Sec. 4.1), which is equivalent to the number of observations o t an agent receives from the environment, considering all the parallel actors and all the iterations. Moreover, if we consider the wall-clock time (and with the same training budget), in all our experiments PPO-RNN is on average about two times slower then our EDHR. In Appendix B we include additional experiments with different hyperparameters of EDHR, an ablation model without self-supervision and another common RNN-based model.

The high score of PPO-RNN in Tab. 2 shows the importance of past information in the Obstacle Tower environment, where instantaneous observations are not sufficiently informative. EDHR successfully exploits the discovered events outperforming PPO. However, in this game our method underperforms PPO-RNN, indicating that in some cases the compact history representation can loose useful information.

The higher efficiency of our EDHR with respect to the training budget is shown in Fig. 3 , where, for each game, we plot the scores reached by each method over the training iterations. These plots show that EDHR effectively makes use of the knowledge acquired in the unsupervised training stages and that, using an event-based history representation, most of the times training is easier. We believe this is due to the fact that the agent, in making a decision, can use the events, recorded in H(t), as a sort of "summary" of the important information in a past trajectory.

5.3. DISCOVERED EVENT ANALYSIS:

What type of events does the method discover? To answer this question, we visually examined the observations and the corresponding events of a trained agent in two environments, MsPacman and Obstacle Tower. Fig. 4 shows how the events are distributed during an episode of MsPacman. Specifically, for each time step t, we plot the predominant event c * t corresponding to: This figure shows that the predominant event tend to be constant for a while during an episode, which is coherent with the expectation that important information about the past (an "event") changes with a low frequency. Moreover, we observed that these events correspond to specific changes of the environment. For instance, when the agent collects an object called "Power Pellet", which produces a switching of the color of the ghosts to blue and it allows these ghosts to be eaten by the agent, then at that moment there is a predominant event change in H(t).

EQUATION

Figure 5 shows some observations and the corresponding predominant events extracted from an episode of the Obstacle Tower environment. The observations have a sequential order (but they are not each other time-adjacent). New events correspond to the transition to a new room, or to different viewpoints of the same room. As shown in this figure, in the middle of the trajectory, the agent observes a green door, and when it later returns to it, the event changes back to the corresponding value. In Appendix D we plot another episode of the same environment, with high-frequency selected observations and the corresponding predominant events, which more clearly shows the semantics behind the clusters discovered by the encoder and used by the agent.

Sparsity of our history representation. Using Obstacle Tower we computed the probability that, in a trained agent, a random element in the S ×C matrix H(t) has a value q such that 0.1 < q < 0.9. This probability is 0.00146. Using the range of values 0.01 < q < 0.99, the probability is 0.0031. These empirical results show that our representation is very sparse and "binary", meaning that all the information is concentrated around the values 0 and 1. 

6. CONCLUSION:

We presented a method for history representation in RL which is an alternative to common solutions based on RNNs. Our EDHR is based on the idea that important information about the past can be "compressed" using events. Specifically, these events are automatically discovered using a modification of the ICC clustering method (Ji et al., 2019) , which is performed jointly with the agent training and iterated through time, to adapt the discovered events to the new observations. In EDHR, visual information is represented using two different networks: Φ and Ψ. The latter is trained using a reward signal, so it presumably extracts task-specific information from the observations. On the other hand, the encoder Φ is trained using self-supervision, and thus it focuses on patterns which are repeated in the data stream, potentially leveraging a larger quantity of supervision signal. Although self-supervision has been explored in other RL and non-RL works, this is the first work to show how the discovered information can be exploited in the form of discrete events used for history representation.

Our results, based on the challenging deep RL benchmarks, ALE and Obstacle Tower, show that EDHR can more effectively represent information about the past in comparison with task-oriented representations such as those used in PPO and PPO-RNN. A HYPERPARAMETERS

In Tab. 3 we list all our hyperparameters which are different from (Schulman et al., 2017) .

B ADDITIONAL EXPERIMENTS:

In Tab. 4 we use three environments of the ALE benchmark to show the impact of different EDHR hyperparameter choices together with an ablation study. Using a minimum length temporal window (L = 1) has a similar performance to L = 3, whilst a much larger value L = 8 degrades the performance. This is probably due to the fact that a large L value blurs out the borders between clusters, thus the event assignment is more ambiguous, less stable and less useful for the agent training.

As expected, a short history (S = 16) degrades the performance for the MsPacman and the SpaceInvaders environments. However, in this case Breakout performs slightly better, demonstrating that in a reactive environment the agent focuses on the instantaneous representation and the shorter history stimulates this. On the other hand, a longer history (S = 64) degrades the score for all the environments: distant events are less useful for the training, whereas the doubled size of H(t) makes the optimisation harder.

To show the impact of self-supervision, we include in Tab. 4 a variant of EDHR without the event discovery stage. Specifically, "Without self-supervision" is equal to EDHR, but Φ is trained using the gradient of the reward signal, jointly with Ψ. This experiment shows a dramatic degradation in the MsPacman and the SpaceInvaders environments, which highlights the gain which can be obtained using the proposed additional self-supervision. However, as above observed, in the more reactive environment Breakout the past information is less important, and the score achieved by "Without self-supervision" is equal to the plain PPO (Tab. 1), which indicates that the model ignores the history.

Finally, the last row in Tab. 4 concerns a variant of the PPO-RNN algorithm, where the input I(t) and the corresponding CNN Ψ are removed. In this case the network receives the last S observations of the environment as images, each image is processed using the CNN, the obtained features are input to the RNN, and the network is trained using the PPO algorithm. This model is similar to the common RNN-based solutions (e.g. Mnih et al. (2016) ). It is trained using the same hyperparameters of the other experiments. Empirically we observed that this solution is prone to a higher variance in the CNN parameter updates, requiring larger training budget.

C INVARIANT INFORMATION CLUSTERING:

In this section we briefly describe the self-supervised method we used for clustering, referring the reader to the original paper (Ji et al., 2019) for more details. Let x, x ∈ X be two different images sharing similar semantic information. In our case, x, x are two temporally-close frames, both contained in the temporal translation window (Sec. 3), while in (Ji et al., 2019) x, x represent the same image with some random perturbations (e.g. scaling, rotation, color adjustment, etc.).

Φ is a network mapping an image into a probability distribution Φ : X → [0, 1] C , where C is the number of clusters (in our case, events). Φ(x) is the probability distribution of a discrete random variable z which (soft-)assigns x to the elements in Y = {1, ..., C} and can be defined as: Φ c (x) = P (z = c|x). If x and x share the same semantic content, then they should be classified equally, otherwise they should be assigned to different clusters. This objective is obtained by maximizing the Mutual Information between the corresponding representations:

EQUATION

In more detail, a pair (x, x ) corresponds to the two variables (z, z ), with conditional joint distribution P (z = c, z = c |x, x ) = Φ c (x) • Φ c (x ), where the equality holds when z and z are independent (Ji et al., 2019) . After marginalization over the batch, the joint probability distribution is given by the C × C matrix P, where each element at position c, c is P cc = P (z = c, z = c ):

EQUATION

The marginals P c = P (z = c) and P c = P (z = c ) correspond to the sums over the rows and the columns of P, rispectively. Finally, the optimisation objective in Eq. 2 is computed directly using the definition of Mutual Information:

EQUATION

D EVENT VISUALIZATION:

Figure 6 shows a detailed sequence of observations and the corresponding predominant events from an episode of Obstacle Tower. We observe that some events correspond to semantically general classes and do not focus only on visual appearance. For instance, event 15 indicates when the agent passes from one room to another, despite each transition (room) is visually different. Events 2 and 12 correspond to specific rooms, event 8 detects the door, events 4 and 14 indicates specific views of a room.

:



Figures and Tables:

Figure 1: Event discovery. This figure shows two replicas of our unsupervised encoder architecture (Φ) which separately map two different observations in corresponding event probability distributions. The latter are used to maximize the MI over the latent assignment variable z, in this way obtaining a set of events Y and the supervision signal used to update the parameters (φ) of Φ.

Figure 3: Training dynamics, smoothed for readability

Figure 5: Obstacle Tower predominant events associated with sequentially ordered observations of an episode.

3 } and a history H(t) representing past events. Specifically, for each t ∈ {t, ..., t -S + 1} let Φ(o t ; φ) be the probability distribution of o t for the current set of events Y . Our proposed history representation is simply given to update θ I and θ A in n P epochs Event discovery stage: 9Use B to update φ in n E epochs by the concatenation of these cluster soft-assignment vectors, computed over the past S steps:

Table content: <table><tr><td colspan="2">Algorithm 1 Training</td></tr><tr><td colspan="2">1 Initialize φ and θ.</td></tr><tr><td colspan="2">2 For iteration = 1, ...N do:</td></tr><tr><td>3</td><td>For actor = 1, 2, ... do:</td></tr><tr><td>4</td><td>For t = 1, ..., r do:</td></tr><tr><td/><td>Observation collection stage:</td></tr><tr><td>5</td><td>Run π θ (I(t), H(t)) in the environment and get o t+1</td></tr><tr><td>6</td><td>Use Φ(o t+1 ; φ) to compute H(t + 1).</td></tr><tr><td>7</td><td>Collect I(t + 1), H(t + 1) and other PPO-specific information in a buffer B.</td></tr><tr><td/><td>Policy optimisation stage:</td></tr><tr><td>8</td><td>Use B</td></tr></table>



Table content: <table><tr><td/><td/><td colspan="3">: Scores on the ALE benchmark</td></tr><tr><td/><td colspan="2">PPO (Sch.) PPO</td><td/><td>PPO-RNN</td><td>EDHR (Ours)</td></tr><tr><td>MsPacman</td><td>2096.5</td><td colspan="2">1727.00 ± 383.81</td><td>2040.63 ± 104.88</td><td>2431.33 ± 216.26</td></tr><tr><td colspan="2">SpaceInvaders 942.5</td><td colspan="2">912.63 ± 5.20</td><td>952.30 ± 126.78</td><td>1089.48 ± 155.26</td></tr><tr><td>Breakout</td><td>274.8</td><td colspan="2">413.07 ± 22.12</td><td>410.40 ± 2.21</td><td>368.24 ± 36.54</td></tr><tr><td>Gravitar</td><td>737.2</td><td colspan="2">581.17 ± 23.01</td><td>600.33 ± 197.36</td><td>587.50 ± 46.29</td></tr><tr><td>Qbert</td><td>14293.3</td><td colspan="4">11340.75 ± 1463.59 9388.58 ± 2408.55 11816.67 ± 1952.70</td></tr><tr><td>Seaquest</td><td>1204.5</td><td colspan="2">1227.73 ± 481.22</td><td>1028.67 ± 125.69</td><td>1342.13 ± 339.85</td></tr><tr><td>Enduro</td><td>758.3</td><td colspan="2">539.75 ± 361.81</td><td>499.00 ± 432.16</td><td>746.69 ± 45.93</td></tr><tr><td/><td colspan="4">Table 2: Scores on the Obstacle Tower benchmark</td></tr><tr><td/><td/><td>PPO</td><td colspan="2">PPO-RNN EDHR (Ours)</td></tr><tr><td/><td colspan="4">Train 1.54 ± 0.07 3.34 ± 0.45 2.24 ± 0.18</td></tr><tr><td/><td>Test</td><td colspan="3">1.35 ± 0.24 2.46 ± 0.11 1.52 ± 0.23</td></tr></table>



Table content: <table><tr><td/><td colspan="2">: Hyperparameters</td><td/></tr><tr><td colspan="2">Number of clusters C</td><td>16</td><td/></tr><tr><td colspan="2">History size S</td><td>32</td><td/></tr><tr><td colspan="2">Temporal window L</td><td>3</td><td/></tr><tr><td colspan="2">Number of epochs n E</td><td>10</td><td/></tr><tr><td colspan="2">Batch size (to update φ)</td><td>512</td><td/></tr><tr><td colspan="2">Learning rate (to update φ)</td><td>.0005</td><td/></tr><tr><td colspan="3">Pretraing iterations ratio p pretrain 15%</td><td/></tr><tr><td/><td colspan="2">Table 4: Additional experiments</td><td/></tr><tr><td/><td>MsPacman</td><td>SpaceInvaders</td><td>Breakout</td></tr><tr><td colspan="4">EDHR L = 3, S = 32 (default) 2431.33 ± 216.26 1089.48 ± 155.26 368.24 ± 36.54</td></tr><tr><td>EDHR L = 1, S = 32</td><td colspan="3">2473.87 ± 355.57 1048.72 ± 128.57 375.54 ± 31.90</td></tr><tr><td>EDHR L = 8, S = 32</td><td colspan="2">2038.33 ± 435.01 910.85 ± 44.65</td><td>389.07 ± 6.31</td></tr><tr><td>EDHR L = 3, S = 16</td><td colspan="2">2148.03 ± 247.37 864.05 ± 122.53</td><td>411.09 ± 11.89</td></tr><tr><td>EDHR L = 3, S = 64</td><td>1916.67 ± 76.67</td><td>875.52 ± 58.46</td><td>373.30 ± 9.49</td></tr><tr><td>Without self-supervision</td><td>1897.70 ± 90.54</td><td>981.20 ± 171.63</td><td>411.58 ± 2.86</td></tr><tr><td>PPO-RNN without I(t)</td><td colspan="2">1317.47 ± 136.67 746.80 ± 33.82</td><td>189.40 ± 29.84</td></tr></table>

Peer review (iclr2020):
The paper focuses on the problem learning state representations that can effectively capture historical information in POMDPs. Specifically, the paper proposes an alternative approach to using RNN*s for capturing such history - the authors adopt a variant of recently proposed Invariant Information Clustering approach to discover important events in past observation and then use these events to learn a probability distribution over observations to represent the state information. The goal here is to address the unstable and inefficient training issues associated with RNN based architectures. The authors validate their approach with experiments on seven tasks on Atari 57 benchmark and Obstacle Tower, both with discrete action space and compare their performance against both original and RNN versions of PPO. The paper addresses an interesting problem to help learning better state representation in absence of complete information about the environment. The approach of using IIC clustering (or any clustering approach) for learning state representations is novel. The overall goal of replacing RNN based methods in order to achieve stable, efficient learning in presence of budget constraints is very useful and hence this approach is potentially a good step in that direction. Although not adequate, the results in figure 3 provides good insight into effectiveness of the method in making training easier. However, I am inclining to reject this paper for the following reasons: (1) The motivation for using proposed clustering approach for history representation is neither clear and nor well exposed. MI based methods are inherently difficult to learn and hence this approach needs rigorous analysis on why it works when it does and how it fails. (2) The paper fails to position the new approach in comparison to related works both in discussions and experiments. (3) The experiments are very limited in nature and fails to demonstrate the efficacy of the proposed approach effectively. Further, the key contribution focusing on learning effective representations under constrained budget is not adequately tested. Major concerns: Motivation ------------- - It is not clear how the proposed clustering mechanism to discover events allows to successfully capture information that an RNN based approach does. For instance, RNN helps to capture long-term dependencies but it is very hard to interpret the proposed model from that aspect. Also, why is this particular clustering approach (Invariant Information Clustering) chosen? The motivation for using this approach is not clear and overall combination appears adhoc. - Further, RNN based methods can retain order information in sequential history of observations. However, this method appears to not consider that information. Is this is true or am I mistaken here? If this is true, why would this not create issues in learning good representations for task where order is indeed important? - The paper discusses several articles that provide background on the methods used however it fails to position the exposition in comparison to existing RNN based approaches which is a big miss as the goal of the paper is to replace RNN based approaches [1,2,3,4]. Method ------ - In the event discovery stage, it is not clear why using consecutive observations is not useful. Also, if one does use L=1 (consecutive observations), can one reduce this method to RNN as you end of capturing all the previous history? Also, why L=3 is good across all different tasks? does it have any relation with the use of 4 frames in I(t)? - The authors mention tat H(t) matrix is highly sparse but also low-dimensional in all their experiments. Would this is be the case for any other task? If not, is this a limitation of the method that you need S and C to be small? - The authors attempt to use clustering based approach with the hope of recovering important information, however mention that H(t) stores all the past events. Isn*t this contradictory? Also, why can RNN with an attention based mechanism not achieve similar effect? - It is also useful to analyse how will this method work in presence of long vs short history? Will the clustering itself and hence the learned representations get affected by length of available history? Experiments ---------- - Focusing only on PPO and designing an RNN version of PPO constrains the effectiveness of experiments in validating the approach. Authors mention difficulty of training with RNN as one reason for PPO with RNN*s under performance but this is not convincing Could the performance of PPO with RNN be limited only due to specific RNN architecture used? - Authors must compare with other methods that use RNN approaches (e.g. [1]). Also, if methods such [2],[3],[4] are not directly applicable, they must atleast, use their RNN based architectures to modify PPO and compare several baselines. - Why do the authors not report PPO with RNN for Obstacle Tower? - For the experiments, authors use a specific 10000 steps budget but this seems to be highly curated. The experiments would be stronger if the authors show experiment over a range of budget. This will also give insights on when does RNN becomes better and is there a budget after which both methods perform equally well or RNN based approaches surpass the current approaches. - Figure 3: Training dynamics seem to favor RNN approach for BreakOut, Gravitar. Do the authors have insight on why this is the case? Minor points to improve submission not affecting the score: - The paper needs to be proofread for various typos and sentence construction issues - Table 1: For Qbert, original PPO (Sch.) is best performing, not EDHR - [5] talks about difficulties in MI based methods and I encourage the authors to look at the analysis in this paper. As it is only an arXiv version and not published yet, I have not based my assessment on the existence of this paper but still connection to such analysis will make this paper stronger. [1] Deep Variational Reinforcement Learning for POMDPs, Igl et. al. [2] On improving deep reinforcement learning for POMDPs, Zhu et. al. [3] Policy Learning with continuous memory states in partially observed robotic control, Zhang et. al. [4] Memory-based control with recurrent neural networks, Heess et. al. [5] On Mutual Information Maximization and Representation Learning, Tschannen et. al.