atlas

Research map / Reinforcement learning / Imitation learning

Vision-Language Navigation with Energy-Based Policy

NeurIPS 2024 · Reinforcement learning › Imitation learning · approach: Imitation combined with RL · 11 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Energy-based Navigation Policy (ENP), which models the joint state-action distribution of VLN agents with an energy-based model trained via persistent contrastive divergence and SGLD, and plugs it into existing VLN architectures.

Problem

Vision-language navigation agents trained by behavioural cloning suffer from compounding errors and covariate shift, and fail to match the expert's state-action distribution over full trajectories.

Approach

Reinterpret the VLN action classifier logits as negative energies over state-action pairs, decompose the joint log-likelihood into a cross-entropy action loss plus a marginal state energy term, and estimate its gradient with SGLD sampling initialized from a persistent marginal state memory. The objective is shown to equal forward-KL occupancy-measure matching with the expert.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Enp joint state-action policy instead of EnvDrop behavioral cloningEnvDrop · R2R val unseenSR: 52 -> 53Helped
Use Enp joint state-action policy instead of VLN↻BERTVLN↻BERT · R2R test unseenSR: 63 -> 65Helped
Use Enp joint state-action policy instead of DUETDUET · R2R test unseenSR: 69 -> 71Helped
Use Enp joint state-action policy instead of VLN↻BERTVLN↻BERT · REVERIE val unseenSR: 35.02 -> 35.80Helped
Use Enp joint state-action policy instead of ETPNavETPNav · RxR-CE val unseenNDTW: 61.90 -> 62.97Helped
Add marginal state distribution matching loss to action loss (ablation)Enp–DUET · R2R valSR: 72 -> 74Helped
Add marginal state distribution matching loss to action loss (ablation)Enp–VLN↻BERT · R2R valSR: 61 -> 65Helped
Set SGLD step size ε=1 instead of ε=1.5 (ablation)Enp–VLN↻BERT · R2R valSR: 65 -> 64Hurt
Set SGLD step size ε=2 instead of ε=1.5 (ablation)Enp–VLN↻BERT · R2R valSR: 65 -> 64Hurt
Increase SGLD noise variance from 0.01 to 0.1 (ablation)Enp–VLN↻BERT · R2R valSR: 65 -> 63Hurt
Reduce SGLD iterations from I=15 to I=5 (ablation)Enp–VLN↻BERT · R2R valSR: 65 -> 64Hurt
Increase SGLD iterations from I=15 to I=20 (ablation)Enp–VLN↻BERT · R2R valSR: 65 -> 65No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: R2R REVERIE R2R-CE RxR-CE

Benchmarks: R2R REVERIE R2R-CE RxR-CE

Metrics: SR SPL NE TL OSR RGS RGSPL NDTW SDTW

Baselines: Seq2Seq SF AuxRN HAMT BEVBert VER BSG LANA HNR AIRL

Builds on: EnvDrop VLN-BERT DUET ETPNav AIRL

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy