Learning to Navigate Unseen Environments: Back Translation with Environmental Dropout

NAACL 2019  ·  Hao Tan, Licheng Yu, Mohit Bansal ·

A grand goal in AI is to build a robot that can accurately navigate based on natural language instructions, which requires the agent to perceive the scene, understand and ground language, and act in the real-world environment. One key challenge here is to learn to navigate in new environments that are unseen during training... Most of the existing approaches perform dramatically worse in unseen environments as compared to seen ones. In this paper, we present a generalizable navigational agent. Our agent is trained in two stages. The first stage is training via mixed imitation and reinforcement learning, combining the benefits from both off-policy and on-policy optimization. The second stage is fine-tuning via newly-introduced 'unseen' triplets (environment, path, instruction). To generate these unseen triplets, we propose a simple but effective 'environmental dropout' method to mimic unseen environments, which overcomes the problem of limited seen environment variability. Next, we apply semi-supervised learning (via back-translation) on these dropped-out environments to generate new paths and instructions. Empirically, we show that our agent is substantially better at generalizability when fine-tuned with these triplets, outperforming the state-of-art approaches by a large margin on the private unseen test set of the Room-to-Room task, and achieving the top rank on the leaderboard. read more

PDF Abstract NAACL 2019 PDF NAACL 2019 Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Vision-Language Navigation Room2Room R2R+EnvDrop spl 0.61 # 1
Vision and Language Navigation VLN Challenge Back Translation with Environmental Dropout (no beam search) success 0.51 # 68
length 11.66 # 62
error 5.23 # 40
oracle success 0.59 # 70
spl 0.47 # 47
Vision and Language Navigation VLN Challenge null success 0.69 # 11
length 686.82 # 8
error 3.26 # 91
oracle success 0.99 # 1
spl 0.01 # 90

Methods


No methods listed for this paper. Add relevant methods here