AR Forcing: Towards Long-Horizon Robot Navigation World Model
TLDR
AR Forcing trains diffusion navigation world models autoregressively to reduce distribution shift, improving long-horizon prediction consistency and trajectory accuracy.
Reasoning
Strengths: Addresses a key distribution shift problem in diffusion-based navigation world models with a simple, plug-in training strategy that requires no extra losses or discriminators, validated on multiple real-world datasets. Weaknesses: Limited to robot navigation domain; no comparison to non-diffusion methods or analysis of computational overhead; code not yet released.
Read-first score
Read-first score 62.1, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 43.
Field roles
Rank sensitivity
Stability: volatile; rank range: 380.
Keyword Scores
Deep Analysis
Innovations
- AR Forcing: an autoregressive training strategy for diffusion-based robot navigation world models that integrates standard diffusion loss into the autoregressive training loop.
- The method uses the model's own predictions to update context and optimize the single step noise prediction objective, explicitly exposing the model to inference state distribution during training.
- No additional discriminators or distribution-matching losses are required; the original diffusion framework and sampler are retained.
Methodology
AR Forcing trains a diffusion model autoregressively: at each step, the model uses its own predictions to update the context and optimizes the single step noise prediction objective using the standard diffusion loss. This exposes the model to the inference state distribution during training, addressing the distribution shift between parallel supervision training and autoregressive inference. The method retains the original diffusion framework and sampler without requiring additional discriminators or distribution-matching losses.
Key Results
On multi-domain navigation datasets (RECON, SCAND, HuRoN, TartanDrive), AR Forcing improved the consistency of generated images during long-horizon navigation and the accuracy of predicted trajectories compared to strong baselines, enhancing robustness in complex known and unknown environments.