Let's Reward Step-by-Step: Step-Aware Contrastive Alignment for Vision-Language Navigation in Continuous Environments
Organizations: College of Computer Science and Technology, Zhejiang University, Hangzhou, China
Abstract
Vision-Language Navigation in Continuous Environments (VLN-CE) requires agents to learn complex reasoning from long-horizon human interactions. While Multi-modal Large Language Models (MLLMs) have driven recent progress, current training paradigms struggle to balance generalization capability, error recovery and training stability. Specifically, (i) policies derived from SFT suffer from compounding errors, struggling to recover from out-of-distribution states, and (ii) Reinforcement Fine-Tuning (RFT) methods e.g. GRPO are bottlenecked by sparse outcome rewards. Their binary feedback fails to assign credit to individual steps, leading to gradient signal collapse in failure dominant batches. To address these challenges, we introduce Step-Aware Contrastive Alignment (SACA), a framework designed to extract dense supervision from imperfect trajectories. At its core, the Perception-Grounded Step-Aware auditor evaluates progress step-by-step, disentangling failed trajectories into valid prefixes and exact divergence points. Leveraging these signals, Scenario-Conditioned Group Construction mechanism dynamically routes batches to specialized resampling and optimization strategies. Extensive experiments on VLN-CE benchmarks demonstrate that SACA achieves state-of-the-art performance.
Figures & tables
| Method | Observation Encoder | R2R-CE Val-Unseen | RxR-CE Val-Unseen | |||||||||
| Pano. | Odo. | Depth | S.RGB | NE | OS | SR | SPL | NE | SR | SPL | nDTW | |
| HPN+DN ∗ [ 24 ] | 6.31 | 40.0 | 36.0 | 34.0 | - | - | - | - | ||||
| CMA ∗ [ 20 ] | 6.20 | 52.0 | 41.0 | 36.0 | 8.76 | 26.5 | 22.1 | 47.0 | ||||
| VLN BERT ∗ [ 20 ] | 5.74 | 53.0 | 44.0 | 39.0 | 8.98 | 27.0 | 22.6 | 46.7 | ||||
| Sim2Sim ∗ [ 25 ] | 6.07 | 52.0 | 43.0 | 36.0 | - | - | - | - | ||||
| GridMM ∗ [ 53 ] | 5.11 | 61.0 | 49.0 | 41.0 | - | - | - | - | ||||
| Row | Components | R2R-CE Val-Unseen | RxR-CE Val-Unseen | ||||||||
| SS | AFR | RR | NE | OS | SR | SPL | NE | SR | SPL | nDTW | |
| 0 | SFT Baseline [ 55 ] | 5.43 | 62.5 | 52.8 | 47.2 | 6.72 | 48.6 | 42.5 | 60.2 | ||
| 1 | 5.39 | 55.3 | 54.1 | 48.8 | 6.54 | 49.9 | 43.0 | 60.3 | |||
| 2 | 5.21 | 63.2 | 55.4 | 49.6 | 6.15 | 51.8 | 44.3 | 60.7 | |||
| 3 | 4.82 | 64.1 | 58.2 | 52.4 | 5.48 | 56.4 | 47.6 | 61.3 | |||
| 4 | 4.57 | 64.9 | 60.3 | 55.1 | 4.90 | 60.3 | 49.8 | 62.1 | |||
| Model Variant | R2R-CE Val-Unseen | RxR-CE Val-Unseen | ||||||
|---|---|---|---|---|---|---|---|---|
| NE | OS | SR | SPL | NE | SR | SPL | nDTW | |
| SACA | 4.57 | 64.9 | 60.3 | 55.1 | 4.90 | 60.3 | 49.8 | 62.1 |
| Ablation on Objective Constraints | ||||||||
| w/o Consistency Align ( ) | 4.89 | 63.8 | 58.0 | 51.6 | 5.34 | 57.1 | 46.2 | 60.5 |
| w/o Contrastive Correct ( ) | 4.98 | 63.5 | 57.3 | 52.8 | 5.51 | 56.4 | 46.5 | 59.4 |
| Ablation on Robustness Mechanisms | ||||||||
| Group Size ( ) | Max Repairs ( ) | R2R-CE Val-Unseen | RxR-CE Val-Unseen | ||||||
|---|---|---|---|---|---|---|---|---|---|
| NE | OS | SR | SPL | NE | SR | SPL | nDTW | ||
| 4 | 3 | 4.92 | 62.1 | 56.4 | 51.2 | 5.30 | 55.2 | 45.3 | 59.5 |
| 8 | 3 | 4.57 | 64.9 | 60.3 | 55.1 | 4.90 | 60.3 | 49.8 | 62.1 |
| 16 | 3 | 4.59 | 64.7 | 60.1 | 54.8 | 4.93 | 59.9 | 49.5 | 61.8 |
| 8 | 1 | 4.81 | 63.5 | 58.1 | 53.4 | 5.15 | 57.5 | 47.1 | 60.2 |
| 8 | 5 | 4.60 | 64.8 | 60.0 | 54.9 | 4.92 | 60.1 | 49.6 | 62.0 |
| Row | Perception Modules | R2R-CE Val-Unseen | RxR-CE Val-Unseen | ||||||||
| Global | BBox | Mask | NE | OS | SR | SPL | NE | SR | SPL | nDTW | |
| 1 | 4.95 | 63.2 | 57.6 | 52.1 | 5.45 | 56.5 | 46.2 | 59.5 | |||
| 2 | 4.71 | 64.1 | 59.2 | 54.0 | 5.11 | 58.7 | 48.3 | 61.1 | |||
| 3 | 4.57 | 64.9 | 60.3 | 55.1 | 4.90 | 60.3 | 49.8 | 62.1 | |||
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Trajectory Phase | Temporal Condition | Hard Mask | Dominant Reward Signal ( ) | Optimization Objective |
|---|---|---|---|---|
| Valid Prefix | Dense Positive (Continuous Baseline + Peaks) | (Behavior Cloning) | ||
| Divergence Point | Sharp Negative (Explicit Penalty) | (Contrastive Correction) | ||
| Resampled Suffix | Dense Positive (Continuous Baseline + Peaks) | (Suffix Cloning) |
| General & SFT Configuration | SACA Hyperparameters | ||
|---|---|---|---|
| Hyperparameter | Value | Hyperparameter | Value |
| General Infrastructure | Scenario & Grouping | ||
| Policy Model | Video-LLaVA-8B | Group size ( ) | 8 |
| Max Sequence Length | 4096 | Valid-prefix ratio ( ) | 0.5 |
| Number of GPUs | 8 | Max resampling attempts ( ) | 3 |
| Parallel Strategy | DDP | PGSA Auditor Settings | |
Vision-Language Navigation
Vision-Language Navigation studies how embodied agents use natural-language instructions and visual observations to plan and execute routes.
See all 310 papers