用分层记忆系统提升城市导航中语言与视觉的结合能力
Mem4Nav: Boosting Vision-and-Language Navigation in Urban Environments with a Hierarchical Spatial-Cognition Long-Short Memory System
- 构建分层空间认知记忆:稀疏八叉树+语义拓扑图,实现细粒度定位与高层连接
- 双记忆机制:短期记忆实时避障,长期记忆无损回溯历史观测,提升路径规划
- 适配多种导航模型,显著提升任务完成率和路径质量,适合复杂城市环境研究
在大规模城市环境中进行视觉-语言导航(VLN)要求智能体将语言指令与复杂场景对齐,并在长时间跨度内回忆相关经验。以往的模块化流程虽具可解释性,但缺乏统一记忆;而端到端的多模态大模型(MLLM)虽能融合视觉与语言,却受限于固定上下文窗口和隐式空间推理。我们提出 extbf{Mem4Nav},一种分层空间-认知长短时记忆系统,可增强任意VLN主干网络。该系统融合稀疏八叉树实现细粒度体素索引,以及语义拓扑图实现高层地标连通性,二者均以可训练记忆令牌嵌入。长期记忆(LTM)在八叉树与图节点上压缩并保留历史观测,短期记忆(STM)则以相对坐标缓存近期多模态输入,用于实时避障与局部规划。每一步中,STM检索快速剪枝动态上下文;当需更深层历史时,通过无损解码重建过往嵌入。在Touchdown与Map2Seq数据集上,基于三种主干模型(模块化、基于提示的LLM、基于步进注意力的MLLM)评估,Mem4Nav实现7-13个百分点的任务完成率提升,足够降低SPD,且nDTW改善超10个百分点。消融实验证实分层地图与双记忆模块不可或缺。代码已开源:https://github.com/tsinghua-fib-lab/Mem4Nav。
原文摘要 · Abstract (English)
Vision-and-Language Navigation (VLN) in large-scale urban environments requires embodied agents to ground linguistic instructions in complex scenes and recall relevant experiences over extended time horizons. Prior modular pipelines offer interpretability but lack unified memory, while end-to-end (M)LLM agents excel at fusing vision and language yet remain constrained by fixed context windows and implicit spatial reasoning. We introduce \textbf{Mem4Nav}, a hierarchical spatial-cognition long-short memory system that can augment any VLN backbone. Mem4Nav fuses a sparse octree for fine-grained voxel indexing with a semantic topology graph for high-level landmark connectivity, storing both in trainable memory tokens embedded via a reversible Transformer. Long-term memory (LTM) compresses and retains historical observations at both octree and graph nodes, while short-term memory (STM) caches recent multimodal entries in relative coordinates for real-time obstacle avoidance and local planning. At each step, STM retrieval sharply prunes dynamic context, and, when deeper history is needed, LTM tokens are decoded losslessly to reconstruct past embeddings. Evaluated on Touchdown and Map2Seq across three backbones (modular, state-of-the-art VLN with prompt-based LLM, and state-of-the-art VLN with strided-attention MLLM), Mem4Nav yields 7-13 pp gains in Task Completion, sufficient SPD reduction, and >10 pp nDTW improvement. Ablations confirm the indispensability of both the hierarchical map and dual memory modules. Our codes are open-sourced via https://github.com/tsinghua-fib-lab/Mem4Nav.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。