Code for CoRL 2019 paper: HRL4IN: Hierarchical Reinforcement Learning for Interactive Navigation with Mobile Manipulators
Most common navigation tasks in human environments require auxiliary arm interactions, e.g. opening doors, pressing buttons and pushing obstacles away. This type of navigation tasks, which we call Interactive Navigation, requires the use of mobile manipulators: mobile bases with manipulation capabilities. Interactive Navigation tasks are usually long-horizon and composed of heterogeneous phases of pure navigation, pure manipulation, and their combination. Using the wrong part of the embodiment is inefficient and hinders progress. We propose HRL4IN, a novel Hierarchical RL architecture for Interactive Navigation tasks. HRL4IN exploits the exploration benefits of HRL over flat RL for long-horizon tasks thanks to temporally extended commitments towards subgoals. Different from other HRL solutions, HRL4IN handles the heterogeneous nature of the Interactive Navigation task by creating subgoals in different spaces in different phases of the task. Moreover, HRL4IN selects different parts of the embodiment to use for each phase, improving energy efficiency. We evaluate HRL4IN against flat PPO and HAC, a state-of-the-art HRL algorithm, on Interactive Navigation in two environments - a 2D grid-world environment and a 3D environment with physics simulation. We show that HRL4IN significantly outperforms its baselines in terms of task performance and energy efficiency. More information is available at https://sites.google.com/view/hrl4in.READ FULL TEXT VIEW PDF
Many Reinforcement Learning (RL) approaches use joint control signals
We present Interactive Gibson, the first comprehensive benchmark for tra...
This paper presents an approach for improving navigation in dynamic and
Model-based reinforcement learning (RL) enables an agent to learn world
We introduce Habitat 2.0 (H2.0), a simulation platform for training virt...
Hierarchical reinforcement learning has demonstrated significant success...
We have compared three common visualisations for hierarchical quantitati...
Code for CoRL 2019 paper: HRL4IN: Hierarchical Reinforcement Learning for Interactive Navigation with Mobile Manipulators
In recent years, agents learning with reinforcement have achieved new levels of proficiency, solving from complex games [silver2017mastering, gu2017deep] and video games [mnih2013playing, vinyals2017starcraft], to real-world robotic tasks [andrychowicz2018learning, kalashnikov2018qt]. In robotics, this success has mostly been restricted to either navigation [zhu2017target, tai2017virtual, chiang2019learning, faust2018prm] or manipulation with stationary arms [gu2017deep, vevcerik2017leveraging, levine2018learning, levine2016end, beyret2019dottodot]. The goal in navigation is to change the agent’s location without changing the environment’s configuration, whereas the goal in manipulation with stationary arms is to change the environment’s configuration without changing the agent’s location. However, many tasks in real human environments require an agent capable of achieving both, sometimes even simultaneously.
A common case is robot navigation in realistic human environments. This task requires interactions with the environment such as opening doors or pushing obstacles away. We call the family of navigation tasks that require interactions Interactive Navigation. Solving interactive navigation problems requires a mobile base with interaction capabilities, i.e. a mobile manipulator. With this embodiment, the agent can change its location and/or the configuration of the environment, by using only the base, the arm(s) or a combination of both. A naive use of the entire embodiment for every phase of the task is inefficient: large parts of the task can be solved with only the base or the arm. Therefore, an efficient solution for Interactive Navigation needs to, first, understand what phases of the task are pure navigation, manipulation or both, and second, solve these subtasks efficiently.
The structure mentioned above is well suited for a hierarchical reinforcement learning (HRL) solution, where the high level learns the phase type and sets a subgoal, and the low level learns to achieve it. However, most existing HRL approaches focus on subgoals that are in the same space as the final goal, e.g. intermediate locations towards a final location [nachum2018data, kulkarni2016hierarchical, sutton1999between] or intermediate joint configurations towards a final one [levy2018learning]. In our problem setup, however, the subgoals are heterogeneous: while the final goal is to move the base to the desired location, the intermediate subgoals may require the agent to move its end-effector to a given pose, e.g., to the location of a door handle.
In this work, we present HRL4IN, a hierarchical reinforcement learning solution for heterogeneous subtask spaces that can be applied to solve Interactive Navigation tasks. Our solution learns the most efficient use of pure navigation, pure manipulation, or their combination. The key for efficiency is to allow the high level to decide not only on the subgoals for the low level, but also on the capabilities that the low level is allowed to use to achieve it.
In summary, the main contributions of our work are: first, we propose HRL4IN, a novel learning algorithm suited for mobile manipulators. Second, our solution is, to the best of our knowledge, the first deep HRL approach for continuous control that shows success in heterogeneous tasks - tasks with multiple phases where the goals lay on different spaces. Third, we show that HRL4IN can solve Interactive Navigation tasks, navigation tasks that require interaction with the environment. We believe that explicitly defining and addressing this new type of navigation task is relevant and practically impactful for robots that aim to operate in human environments.
Hierarchical Reinforcement Learning: One of the hardest types of tasks for reinforcement learning is long-horizon tasks with sparse reward. Exploring the environment through random sequences of actions requires a prohibitive number of samples until a solution can be found. To alleviate this problem, researchers have proposed hierarchical RL to incorporate temporal abstraction [sutton1999between].
bacon2017option proposes an option-critic architecture that learns temporally extended options and their termination conditions, together with the policy over them. heess2016learning pretrains the low-level policy on simple tasks and then trains the high-level policy to output appropriate signals that modulate low-level behaviors. Our proposed method, HRL4IN, is closer to the hierarchical approaches where a high-level policy explores by setting subgoals for a low-level policy to achieve [kulkarni2016hierarchical, vezhnevets2017feudal, nachum2018data]. The subgoals can be set directly in the same space as the original goal [nachum2018data], or in an embedding space [vezhnevets2017feudal]. While the former is more data efficient, the latter has the potential to learn more informative representation. nachum2018near propose a way to balance between representation efficiency and capabilities.
Similar to [nachum2018data, levy2018learning], our subgoals represent a desired observation that the high-level policy wants to induce. However, we also exploit the fact that a heterogeneous task can be solved more efficiently by setting subgoals in different spaces during different phases of the task. Specifically, our high-level policy not only outputs a subgoal, but also selects a part of the embodiment to use for the current phase. Finally, unlike [kulkarni2016hierarchical], we do not manually construct subgoals apriori, which may require extensive domain knowledge, allowing the high-level policy to learn subgoals from scratch.
Interactive Navigation: While the literature on autonomous robot navigation is vast and prolific [desouza2002vision, bonin2008visual, chenbehavioral], less attention has been paid to navigation problems that require interactions with the environment, what we call Interactive Navigation. In the robot control literature, several papers have approached the problem of opening doors with mobile manipulators [peterson2000high, schmid2008opening, petrovskaya2007probabilistic, jain2009behavior]. However, these approaches focus on this single phase and not on the entire Interactive Navigation task.
stilman2005navigation studied interactive navigation from a geometric motion planning perspective. In their problem setup, the agent has to reason about the geometry and arrangement of obstacles to decide on a sequence of pushing/pulling actions to rearrange them to allow navigation. They named this type of problems Navigation Among Movable Objects (NAMO) [stilman2008planning, stilman2007manipulation, van2009path, levihn13]. Their solution requires knowledge of the geometry of the objects to plan, and the search problem is restricted to 2D space. Our work has two main differences from the previous line of work on NAMO. First, most NAMO solutions assume full observability of the environment as it is required for sampling-based motion planning. On the contrary, HRL4IN only assumes partial observability through virtual depth images. Second, NAMO solutions find a suitable plan in configuration space and rely on a separate controller for the execution of the plan. HRL4IN generates full policies that can not only overcome local minimum in the reward function but also directly output agent commands. The aforementioned properties of our solution render it model-free and end-to-end trainable.
The work by konidaris2011autonomous is close to us because it approaches Interactive Navigation as a hierarchical reinforcement learning problem. Their algorithm learns to sequence actions to solve simple tasks and reuses these sequences to solve tasks composed of simple subtasks. However, different from ours, their solution requires a set of predefined controllers as actions for the policy during training. The options are learned in the low-level space and reused during testing. HRL4IN directly controls lower level commands for the mobile manipulator, without hard-coded controllers.
Task and Motion Planning: Interactive Navigation and NAMO can alternatively be solved by task and motion planning (TaMP) methods. Task and motion planning solves long-horizon problems by combining task planner and motion planner. Task planner can reason over large sets of states and motion planner can compute exact paths for the robot to execute. An integration of task and motion planner often uses a hierarchical approach [wolfe2010combined]. Much of the work in TaMP factorizes the task into sub-problems and abstractions that the solutions learn to use. Existing TaMP approaches are capable of solving complicated mobile manipulation tasks [srivastava2014combined, garrett2018ffrob]. konidaris2018skills learns symbolic representations by interacting with an environment and observing the effects of the actions. wolfe2010combined build a system that reuses optimal solutions to state-abstracted sub-problems across the search space. Our work differs from the above as we don’t explicitly formulate our problem as TaMP but rather implicitly allow the high-level policy to learn to do the planning and the low-level policy to learn to do the control.
While most of the previous work on abstraction selection refers to state abstraction [van2014efficient] (mapping primitive state space to an abstract state space ), our work performs abstraction selection in the action space. The work by konidaris2008sensorimotor relates to us as they also learns abstractions for the action space.
We assume that the underlying problem can be modelled as a discrete-time goal-conditioned MDPs , where is the state space, is the goal space, is the action space, is the reward function, is the transition dynamics, and is the discount factor. In our work, is the space of final goals for the high level and subgoals for the low level. At each time step , an agent observes state , executes action sampled from policy , receives from the environment a new observation and a reward . In reinforcement learning, the goal of the agent is to maximize expected discounted return . To do so, a family of RL solutions use the policy gradient theorem [sutton2000policy] to optimize for the policy parameters . In this work, we adopt Proximal Policy Optimization Algorithms (PPO) [schulman2017proximal], a stable and widely adopted on-policy model-free policy gradient algorithm as the building block and baseline for HRL4IN.
Interactive navigation tasks are intrinsically multi-phase. In some phases the agent only needs to navigate using its base, or to interact with the environment using its arm. In other phases, it needs to move its base and arm in coordination. Using the base or arm when not needed can be counter-productive: it is less energy efficient and increases the probabilities of undesired collisions.
Based on these insights, we propose HRL4IN, a hierarchical reinforcement learning solution for Interactive Navigation that learns to select the elements of the embodiment that are necessary for each phase of the task. To do so, the high-level policy not only sets subgoals for the low-level policy, but also selects which part of the embodiment to use.
The main structure of HRL4IN is depicted in Fig. 1. Our solution is composed by two policies, a high-level (HL) and a low-level (LL) policy. Both policies use the same observations that contain both elements that can be changed by the agent (mutable) such as the poses of its base and arm, and elements that cannot be changed by the agent (immutable) such as the door location. A full description of the observation space is included in Sec. 5.2 and Appendix A. In the following, we explain the main characteristics of the high-level and the low-level policies.
The high-level policy acts at a different time scale from that of the original MDP: it acts at time steps that correspond to every time steps of the low-level policy, or fewer, if the low-level policy converges to the subgoal. The high-level policy receives an observation and generates an action , conditioned on the final goal, .
Similar to previous HRL approaches [nachum2018data, levy2018learning], one component of the high-level action, , is the subgoal . indicates a desired relative change of certain mutable components of the observation, denoted as , yielding a subgoal for the low-level policy to achieve. The subgoal is valid for time steps of the low-level policy, unless it achieves the subgoal within time steps ( is close enough to ), in which case HRL4IN queries the high-level policy for the next high-level action. To represent the subgoal with respect to the current observation , the original subgoal is updated by for every subsequent time step.
The other component of the high-level action, , is an embodiment selector . The embodiment selector is a discrete variable that decides which capabilities the low-level policy can use to achieve the current subgoal: navigation (base), manipulation (arm), or both. The selector plays two crucial roles. First, it is used to compute an action mask, , that deactivates some components of the embodiment that are not necessary for the current phase (i.e., base or arm) by forcing the velocity of certain joints to be 0. Second, it is used to compute a subgoal mask, , that determines the dimensions of the observation that the low-level policy should bring closer to the subgoal to obtain intrinsic reward. Using the embodiment selector, the high-level policy decides on the phase type within the Interactive Navigation task for a period of steps: navigation, manipulation or both.
The reward for a high-level action is the sum of the rewards from the environment during the or fewer than time steps of its execution.
The low-level policy, , acts at every discrete time step of the MDP. It also receives an observation and generates an action, , conditioned on the last high-level action, , which includes the subgoal (transformed to be relative to the current observation as explained above) and the action and subgoal masks derived from the embodiment selector .
The low-level actions, , are robot commands for the base and the arm. In our Interactive Navigation experiments, the exact form of these actions depends on the environment (see Sec. 5.1). The low-level actions are masked by the action mask derived the embodiment selector , which zeroes out the action components that are not necessary for the current task phase: .
We define the current subgoal distance as the distance between the current observation and the subgoal , masked by the subgoal mask derived from the high-level embodiment selector : , where is the dimension of . We train the low-level policy using intrinsic reward defined by . Intuitively, the intrinsic reward encourages the low-level policy to make progress towards the subgoal.
For each subgoal, the low-level policy has T time steps to achieve it. We consider a subgoal achieved if (discrete state space) or (continuous state space). When the subgoal is achieved or the low-level policy runs out of time, HRL4IN queries the high-level policy for the next .
In order to demonstrate the effectiveness of our approach, we build two environments with different levels of abstraction and physical complexity: Interactive ToyEnv and Interactive Gibson Environment. More details are explained below and included in Appendix A.
Interactive ToyEnv: Interactive ToyEnv is a 2D grid-world environment with discrete state and action space. It consists of cells, where each cell could be free space, wall, door, or occupied by the agent. The agent is a simplification of a mobile manipulator that can navigate around the environment and interact with the door using its arm. At each time step, the agent can choose among navigation and/or manipulation actions that result in a deterministic outcome. Navigation actions include turn-left, turn-right, go-forward and no-op. Manipulation actions include slide-up (the door), slide-down, and no-op. Manipulation actions are only effective (i.e., change the door state) when the agent is in the cell in front of the door and is facing the door. Several () consecutive slide-up are necessary to fully open the door. The observation space includes the global map, the door state, the goal, and proprioceptive information such as the pose of the agent.
Interactive Gibson Environment: Interactive Gibson Environment is a 3D environment with continuous state and action space. It is an evolution of the Gibson Environment [xia2018gibson] that offers fast rendering and physics simulation. The main difference is that in Interactive Gibson Environment, additional objects (e.g. doors) apart from the agent can be added and their dynamics can be simulated, which allows us to train agents for Interactive Navigation tasks. To simplify the contact physics, we assume the agent has a grasping primitive that grabs the door handle when the end-effector is close enough (distance under ). Grasping creates a ball joint between the end-effector and the door handle that constrains translation but allows rotation. Once the door is fully opened, the end-effector ends the grasp primitive and releases the door handle.
In Interactive Gibson Environment
, the agent controls the mobile manipulator JackRabbot composed by a six degrees of freedom (DoF) Kinova arm mounted on a non-holonomic two-wheeled Segway mobile base. The agent commands actions for the mobile manipulator in continuous action space: desired velocities for each joint, including the two wheels and five joints of the arm (seven DoF in total). The observations to the agent include a depth map, and proprioceptive information such as the poses of the base and the end-effector, the joint configurations, and the goal position.
In our experiments, we use PPO [schulman2017proximal]
as the policy gradient algorithm for both the high-level and the low-level policies of HRL4IN. When training both policies at the same time, the changing behavior of the low-level policy poses a non-stationary problem for the high-level policy: the same high-level action could result in different low-level actions, and therefore different state transitions and rewards, depending on the current training stage of the low-level policy. We believe an on-policy algorithm like PPO will stabilize the training process and encourage policy convergence. Details about the network structure and hyperparameters of HRL4IN and PPO can be found in AppendixB.2 and B.3.
To evaluate the performance of HRL4IN, we compare with two baselines: flat PPO and Hindsight Actor-Critic (HAC), a state-of-the-art HRL framework. Since the official implementation of HAC published by the authors does not support discrete action, we only compare HRL4IN with HAC in Interactive Gibson Environment. We run experiments in Interactive Gibson Environment and Interactive ToyEnv with 5 and 7 different random seeds, respectively.
Interactive ToyEnv: The goal in Interactive ToyEnv is defined by the discrete coordinates of a desired agent’s location, , where indicates a global world frame. The agent starts at a random location in the left room with a random orientation. Subgoals are defined as the desired relative change in the agent’s , location, orientation and door state, . The embodiment selector can assume three values: base-only, arm-only, and base+arm.
The reward is
, where the reward vector is defined asand the weights to sum them up are . The reward weights are defined in Appendix A. In the reward vector, indicates goal convergence, if there is no actuation, if there is arm or base actuation, and if there are base and arm actuations. An episode ends when the agent reaches the goal or runs out of time. We use time scale for the high-level policy.
Interactive Gibson Environment: The goals in Interactive Gibson Environment is defined by the continuous coordinates of a desired agent’s base location, . The agent starts at a fixed location with a random orientation and is assumed to converge to the goal when the distance between the agent base position and the goal is smaller than . Subgoals are defined as the desired relative change in the agent’s end-effector position in either 3D or 2D (xy-plane) space, depending on the embodiment selection. The embodiment selector can assume two values: base-only and base+arm. The threshold to assume subgoal convergence is .
The environment returns different rewards in three different phases of the Interactive Navigation task: moving towards the door, opening the door, and moving towards to the goal. The reward is , where the reward vector is defined as: and the weights to sum them up are . The reward weights are defined in Appendix A. In the reward vector, indicates goal convergence; is a small positive value if the agent is making progress for the current stage (e.g. getting closer to the door handle or opening the door partially); , where and are joint velocities of the base wheels and the arm joints; if the agent collides with the environment or itself. An episode ends when the agent reaches the goal, runs out of time or tips over. We use time scale (equal to 5 seconds of world time) for the high-level policy.
Interactive ToyEnv: Our experimental results for the Interactive ToyEnv are depicted in Fig. 3. While being initially less sample efficient, HRL4IN achieves higher reward than flat PPO. HRL4IN learns to solve the task for 5 out of 7 random seeds while flat PPO only does so for 3 out of 7 random seeds. Given the sparse reward, HRL4IN is able to explore more efficiently than flat PPO through temporal extended commitments towards subgoals. The plot also shows that once flat PPO learns to solve the task, the policy consistently repeats the successful strategy. HRL4IN, on the other hand, has less stable training process since the same high-level action could result in different executions by the low-level policy that is being trained simultaneously. Our best performing model achieves 100% success rate with average episode length of 19.2 steps. The optimal policy, computed analytically using the shortest path, takes on average 14 steps, so our solution is optimal.
We also analyze the probability of using different parts of the embodiment at different locations of the rooms (Fig. 2(b)). We observe that the high-level policy of HRL4IN learns to use only base at almost all locations except those near the door, where it uses both base and arm. It is energy-efficient to use only base when far away from the door because the agent cannot interact with it. The agent never uses arm-only because unless the agent is 1) at the position in front of the door and 2) facing it, it needs to use both base and arm to make progress towards the goal and arm-only is overall too restrictive. Fig. 4 illustrates three different phases of a sample trajectory of our best model.
Interactive Gibson Environment: Our experimental results for the Interactive Gibson Environment are depicted in Fig. 5. Similar to the pattern observed for the Interactive ToyEnv, the reward increases faster initially for flat PPO than for HRL4IN: flat PPO quickly starts to exploit the experiences from the initial exploration and gets stuck in a local maximum of return that corresponds to the scenario in which the agent moves towards the wall separating the initial room and the final room (closer to the final goal). On the contrary, HRL4IN performs a deeper exploration in subgoal directions and achieves much higher reward in the long run. The other baseline HAC completely fails to solve the task. Our hypothesis for HAC’s poor performance is that our task poses a difficult exploration problem in which there exists a discrepancy between the final goal distribution and the hindsight goal distribution. The subgoals that HAC learns to achieve during training does not contribute to achieving the final goal. We observe that most transitions sampled from the replay buffer include hindsight goals very close to the initial position and high virtual rewards, and few include real goals and low real rewards. Our best performing HRL4IN policies has close to 100% success rate after 50M training steps. A few failure cases are caused by the agent accidentally colliding with the door and closing it back on its way to the final goal.
We have run ablation studies to understand the effect of embodiment selection and different reward terms in Interactive Gibson Environment on model performance (Table 1). First, we ablate the embodiment selector for HRL4IN and find that it achieves comparable reward and success rate, but fails to save any energy. We compute energy in a symbolic sense since we can’t accurately measure energy consumption from the simulator: the agent consumes 2 units of energy when using base and arm, and 1 unit of energy when using only base or arm. Furthermore, we ablate each reward term except and observe that all reward terms are necessary for training a successful, energy-efficient policy: without and , the agent fails to solve the task; without , the agent fails to save energy or learn meaningful embodiment selection.
We also analyze the use of different parts of the embodiment at different locations of the environment (Fig. 4(b)). We plot the embodiment selections from 100 evaluation episodes on the room layout. We observe that the high-level policy learns to use only base most of the time to save energy. It sometimes uses base and arm when in front of the door to grasp the door handle. Then it tends to switch back to use only base again because now that the end-effector is attached to the door handle, moving the base is sufficient to open the door. Finally, it uses only base in the final room since the door is already open. Fig. 6 illustrates four different phases of a sample trajectory of our best model. More experimental results and policy visualizations can be found in Appendix C.
|Model||Reward Terms||Success Rate||Reward||Energy-Saving|
|HRL4IN (w/o emb. sel.)||All|
We presented HRL4IN, a hierarchical Reinforcement Learning solution for Interactive Navigation tasks, tasks that require motion of the base and the arm of a mobile manipulator. The high-level policy of HRL4IN sets subgoals in different subspaces (navigation and/or manipulation) for the low-level policy to achieve. Moreover, the high level can also decide on the part of the embodiment (base, arm, base and arm) to use at each temporally extended phase to perform deep exploration. We evaluate HRL4IN against flat PPO and HAC in two environments, a grid-world environment with discrete action space, and a complex, realistic robotics environment with continuous action space. Our results indicate that HRL4IN achieves higher reward and solve the tasks more successfully and efficiently. In future work, we would like to 1) examine other relevant tasks that require interactive navigation and 2) transfer the learned policy to the real world.
We acknowledge the support of Toyota Research Institute (“TRI”) but this article solely reflects the opinions and conclusions of its authors and not TRI or any other Toyota entity. Fei Xia would like to acknowledge the support of Stanford Graduate Fellowship.
Interactive ToyEnv: We use for our grid size. The room configuration is depicted in Fig. 2. The door state has a minimum value of 1 (closed) and a maximum value of 5 (open). The agent will increment the door state by 1 if it outputs slide-up at the front of the door while facing it and decrease by 1 if it outputs slide-down. The maximum episode length is 500.
The full observation includes:
agent orientation represented by yaw
cosine and sine of yaw
whether the agent is next to the door
a top-down global map with four channels: the first channel encodes the static environment, the second channel encodes agent’s position and orientation, the third channel encodes the goal position and the last channel encodes the door position and state.
The high-level policy only sets subgoals for the first three elements of the observation because the other elements are either immutable or redundant information that solely helps the neural networks to learn policies more efficiently.
The reward weights defined in Sec. 5.2 are and .
Interactive Gibson Environment: We used a room size of approximately . The room configuration is depicted in Fig. 2. The door is considered fully open if the door angle is more than . The maximum episode length is .
The full observation includes:
linear and angular velocity
roll, pitch, yaw of the base
joint space configuration, velocities and torques for wheels and arm joints
door angle and location
a flag that indicates collision or self-collision
a flat that indicates if the end-effector is currently grabbing to the door handle
cosine and sine of the robot’s base orientation, joint configurations for revolute joints, and door angle
a depth map clipped to a range of
For similar reasons as in the Interactive ToyEnv, the high-level policy only sets subgoals for the first element of the observation vector, the end-effector position . When the high level selects to use only the base embodiment (base-only), the coordinate is masked out for the computation of intrinsic reward by the associated subgoal mask and the desired velocities for all joints of the arm are set to zero by the associated action mask. The reward weights defined in Sec.5.2 are included in Table 6(a). The physics simulation parameters are included in Table 6(b).
A detailed pseudo-code of our HRL4IN solution is included below in Algorithm 1:
We implement the high-level and low-level policies as Gated Recurrent Neural Networks (GRU[chung2014empirical]
). For feature extraction, the observation space that contains spatial information (e.g. global map inInteractive ToyEnv and depth map in Interactive Gibson Environment
) are processed by three convolutional layers and one fully connected layer with ReLU activation after flattening. Other observations are concatenated together and processed by one fully connected layer with ReLU activation. Finally, the features from two branches are concatenated together and fed into the GRU. The output features from the GRU are used to estimate the value function and the action distribution. Note that the embodiment selection component of the high-level action is modeled as a discrete action (e.g. choose between base-only and base-arm). The action losses of the subgoal and the embodiment selector are aggregated together for back-propagation. We follow the same training procedure as PPO[schulman2017proximal] and the training hyper-parameters can be found below.
Table 3 summarizes the hyperparameters for HRL4IN.
|Learning Rate for High-Level Policy|
|Learning Rate for Low-Level Policy|
|Freeze High-Level Policy for First N Updates|
|Intrinsic Reward Scaling|
|Discount Factor for the High-Level policy|
|Discount Factor for the Low-Level policy|
|Subgoal Init. Std. Dev. (Gibson Env)|
|Subgoal Min. Std. Dev. (Gibson Env)|
|Subgoal Init. Std. Dev. (Toy Env)|
|Subgoal Min. Std. Dev. (Toy Env)|
Table 4 summarizes the hyperparameters for PPO.
|PPO Clip Parameter|
|Num of Mini-Batches for PPO|
|Value Loss Coefficient|
|Max Gradient Norm|
|Hidden layer size for the GRUs|
|Use Linear LR Decay||True|
|Use Generalized Advantage Estimation (GAE)||True|
|Tau for GAE||0.95|
|Primitive Action Init. Std. Dev.||0.33|
|Primitive Action Min. Std. Dev.||0.1|
We normalize the observation space and the action space to during training and inference.
In this section we show some additional results from the experiments described in Sec 5.
Fig. 7 shows additional training curves of success rate on Interactive ToyEnv and Interactive Gibson Environment, which complements Fig. 2(a) and Fig. 4(a). It shows that our approach is able to achieve higher final success rate against both baselines in both environments.
Fig. 8 depicts the training curves of our ablation studies, which complements Table 1. We observe that and are required for the agent to solve the task, and and embodiment selection are required for the agent to learn embodiment selection and solve the task efficiently.
Fig. 9 shows additional visualizations of subgoals set by the high-level policy of HRL4IN in Interactive Gibson Environment. These visualizations illustrate that the high level policy of HRL4IN learns to command meaningful subgoals to guide the low-level policy to achieve different types of subtasks, including pure navigation, reaching for the door handle and pulling the door open.