We introduce a new RL problem where the agent is required to generalize to a
previously-unseen environment characterized by a subtask graph which describes
a set of subtasks and their dependencies. Unlike existing hierarchical
multitask RL approaches that explicitly describe what the agent should do at a
high level, our problem only describes properties of subtasks and relationships
among them, which requires the agent to perform complex reasoning to find the
optimal subtask to execute. To solve this problem, we propose a neural subtask
graph solver (NSGS) which encodes the subtask graph using a recursive neural
network embedding. To overcome the difficulty of training, we propose a novel
non-parametric gradient-based policy, graph reward propagation, to pre-train
our NSGS agent and further finetune it through actor-critic method. The
experimental results on two 2D visual domains show that our agent can perform
complex reasoning to find a near-optimal way of executing the subtask graph and
generalize well to the unseen subtask graphs. In addition, we compare our agent
with a Monte-Carlo tree search (MCTS) method showing that our method is much
more efficient than MCTS, and the performance of NSGS can be further improved
by combining it with MCTS.

提出了一种新的强化学习问题，该问题需要代理在一个由子任务图描述的先前未见的环境中进行泛化，我们提出了一种神经子任务图求解器，通过使用递归神经网络嵌入来编码子任务图，并用基于梯度的策略，图奖励传播来预训练它，进一步通过演员 - 评论家方法微调它，实验结果表明，我们的代理可以进行复杂的推理，找到执行子任务图的近似最优方式，并且在未见的子任务图上具备很好的泛化性能。