Bayesian Relational Memory for Semantic Visual Navigation
Bayesian Relational Memory for Semantic Visual Navigation
复制标题
DOI:
10.1109/iccv.2019.00286
复制
发表时间:
2019-09
期刊:
影响因子:
--
通讯作者:
Yi Wu;Yuxin Wu;Aviv Tamar;Stuart J. Russell;Georgia Gkioxari;Yuandong Tian
中科院分区:
文献类型:
--
作者:
Yi Wu;Yuxin Wu;Aviv Tamar;Stuart J. Russell;Georgia Gkioxari;Yuandong Tian
We introduce a new memory architecture, Bayesian Relational Memory (BRM), to improve the generalization ability for semantic visual navigation agents in unseen environments, where an agent is given a semantic target to navigate towards. BRM takes the form of a probabilistic relation graph over semantic entities (e.g., room types), which allows (1) capturing the layout prior from training environments, i.e., prior knowledge, (2) estimating posterior layout at test time, i.e., memory update, and (3) efficient planning for navigation, altogether. We develop a BRM agent consisting of a BRM module for producing sub-goals and a goal-conditioned locomotion module for control. When testing in unseen environments, the BRM agent outperforms baselines that do not explicitly utilize the probabilistic relational memory structure.