Distilling LLM Prior to Flow Model for Generalizable Agent's Imagination in Object Goal Navigation
This addresses the problem of limited generalization in object navigation for AI agents by incorporating uncertainty modeling, though it appears incremental as it builds on existing flow models and LLM priors.
The paper tackles the Object Goal Navigation task by proposing GOAL, a generative flow-based framework that models semantic distributions of indoor environments using LLM-enriched priors, achieving state-of-the-art performance on MP3D and Gibson benchmarks with strong generalization to HM3D.
The Object Goal Navigation (ObjectNav) task challenges agents to locate a specified object in an unseen environment by imagining unobserved regions of the scene. Prior approaches rely on deterministic and discriminative models to complete semantic maps, overlooking the inherent uncertainty in indoor layouts and limiting their ability to generalize to unseen environments. In this work, we propose GOAL, a generative flow-based framework that models the semantic distribution of indoor environments by bridging observed regions with LLM-enriched full-scene semantic maps. During training, spatial priors inferred from large language models (LLMs) are encoded as two-dimensional Gaussian fields and injected into target maps, distilling rich contextual knowledge into the flow model and enabling more generalizable completions. Extensive experiments demonstrate that GOAL achieves state-of-the-art performance on MP3D and Gibson, and shows strong generalization in transfer settings to HM3D. Codes and pretrained models are available at https://github.com/Badi-Li/GOAL.