Title: EXPLORATION BASED LANGUAGE LEARNING FOR TEXT-BASED GAMES

Abstract: This work presents an exploration and imitation-learning-based agent capable of state-of-the-art performance in playing text-based computer games. Text-based computer games describe their world to the player through natural language and expect the player to interact with the game using text. These games are of interest as they can be seen as a testbed for language understanding, problem-solving, and language generation by artificial agents. Moreover, they provide a learning environment in which these skills can be acquired through interactions with an environment rather than using fixed corpora. One aspect that makes these games particularly challenging for learning agents is the combinatorially large action space. Existing methods for solving text-based games are limited to games that are either very simple or have an action space restricted to a predetermined set of admissible actions. In this work, we propose to use the exploration approach of Go-Explore (Ecoffet et al., 2019) for solving text-based games. More specifically, in an initial exploration phase, we first extract trajectories with high rewards, after which we train a policy to solve the game by imitating these trajectories. Our experiments show that this approach outperforms existing solutions in solving textbased games, and it is more sample efficient in terms of the number of interactions with the environment. Moreover, we show that the learned policy can generalize better than existing solutions to unseen games without using any restriction on the action space.

1. INTRODUCTION:

Text-based games became popular in the mid 80s with the game series Zork (Anderson & Galley, 1985) resulting in many different text-based games being produced and published (Spaceman, 2019) . These games use a plain text description of the environment and the player has to interact with them by writing natural-language commands. Recently, there has been a growing interest in developing agents that can automatically solve text-based games (Côté et al., 2018) by interacting with them. These settings challenge the ability of an artificial agent to understand natural language, common sense knowledge, and to develop the ability to interact with environments using language (Luketina et al., 2019; Branavan et al., 2012) .

Since the actions in these games are commands that are in natural language form, the major obstacle is the extremely large action space of the agent, which leads to a combinatorially large exploration problem. In fact, with a vocabulary of N words (e.g. 20K) and the possibility of producing sentences with at most m words (e.g. 7 words), the total number of actions is O(N m ) (e.g. 20K 7 ≈ 1.28e 30 ). To avoid this large action space, several existing solutions focus on simpler text-based games with very small vocabularies where the action space is constrained to verb-object pairs (DePristo & Zubek, 2001; Narasimhan et al., 2015; Yuan et al., 2018; Zelinka, 2018) . Moreover, many existing works rely on using predetermined sets of admissible actions (He et al., 2015; Tessler et al., 2019; Zahavy et al., 2018) . However, a more ideal, and still under explored, alternative would be an agent that can operate in the full, unconstrained action space of natural language that can systematically generalize to new text-based games with no or few interactions with the environment.

To address this challenge, we propose to use the idea behind the recently proposed Go-Explore (Ecoffet et al., 2019) algorithm. Specifically, we propose to first extract high reward trajectories of states and actions in the game using the exploration methodology proposed in Go-Explore and then train a policy using a Seq2Seq (Sutskever et al., 2014) model that maps observations to actions, in an imitation learning fashion. To show the effectiveness of our proposed methodology, we first benchmark the exploration ability of Go-Explore on the family of text-based games called CoinCollector (Yuan et al., 2018) . Then we use the 4,440 games of "First TextWorld Problems" (Côté, 2018) , which are generated using the machinery introduced by Côté et al. (2018) , to show the generalization ability of our proposed methodology. In the former experiment we show that Go-Explore finds winning trajectories faster than existing solutions, and in the latter, we show that training a Seq2Seq model on the trajectories found by Go-Explore results in stronger generalization, as suggested by the stronger performance on unseen games, compared to existing competitive baselines (He et al., 2015; Narasimhan et al., 2015) .

Reinforcement Learning Based Approaches for Text-Based Games Among reinforcement learning based efforts to solve text-based games two approaches are prominent. The first approach assumes an action as a sentence of a fixed number of words, and associates a separate Qfunction (Watkins, 1989; Mnih et al., 2015) with each word position in this sentence. This method was demonstrated with two-word sentences consisting of a verb-object pair (e.g. take apple) (De-Pristo & Zubek, 2001; Narasimhan et al., 2015; Yuan et al., 2018; Zelinka, 2018; Fulda et al., 2017) . In the second approach, one Q-function that scores all possible actions (i.e. sentences) is learned and used to play the game (He et al., 2015; Tessler et al., 2019; Zahavy et al., 2018) . The first approach is quite limiting since a fixed number of words must be selected in advance and no temporal dependency is enforced between words (e.g. lack of language modelling). In the second approach, on the other hand, the number of possible actions can become exponentially large if the admissible actions (a predetermined low cardinality set of actions that the agent can take) are not provided to the agent. A possible solution to this issue has been proposed by Tao et al. (2018) , where a hierarchical pointer-generator is used to first produce the set of admissible actions given the observation, and subsequently one element of this set is chosen as the action for that observation. However, in our experiments we show that even in settings where the true set of admissible actions is provided by the environment, a Q-scorer (He et al., 2015) does not generalize well in our setting (Section 5.2 Zero-Shot) and we would expect performance to degrade even further if the admissible actions were generated by a separate model. Less common are models that either learn to reduce a large set of actions into a smaller set of admissible actions by eliminating actions (Zahavy et al., 2018) or by compressing them in a latent space (Tessler et al., 2019) .

Exploration in Reinforcement Learning:

In most text-based games rewards are sparse, since the size of the action space makes the probability of observing a reward extremely low when taking only random actions. Sparse reward environments are particularly challenging for reinforcement learning as they require longer term planning. Many exploration based solutions have been proposed to address the challenges associated with reward sparsity. Among these exploration approaches are novelty search (Lehman & Stanley, 2008; 2011; Conti et al., 2018; Achiam & Sastry, 2017; Burda et al., 2018) , intrinsic motivation (Schmidhuber, 1991b; Oudeyer & Kaplan, 2009; Barto, 2013) , and curiosity based rewards (Schmidhuber, 2006; 1991a; Pathak et al., 2017) . For text based games exploration methods have been studied by Yuan et al. (2018) , where the authors showed the effectiveness of the episodic discovery bonus (Gershman & Daw, 2017) in environments with sparse rewards. This exploration method can only be applied in games with very small action and state spaces, since their counting methods rely on the state in its explicit raw form.

2. METHODOLOGY:

Go-Explore (Ecoffet et al., 2019) differs from the exploration-based algorithms discussed above in that it explicitly keeps track of under-explored areas of the state space and in that it utilizes the determinism of the simulator in order to return to those states, allowing it to explore sparse-reward environments in a sample efficient way (see Ecoffet et al. (2019) as well as section 4.1). For the experiments in this paper we mainly focus on the final performance of our policy, not how that policy is trained, thus making Go-Explore a suitable algorithm for our experiments. Go-Explore is composed of two phases. In phase 1 (also referred to as the "exploration" phase) the algorithm explores the state space through keeping track of previously visited states by maintaining an archive. During this phase, instead of resuming the exploration from scratch, the algorithm starts exploring from promising states in the archive to find high performing trajectories. In phase 2 (also referred to as the "robustification" phase, while in our variant we will call it "generalization") the algorithm trains a policy using the trajectories found in phase 1. Following this framework, which is also shown in Figure 3 (Appendix A.2), we define the Go-Explore phases for text-based games.

Let us first define text-based games using the same notation as Yuan et al. (2018) Let us also define the observation o t ∈ Ω and the action a t ∈ A. Text-based games provide some information in plain text at each turn and, without loss of generality, we define an observation o t as the sequence of tokens {o 0 t , • • • , o n t } that form that text. Similarly, we define the tokens of an action a t as the sequence {a 0 t , • • • , a m t }. Furthermore, we define the set of admissible actions

A t ∈ A as A t = {a 0 , • • • , a z }

, where each a i , which is a sequence of tokens, is grammatically correct and admissible with reference to the observation o t .

2.1. PHASE 1: EXPLORATION:

In phase 1, Go-Explore builds an archive of cells, where a cell is defined as a set of observations that are mapped to the same, discrete representation by some mapping function f (x). Each cell is associated with meta-data including the trajectory towards that cell, the length of that trajectory, and the cumulative reward of that trajectory. New cells are added to the archive when they are encountered in the environment, and existing cells are updated with new meta-data when the trajectory towards that cells is higher scoring or equal scoring but shorter.

At each iteration the algorithm selects a cell from this archive based on meta-data of the cell (e.g. the accumulated reward, etc.) and starts to randomly explore from the end of the trajectory associated with the selected cell. Phase 1 requires three components: the way that observations are embedded into cell representations, the cell selection, and the way actions are randomly selected when exploring from a selected cell. In our variant of the algorithm, f (x) is defined as follows: given an observation, we compute the word embedding for each token in this observation, sum these embeddings, and then concatenate this sum with the current cumulative reward to construct the cell representation. The resulting vectors are subsequently compressed and discretized by binning them in order to map similar observations to the same cell. This way, the cell representation, which is the key of the archive, incorporates information about the current observation of the game. Adding the current cumulative reward to the cell representation is new to our Go-Explore variant as in the original algorithm only down-scaled image pixels were used. It turned out to be a very very effective to increase the speed at which high reward trajectories are discovered. In phase 1, we restrict the action space to the set of admissible actions A t that are provided by the game at every step of the game1 . This too is particularly important for the random search to find a high reward trajectory faster. Finally, we denote the trajectory found in phase 1 for game g as 2.2 PHASE 2: GENERALIZATION Phase 2 of Go-Explore uses the trajectories found in phase 1 and trains a policy based on those trajectories. The goal of this phase in the original Go-Explore algorithm is to turn the fragile policy of playing a trajectory of actions in sequence into a more robust, state-conditioned policy that can thus deal with environmental stochasticity. In our variant of the algorithm the purpose of the second phase is generalization: although in our environment there is no stochasticity, our goal is to learn a general policy that can be applied across different games and generalize to unseen games. In the original Go-Explore implementation, the authors used the backward Proximal Policy Optimization algorithm (PPO) (Salimans & Chen, 2018; Schulman et al., 2017) to train this policy. In this work we opt for a simple but effective Seq2Seq imitation learning approach that does not use the reward directly in the loss. More specifically, given the trajectory T g2 , we train a Seq2Seq model to minimize the negative log-likelihood of the action a t given the observation o t . In other words, consider a word embedding matrix E ∈ R d×|V | where d is the embedding size and |V | is the cardinality of the vocabulary, which maps the input token to an embedded vector. Then, we define an encoder LSTM enc and a decoder LSTM dec . Every token o t from the trajectory T g is converted to its embedded representation using the embedding matrix E and the sequence of these embedding vectors is passed through LSTM enc :

T g = [(o 0 , a 0 , r 0 ), • • • , (o t , a t , r t )].

h enc i = LSTM enc (E(o i t ), h enc i-1

).

(1) The last hidden state h enc |ot| is used as the initial hidden state of the decoder which generates the action a t token by token. Specifically, given the sequence of hidden states H ∈ R d×|ot| of the encoder, tokens a j t are generated as follows:

EQUATION

EQUATION

where W ∈ R 2d×|V | is a matrix that maps the decoder hidden state, concatenated with the context vector, into a vocabulary size vector. During training, the parameters of the model are trained by minimizing:

EQUATION

which is the sum of the negative log likelihood of each token in a t (using teacher forcing (Williams & Zipser, 1989) ). However, at test time the model produces the sequence in an auto-regressive way (Graves, 2013) using greedy search.

3. EXPERIMENTS:



3.1. GAMES AND EXPERIMENTS SETUP:

A set of commonly used standard benchmarks (Yuan et al., 2018; Côté et al., 2018; Narasimhan et al., 2015) for agents that play text-based games are simple games which require no more than two words in each step to solve the game, and have a very limited number of admissible actions 205.00K ± 29.41K 30 ± 0.0 per observation. While simplifying, this setting limits the agent's ability to fully express natural language and learn more complex ways to speak. In this paper, we embrace more challenging environments where multiple words are needed at each step to solve the games and the reward is particularly sparse. Hence, we have selected the following environments:

• CoinCollector (Yuan et al., 2018) is a class of text-based games where the objective is to find and collect a coin from a specific location in a given set of connected rooms3 . The agent wins the game after it collects the coin, at which point (for the first and only time) a reward of +1 is received by the agent. The environment parses only five admissible commands (go north, go east, go south, go west, and take coin) made by two worlds;

• CookingWorld4 (Côté, 2018) in this challenge, there are 4,440 games with 222 different levels of difficulty, with 20 games per level of difficulty, each with different entities and maps. The goal of each game is to cook and eat food from a given recipe, which includes the task of collecting ingredients (e.g. tomato, potato, etc.), objects (e.g. knife), and processing them according to the recipe (e.g. cook potato, slice tomato, etc.). The parser of each game accepts 18 verbs and 51 entities with a predefined grammar, but the overall size of the vocabulary of the observations is 20,000. In Appendix A.1 we provide more details about the levels and the games' grammar.

In our experiments, we try to address two major research questions. First, we want to benchmark the exploration power of phase 1 of Go-Explore in comparison to existing exploration approaches used in text-based games. For this purpose, we generate 10 CoinCollector games with the hardest setting used by Yuan et al. (2018) , i.e. hard-level 30 (refer to the Appendix A.1 for more information) and use them as a benchmark. In fact, CoinCollector requires many actions (at least 30 on hard games) to find a reward, which makes it suitable for testing the exploration capabilities of different algorithms. Secondly, we want to verify the generalization ability of our model in creating complex strategies using natural language. CoinCollector has a very limited action space, and is mainly designed to benchmark models on their capability of dealing with sparse rewards. Therefore we use the more complex CookingWorld games to evaluate the generalization capabilities of our proposed approach. We design three different settings for CookingWorld: 1) Single: treat each game independently, which means we train and test one agent for each game to evaluate how robust different models are across different games.; 2) Joint: training and testing a single policy on all the 4,440 CookingWorld games at the same time to verify that models can learn to play multiple games at the same time;

3) Zero-Shot: split the games into training, validation, and test sets, and then train our policy on the training games and test it on the unseen test games. This setting is the hardest among all, since it requires generalization to unseen games.

In both CoinCollector and CookingWorld games an observation o t provided by the environment consists of a room description D, inventory information I, quest Q, previous action P and feedback F provided in the previous turn. Table 1 shows an example for each of these components. In our experiments for phase 1 of Go-Explore we only use D as the observation.

3.2. BASELINES:

For the CoinCollector games, we compared Go-Explore with the episodic discovery bonus (Gershman & Daw, 2017) that was used by Yuan et al. (2018) to improve two Q-learning-based baselines: DQN++ and DRQN++. We used the code provided by the authors and the same hyper-parameters5 .

For the CookingWorld games, we implemented three different treatments based on two existing methods:

• LSTM-DQN (Narasimhan et al., 2015; Yuan et al., 2018) : An LSTM based state encoder with a separate Q-functions for each component (word) of a fixed pattern of Verb, Adjective1, Noun1, Adjective2, and Noun2. In this approach, given the observation o t , the tokens are first converted into embeddings, then an LSTM is used to extract a sequence of hidden states H dqn ∈ R d×|ot| . A mean-pool layer is applied to H dqn to produce a single vector h o t that represents the whole sequence. Next, a linear transformation W type ∈ R d×|Vtype| is used to generate each of the Q values, where |V type | |V | is the subset of the original vocabulary restricted to the word type of a particular game (e.g for Verb type: take, drop, etc.). Formally, we have:

Q(o t , a type ) = h o t W type

where, type ∈ {Verb, Obj, Noun, Obj2,Noun2}. ( 6)

Next, all the Q-functions are jointly trained using the DQN algorithm with -greedy exploration (Watkins, 1989; Mnih et al., 2015) . At evaluation time, the argmax of each Q-function is concatenated to produce a t . Importantly, in V type a special token 
 is used to denote the absence of a word, so the model can produce actions with different lengths. Figure 4 in Appendix A.2 shows a depiction of this model. • LSTM-DQN+ADM: It is the same model as LSTM-DQN, except that the random actions for -greedy exploration are sampled from the set of admissible actions instead of creating them by sampling each word separately. • DRRN (He et al., 2015) : In this approach a model learns how to score admissible actions instead of directly generating the action token by token. The policy uses an LSTM for encoding the observation and actions are represented as the sum of the embedding of the word tokens they contain. Then, the Q value is defined as the dot product between the embedded representations of the observation and the action. Following the aforementioned notation, h o t is generated as in the LSTM-DQN baseline. Next, we define its embedded representation as

c i = |ai| k E(a k i )

, where E is an embedding matrix as in Equation 1. Thus, the Q-function is defined as:

Q(o t , a i ) = h o t c i (7)

At testing time the action with the highest Q value is chosen. Figure 6 in Appendix A.2 shows a depiction of this model.

3.3. HYPER-PARAMETERS:

In all the games the maximum number of steps has been set to 50. As mentioned earlier, the cell representation used in the Go-Explore archive is computed by the sum of embeddings of the room description tokens concatenated with the current cumulative reward. The sum of embeddings is computed using 50 dimensional pre-trained GloVe (Pennington et al., 2014) vectors. In the Coin-Collector baselines we use the same hyper-parameters as in the original paper. In CookingWorld all the baselines use pre-trained GloVe of dimension 100 for the single setting and 300 for the joint one. The LSTM hidden state has been set to 300 for all the models.

4. RESULTS:



4.1. COINCOLLECTOR:

In this setting, we compare the number of actions played in the environment (frames) and the score achieved by the agent (i.e. +1 reward if the coin is collected). In Go-Explore we also count the actions used to restore the environment to a selected cell, i.e. to bring the agent to the state represented in the selected cell. This allows a one-to-one comparison of the exploration efficiency between Go-Explore and algorithms that use a count-based reward in text-based games. Importantly, Yuan et al. (2018) showed that DQN and DRQN, without such counting rewards, could never find a successful trajectory in hard games such as the ones used in our experiments. Figure 1 shows the number of interactions with the environment (frames) versus the maximum score obtained, averaged over 10 games of the same difficulty. As shown by Yuan et al. (2018) , DRQN++ finds a trajectory with the maximum score faster than to DQN++. On the other hand, phase 1 of Go-Explore finds an optimal trajectory with approximately half the interactions with the environment. Moreover, the trajectory length found by Go-Explore is always optimal (i.e. 30 steps) whereas both DQN++ and DRQN++ have an average length of 38 and 42 respectively.

4.2. COOKINGWORLD:

In CookingWorld, we compared models in the three settings mentioned earlier, namely, single, joint, and zero-shot. In all experiments, we measured the sum of the final scores of all the games and their trajectory length (number of steps). Table 3 summarizes the results in these three settings. Phase 1 of Go-Explore on single games achieves a total score of 19,530 (sum over all games), which is very close to the maximum possible points (i.e. 19,882), with 47,562 steps. A winning trajectory was found in 4,279 out of the total of 4,440 games. This result confirms again that the exploration strategy of Go-Explore is effective in text-based games. Next, we evaluate the effectiveness and the generalization ability of the simple imitation learning policy trained using the extracted trajectories in phase 1 of Go-Explore in the three settings mentioned above.

Single In this setting, each model is trained from scratch in each of the 4,440 games based on the trajectory found in phase 1 of Go-Explore (previous step). As shown in Table 3 , the LSTM-DQN (Narasimhan et al., 2015; Yuan et al., 2018) approach without the use of admissible actions performs poorly. One explanation for this could be that it is difficult for this model to explore both language and game strategy at the same time; it is hard for the model to find a reward signal before it has learned to model language, since almost none of its actions will be admissible, and those reward signals are what is necessary in order to learn the language model. As we see in Table 3 , however, by using the admissible actions in the -greedy step the score achieved by the LSTM-DQN increases dramatically (+ADM row in Table 3 ). DRRN (He et al., 2015) achieves a very high score, since it explicitly learns how to rank admissible actions (i.e. a much simpler task than generating text). Finally, our approach of using a Seq2Seq model trained on the single trajectory provided by phase 1 of Go-Explore achieves the highest score among all the methods, even though we do not use admissible actions in this phase. However, in this experiment the Seq2Seq model cannot perfectly replicate the provided trajectory and the total score that it achieves is in fact 9.4% lower compared to the total score achieved by phase 1 of Go-Explore. Figure 7 (in Appendix A.3) shows the score breakdown for each level and model, where we can see that the gap between our model and other methods increases as the games become harder in terms of skills needed.

Joint In this setting, a single model is trained on all the games at the same time, to test whether one agent can learn to play multiple games. Overall, as expected, all the evaluated models achieved a lower performance compared to the single game setting. One reason for this could be that learning multiple games at the same time leads to a situation where the agent encounters similar observations in different games, and the correct action to take in different games may be different. Furthermore, it is important to note that the order in which games are presented greatly affects the performance of LSTM-DQN and DRRN. In our experiments, we tried both an easy-to-hard curriculum (i.e. sorting the games by increasing level of difficulty) and a shuffled curriculum. Shuffling the games at each epoch resulted in far better performance, thus we only report the latter. In Figure 2 we show the score breakdown, and we can see that all the baselines quickly fail, even for easier games.

Zero-Shot In this setting the 4,440 games are split into training, validation, and test games. The split is done randomly but in a way that different difficulty levels (recipes 1, 2 and 3), are represented with equal ratios in all the 3 splits, i.e. stratified by difficulty. As shown in Table 3 , the zero-shot performance of the RL baselines is poor, which could be attributed to the same reasons why RL baselines under-perform in the Joint case. Especially interesting is that the performance of DRRN is substantially lower than that of the Go-Explore Seq2Seq model, even though the DRRN model has access to the admissible actions at test time, while the Seq2Seq model (as well as the LSTM-DQN model) has to construct actions token-by-token from the entire vocabulary of 20,000 tokens. On the other hand, Go-Explore Seq2Seq shows promising results by solving almost half of the unseen games. Figure 8 (in Appendix A.3) shows that most of the lost games are in the hardest set, where a very long sequence of actions is required for winning the game. These results demonstrate both the relative effectiveness of training a Seq2Seq model on Go-Explore trajectories, but they also indicate that additional effort needed for designing reinforcement learning algorithms that effectively generalize to unseen games.

5. DISCUSSION:

Experimental results show that our proposed Go-Explore exploration strategy is a viable methodology for extracting high-performing trajectories in text-based games. This method allows us to train supervised models that can outperform existing models in the experimental settings that we study. Finally, there are still several challenges and limitations that both our methodology and previous solutions do not fully address yet. For instance:

State Representation The state representation is the main limitation of our proposed imitation learning model. In fact, by examining the observations provided in different games, we notice a large overlap in the descriptions (D) of the games. This overlap leads to a situation where the policy receives very similar observations, but is expected to imitate two different actions. This show especially in the joint setting of CookingWorld, where the 222 games are repeated 20 times with different entities and room maps. In this work, we opted for a simple Seq2Seq model for our policy, since our goal is to show the effectiveness of our proposed exploration methods. However, a more complex Hierarchical-Seq2Seq model (Sordoni et al., 2015) or a better encoder representation based on knowledge graphs (Ammanabrolu & Riedl, 2019a; b) would likely improve the of performance this approach.

Language Based Exploration In Go-Explore, the given admissible actions are used during random exploration. However, in more complex games, e.g. Zork I and in general the Z-Machine games, these admissible actions are not provided. In such settings, the action space would explode in size, and thus Go-Explore, even with an appropriate cell representation, would have a hard time finding good trajectories. To address this issue one could leverage general language models to produce a set of grammatically correct actions. Alternatively one could iteratively learn a policy to sample actions, while exploring with Go-Explore. Both strategies are viable, and a comparison is left to future work.

It is worth noting that a hand-tailored solution for the CookingWorld games has been proposed in the "First TextWorld Problems" competition (Côté et al., 2018) . This solution6 managed to obtain up to 91.9% of the maximum possible score across the 514 test games on an unpublished dataset.

However, this solution relies on entity extraction and template filling, which we believe limits its potential for generalization. Therefore, this approach should be viewed as complementary rather than competitor to our approach as it could potentially be used as an alternative way of getting promising trajectories.

6. CONCLUSION:

In this paper we presented a novel methodology for solving text-based games which first extracts high-performing trajectories using phase 1 of Go-Explore and then trains a simple Seq2Seq model that maps observations to actions using the extracted trajectories. Our experiments show promising results in three settings, with improved generalization and sample efficiency compared to existing methods. Finally, we discussed the limitations and possible improvements of our methodology, which leads to new research challenges in text-based games.

A APPENDIX:

A.1 TEXT-GAMES CoinCollector In the hard setting (mode 2), each room on the path to the coin has two distractor rooms, and the level (e.g. 30) indicates the shortest path from the starting point to the coin room.

CookingWorld The game's complexity is determined by the number of skills and the types of skills that an agent needs to master. The skills are:

• recipe {1,2,3}: number of ingredients in the recipe

• take {1,2,3}: number of ingredients to find (not already in the inventory)

• open: whether containers/doors need to be opened

• cook: whether ingredients need to be cooked

• cut: whether ingredients need to be cut

• drop: whether the inventory has limited capacity

• go {1,6,9,12}: number of locations Thus the hardest game would be a recipe with 3 ingredients, which must all be picked up somewhere across 12 locations and then need to be cut and cooked, and to get access to some locations, several doors or objects need to be opened. The handicap of a limited capacity in the inventory makes the game more difficult by requiring the agent to drop an object and later on take it again if needed. The grammar used for the text-based games is the following:

• go, look, examine, inventory, eat, open/close, take/drop, put/insert

• cook X with Y -→ grilled X (when Y is the BBQ)

• cook X with Y -→ roasted X (when Y is the oven)

• cook X with Y -→ fried X (when Y is the stove)

• slice X with Y -→ sliced X

• chop X with Y -→ chopped X

• dice X with Y -→ diced X • prepare meal

• Where Y is something sharp (e.g. knife). 

A.2 MODELS:



:

Note that the final goal is to generalize to test environments where admissible actions are not available.The assumption that admissible actions are available at training time holds in cases where we build the training environment for an RL agent (e.g. a hand-crafted dialogue system), and a system trained in such an environment can be practically applied as long as the system does not rely on such information at test time. Thus, we assumed that these admissible commands are not available at test time.

Or the set of trajectories, depending on the training setting

Visual example provided at: https://github.com/xingdi-eric-yuan/TextWorld-CoinCollector

We gave this name to this family of games, following the naming convention of TextWorld environments

The authors released their code at https://github.com/xingdi-eric-yuan/TextWorld-Coin-Collector.

https://www.microsoft.com/en-us/research/blog/first-textworld-problems-the-competition-using-textbased-games-to-advance-capabilities-of-ai-agents/

Figures and Tables:

Figure 1: CoinCollector results of DQN++ and DRQN++(Yuan et al., 2018) versus Go-Explore Phase1, i.e. just exploration.

Figure 2: Breakdown of results for the CookingWorld games in the Joint setting. The results are normalized and sorted by increasing difficulty level from left to right, averaged among the 20 games of each level.

Example of the observations provided by the CookingWorld environment.

Table content: <table><tr><td>Description</td><td>-= garden = -well, here we are in a garden. There is a roasted red apple and a red onion on the floor</td><td>Admissible commands</td></tr><tr><td>Inventory</td><td>You are carrying: a black pepper</td><td>-drop black pepper</td></tr><tr><td>Prev. Action</td><td>Open screen door</td><td>-eat black pepper</td></tr><tr><td/><td>Gather all following ingredients and follow the</td><td>-examine red apple</td></tr><tr><td/><td>directions to prepare this tasty meal.</td><td>-examine red onion</td></tr><tr><td>Quest</td><td>Ingredients: black pepper, red apple salt.</td><td>-go north</td></tr><tr><td/><td>Directions: chop the red apple, roast the red apple,</td><td>-look</td></tr><tr><td/><td>prepare meal</td><td>-take red apple</td></tr><tr><td>Feedback</td><td>That 's already open</td><td>-take red onion</td></tr></table>

S is the set of the environment states, T is the state transition function that defines the next state probability, i.e. T (s t+1 |a t ; s t )∀s t ∈ S, A is the set of actions, which in our case is all the possible sequences of tokens, Ω is the set of observations, i.e. text observed by the agent every time has it to take an action in the game (i.e.

Table content: <table><tr><td>. A text-based game</td></tr><tr><td>can be framed as a discrete-time Partially Observable Markov Decision Process (POMDP) (Kael-</td></tr><tr><td>bling et al., 1998) defined by (S, T, A, Ω, O, R), where:</td></tr></table>

Statistics of the two families of text-based games used in the experiments. The average is among the different games in CookingWorld and among different instance of the same game for CoinCollector.

Table content: <table><tr><td/><td>CoinCollector</td><td>CookingWorld</td></tr><tr><td>Vocabulary size (|V |)</td><td>1,250</td><td>20,000</td></tr><tr><td>Action Space</td><td>10</td><td>20,000 5</td></tr><tr><td>#Level</td><td>1 (Hard)</td><td>222</td></tr><tr><td>Max Room</td><td>90</td><td>12</td></tr><tr><td>Avg. #Tok. Description</td><td>64 ± 9</td><td>97 ± 49</td></tr><tr><td>Avg. #Adm. Actions</td><td>10</td><td>14 ± 13</td></tr></table>

CookingWorld results on the three evaluated settings single, joint and zero-shot.

Table content: <table><tr><td/><td/><td>Single</td><td/><td/><td>Joint</td><td/><td/><td>Zero-Shot</td><td/></tr><tr><td>Model</td><td>Score</td><td>Steps</td><td>Win</td><td>Score</td><td>Steps</td><td>Win</td><td>Score</td><td>Steps</td><td>Win</td></tr><tr><td>LSTM-DQN</td><td>2206</td><td>201832</td><td>412</td><td>473</td><td>213618</td><td>172</td><td>31</td><td>21480</td><td>14</td></tr><tr><td>+ADM</td><td>10360</td><td>140940</td><td>1770</td><td>623</td><td>210283</td><td>200</td><td>37</td><td>21530</td><td>14</td></tr><tr><td>DRRN</td><td>16075</td><td>78856</td><td>3195</td><td>4560</td><td>184888</td><td>216</td><td>451</td><td>17243</td><td>37</td></tr><tr><td>Go-Explore Seq2Seq</td><td>17507</td><td>68571</td><td>3757</td><td>11167</td><td>85967</td><td>2326</td><td>1038</td><td>10805</td><td>207</td></tr><tr><td>Go-Explore Phase 1</td><td>19437</td><td>49011</td><td>4279</td><td>-</td><td>-</td><td>-</td><td>-</td><td>-</td><td>-</td></tr><tr><td>Max Possible</td><td>19882</td><td>-</td><td>4440</td><td>19882</td><td>-</td><td>4440</td><td>2034</td><td>-</td><td>444</td></tr></table>

High level intuition of the Go-Exlore algorithm. Figure taken from Ecoffet et al. (2019) with permission.Figure 4: LSTM-DQN high level schema.

Table content: <table><tr><td>take</td><td>red</td><td>pepper</td><td>&lt;/s&gt;</td></tr><tr><td colspan="3">Phase 1: explore until solved Go to state Explore from state LSTM pepper from table Figure 3: LSTM + MeanPool Select state from archive Update archive take red + + + + =</td><td>Phase 2: generalize Run imitation learning on best trajectory</td></tr></table>

Peer review (iclr2020):
This paper considers the task of training an agent to play text-based computer games. One of the key challenges is the high-dimensional action space in these games, which poses a problem for many current methods. The authors propose to learn an LSTM-based decoder to output the action by greedily prediction one word at a time. They achieve this by training a sequence to sequence model on trajectories collected by running the game using a previously proposed exploration method (Go-Explore). While the results are promising, there might be limited novelty beyond training a sequence to sequence model on pre-collected trajectories. Further, the experiments are missing key elements in terms of proper comparison to baselines. Pros: 1. Nice idea for tackling the unbounded action space problem in text-based games. Cons: 1. The method depends on the assumption that we can get a set of trajectories with high rewards. This seems a pretty strong assumption. In fact, the authors use a smaller set of admissible actions in order to collect these trajectories in the first place - this seems to not be in line with the goal of solving the large action space problem. If we assume access to this admissible action function, why not just use it directly? 2. Some of the empirical results may not be fair comparisons (unless I*m missing something). For example, all the baselines for the CookingWorld games use -greedy exploration. Since the Go-Explore method assumes access to extra trajectories at the start, this doesn*t seem fair to the other baselines which may not observe the same high-reward trajectories. Other comments: 1. Do you use the game rewards to train/finetune the seq2seq model or is it only trained in a supervised fashion on the trajectories? (like an imitation learning setup) 2. How critical is the first step of producing high reward trajectories to the overall performance? Some more analysis or discusssion on this would be helpful to disentangle the contribution of GoExplore from the seq2seq action decoder.