IEEE Trans Pattern Anal Mach Intell. 2024 Dec;46(12):11200-11215. doi: 10.1109/TPAMI.2024.3455252. Epub 2024 Nov 6.
We propose an end-to-end visuomotor navigation framework that leverages Neural Radiance Fields (NeRF) for spatial cognition. To the best of our knowledge, this is the first effort to integrate such implicit spatial representation with embodied policy end-to-end for cognitive decision-making. Consequently, our system does not necessitate modularized designs nor transformations into explicit scene representations for downstream control. The NeRF-based memory is constructed online during navigation, without relying on any environmental priors. To enhance the extraction of decision-critical historical insights from the rigid and implicit structure of NeRF, we introduce a spatial information extraction mechanism named Structural Radiance Attention (SRA). SRA empowers the agent to grasp complex scene structures and task objectives, thus paving the way for the development of intelligent behavioral patterns. Our comprehensive testing in image-goal navigation tasks demonstrates that our approach significantly outperforms existing navigation models. We demonstrate that SRA markedly improves the agent's understanding of both the scene and the task by retrieving historical information stored in NeRF memory. The agent also learns exploratory awareness from our pipeline to better adapt to low signal-to-noise memory signals in unknown scenes. We deploy our navigation system on a mobile robot in real-world scenarios, where it exhibits evident cognitive capabilities while ensuring real-time performance.
我们提出了一个端到端的视觉运动导航框架,利用神经辐射场(NeRF)进行空间认知。据我们所知,这是首次将这种隐式空间表示与具身策略端到端集成用于认知决策的尝试。因此,我们的系统不需要模块化设计,也不需要转换为显式场景表示进行下游控制。基于 NeRF 的记忆是在导航过程中在线构建的,不依赖于任何环境先验。为了增强从 NeRF 的刚性和隐式结构中提取决策关键历史见解的能力,我们引入了一种名为结构辐射注意力(SRA)的空间信息提取机制。SRA 使代理能够掌握复杂的场景结构和任务目标,从而为智能行为模式的发展铺平道路。我们在图像目标导航任务中的综合测试表明,我们的方法明显优于现有的导航模型。我们通过从 NeRF 内存中检索存储的历史信息来证明 SRA 显著提高了代理对场景和任务的理解。该代理还从我们的管道中学习探索意识,以更好地适应未知场景中低信噪比的内存信号。我们在真实场景中的移动机器人上部署了我们的导航系统,它在确保实时性能的同时表现出明显的认知能力。