Source-linked AI summary

Neural Modular Control for Embodied Question Answering

Abhishek Das, Georgia Gkioxari, Stefan Lee, Devi Parikh, Dhruv Batra

arXiv:1810.11181v2cs.AIcs.CLcs.CVcs.LG

TL;DR

Embodied question answering requires navigating from egocentric vision over long horizons to find an entity and answer a question. NMC divides this work between a master policy that proposes semantic subgoals and specialized sub-policies that execute them. On EQA in House3D, it outperforms prior work in navigation and question answering.

  • Problem

    EmbodiedQA requires finding a question-relevant entity from egocentric vision in a novel environment, amid partial observability, long planning horizons, and sparse final-answer rewards.

    Source: §1 · p. 1
  • Method

    NMC uses a master policy to propose semantic subgoals and specialized sub-policies to execute them, with behavior cloning followed by reinforcement learning.

    Source: §1 · p. 1, §5 · p. 8
  • Results

    NMC significantly outperforms prior work in both navigation and question answering on EQA in House3D.

    Source: §1 · p. 2
  • Takeaways & Limitations

    Behavior cloning combined with reinforcement learning is more sample-efficient than either regime alone, with cloning providing dense navigation feedback and reinforcement learning enabling recovery from errors.

    Source: §4 · p. 7, §5 · p. 8

    Shortest-path states and actions provide supervision on training environments but cannot be used during test evaluation, where the agent operates from egocentric vision alone.

    Source: §3.2 · p. 5

Abstract

from arXiv · show

We present a modular approach for learning policies for navigation over long planning horizons from language input. Our hierarchical policy operates at multiple timescales, where the higher-level master policy proposes subgoals to be executed by specialized sub-policies. Our choice of subgoals is compositional and semantic, i.e. they can be sequentially combined in arbitrary orderings, and assume human-interpretable descriptions (e.g. 'exit room', 'find kitchen', 'find refrigerator', etc.). We use imitation learning to warm-start policies at each level of the hierarchy, dramatically increasing sample efficiency, followed by reinforcement learning. Independent reinforcement learning at each level of hierarchy enables sub-policies to adapt to consequences of their actions and recover from errors. Subsequent joint hierarchical training enables the master policy to adapt to the sub-policies. On the challenging EQA (Das et al., 2018) benchmark in House3D (Wu et al., 2018), requiring navigating diverse realistic indoor environments, our approach outperforms prior work by a significant margin, both in terms of navigation and question answering.

1 Introduction

The introduction motivates semantic subgoals for long-horizon embodied navigation and presents a modular hierarchy that divides planning from low-level execution. It reports improved navigation and question answering on EQA, alongside benefits from reinforcement learning after behavior cloning.

1 Introduction

  • EmbodiedQA requires an agent to locate a question-relevant entity from egocentric vision across potentially many rooms, with partial observability, long horizons, and sparse final-answer rewards.The agent may execute hundreds of primitive actions before receiving a reward.Source: §1 · p. 1
  • The Neural Modular Controller uses a master policy to select semantic subgoals and specialized sub-policies to execute variable-length primitive actions.For example, a plan can include exiting a room, finding the relevant room and object, then answering.Source: §1 · pp. 1–2
  • Semantic subgoals compress planning: the master makes orders of magnitude fewer decisions than a flat primitive-action policy, allowing the answering reward to influence high-level control more easily.The hierarchy separates semantic subgoal selection from sub-policy execution, shortening learning horizons.Source: §1 · p. 2
  • Modules can be pretrained independently before combination, and semantic master-policy predictions expose the agent’s current intent more clearly than primitive actions.The introduction also reports that reinforcement learning after behavior cloning improves performance over either training regime alone.Source: §1 · p. 2
  • NMC significantly outperforms prior work in both navigation and question answering on the EQA benchmark in House3D.The agent navigates closer to target objects and answers questions correctly more often.Source: §1 · p. 2

2 Related Work

The paper relates its approach to hierarchical learning, neural module networks, policy sketches, grounded language learning, and earlier embodied question-answering systems. Its distinctions include learned subgoal generation at test time, active reevaluation under partial observability, and shared navigation modules across target objects.

Hierarchical Reinforcement and Imitation Learning

  • NMC instantiates the options-style division between a global policy proposing subgoals and local policies pursuing them, using known task structure with hierarchical behavior cloning and actor-critic learning.The authors contrast this with automatic subgoal discovery, arguing that simple heuristics can effectively break long-range planning into sequential subgoals when problem structure is known.Source: §2 · p. 2

Neural Module Networks and Policy Sketches

  • Unlike neural module networks that access a fully observable image, NMC sub-policies use first-person RGB, making active reevaluation of subgoals essential.Unlike policy sketches, NMC learns to produce subgoals rather than receiving a task-specific sketch at test time.Source: §2 · pp. 2–3
  • NMC sub-policies accept room and object arguments, sharing parameters across similar navigation tasks instead of learning a separate policy for each target.The paper contrasts this with policy sketches, which assume subgoal supervision at test time; NMC uses demonstrations and plans during training but must generate its own subgoals at test time.Source: §2 · p. 3

Grounded Language Learning

  • The master grounds question words into semantic subgoals, while sub-policies ground those targets into primitive actions and raw pixels.Both levels are neural control policies trained end-to-end.Source: §2 · p. 3

Embodied Question-Answering Agents

  • Compared with prior hierarchical EQA systems, NMC uses human-interpretable semantic subgoals and is described as navigating longer paths spanning multiple rooms.The cited prior systems include work in House3D and AI2-THOR.Source: §2 · p. 3

3 Neural Modular Control

Neural Modular Control (NMC) divides navigation into semantic subgoals selected by a master policy and primitive actions executed by specialized sub-policies. The paper derives expert subgoal demonstrations from shortest paths, then trains the hierarchy with behavior cloning and reinforcement learning.

3.1 Hierarchical Policy

  • NMC alternates between a master policy that selects subgoals and sub-policies that execute primitive actions until stopping or reaching an action limit.The sub-policy’s terminal state becomes the master policy’s state for selecting the next subgoal.Source: §3.1 · p. 3
  • The hierarchy defines 64 subgoals across exit-room, find-room, find-object, and answer, with room and object arguments for the two search tasks.The 50 object targets and 12 room types are arguments; exit-room and answer take none.Source: §3.1 · p. 4
  • Sharing task parameters across argument-specific sub-policies is intended to make learning more sample-efficient than training a separate policy for every task-argument combination.Source: §3.1 · p. 4

3.2 Hierarchical Behavior Cloning from Expert Trajectories

  • Expert subgoal sequences and their temporal segments are extracted from shortest paths using room transitions and House3D bounding boxes.The shortest paths provide supervision on training environments; agents must navigate from egocentric vision alone at test time.Source: §3.2 · p. 5, Figure 2 · p. 5
  • Hierarchical behavior cloning trains the master and sub-policies as independent supervised problems, with the master effectively assuming perfect sub-policies.The expert paths are segmented into subgoal decisions and sub-policy trajectories before cloning.Source: §3.2 · pp. 5–6

3.3 Asynchronous Advantage Actor-Critic (A3C) Training

  • After behavior cloning, the policies are fine-tuned with reinforcement learning, first independently and then jointly so they can handle errors and coordinate.The master is subsequently trained using the learned sub-policies rather than expert subgoal trajectories.Source: §3.3 · p. 6
  • The reward combines question-answering success with distance-based progress rewards, while sub-policies also receive a small collision penalty.Sub-policy terminal rewards indicate successful completion of their specialized goals.Source: §3.3 · p. 6
  • BC+A3C significantly outperforms behavior cloning or A3C alone on the sub-policy tasks.The figure reports success rates over training iterations averaged across five runs.Source: Figure 3 · p. 7

4 Experiments and Results

On the EQA v1 test set, the final NMC model outperforms PACMAN on the reported navigation and answering measures. The experiments also compare sub-policy training regimes and assess how closely the master policy reproduces expert subgoal sequences.

Dataset

  • The EQA v1 benchmark uses non-overlapping training, validation, and test environments to assess generalization to novel environments.Source: §4 · p. 7

Evaluating sub-policies

  • The sub-policy evaluation measures success on 68 validation environments, with policies started in suitable rooms and limited to 50 steps or termination.Source: §4 · p. 7
  • BC+A3C sub-policies significantly outperform policies trained with BC or A3C alone, nearly doubling BC success rates for some tasks.BC learns faster than A3C from scratch on all tasks and achieves higher success rates for Exit-room and Find-room.Source: §4 · p. 7

Evaluating master policy

  • The master policy achieves approximately 48% intersection-over-union with ground-truth subgoal sequences on the validation set.The authors characterize this as a strict metric because a different subgoal sequence can still reach the target and answer correctly.Source: §4 · p. 7

Evaluating NMC

  • NMC (BC+A3C) outperforms PACMAN with higher d∆, lower dT, and higher answering accuracy.Table 2 reports navigation and answering metrics on the EQA v1 test set; the comparison uses the final NMC model.Source: Table 2 · p. 8, §4 · p. 8

5 Conclusion

The Neural Modular Controller uses a master policy to propose semantic subgoals and specialized sub-policies to execute them. Combining behavior cloning with reinforcement learning supports sample-efficient training and recovery, and NMC outperforms prior work in both navigation and question answering on EQA v1.

  • NMC outperforms prior work in both navigation and question answering on the EQA v1 dataset.Source: §5 · p. 8
  • A master policy proposes semantic subgoals from the question, while specialized sub-policies execute each task.Source: §5 · p. 8
  • Behavior cloning and reinforcement learning together are more sample-efficient than either regime alone; cloning provides dense navigation feedback, while RL supports adapting to action consequences and recovering from errors.Source: §5 · p. 8
Loading 1810.11181v2…