arXiv:2605.31314cs.RO2026-05

用自回归训练让机器人导航模型更稳定,避免长距离预测时出错。

AR Forcing: Towards Long-Horizon Robot Navigation World Model

论文配图:AR Forcing: Towards Long-Horizon Robot Navigation World Model
图 1 · 摘自论文原文
  • 在训练中模拟推理过程,让模型提前适应自身预测的误差。
  • 在多个数据集上显著提升长程路径预测准确率和图像生成一致性。
  • 无需额外模块,可无缝接入现有扩散模型框架,适合做导航系统的研究者。

基于扩散模型的机器人导航世界模型通常采用并行监督进行训练,但在路径规划中使用自回归推理,导致训练与推理分布不一致,影响长程预测的稳定性。本文提出AR Forcing,一种自回归训练策略,将标准扩散损失融入自回归训练循环。每一步中,模型利用自身预测更新上下文,并优化单步噪声预测目标,从而在训练阶段显式暴露于推理状态分布。该方法无需额外判别器或分布匹配损失,保持原有扩散框架与采样器,易于集成。在多领域导航数据集(RECON、SCAND、HuRoN、TartanDrive)上的实验表明,相比强基线,AR Forcing显著提升了长程导航中的图像生成一致性与轨迹预测精度,增强了模型在复杂已知与未知环境下的鲁棒性。代码即将开源。

原文摘要 · Abstract (English)

The diffusion based robot navigation world models are typically trained using parallel supervision, while autoregressive inference is employed during path planning. This results in a distribution shift between training and inference, which destabilizes the performance over long-horizon prediction. We propose AR Forcing, an autoregressive training strategy, which integrates the standard diffusion loss into the autoregressive training loop. At each step, the model uses its own predictions to update the context and optimize the single step noise prediction objective, thereby explicitly exposing the model to the inference state distribution during training. Our method does not require additional discriminators or distribution-matching losses, retains the original diffusion framework and sampler, and is easy to integrate. Experiments on multi-domain navigation datasets (RECON, SCAND, HuRoN, TartanDrive) show that compared with strong baselines, AR Forcing improved the consistency of generated images during long-horizon navigation and the accuracy of predicted trajectories, enhancing robustness of the model in complex known and unknown environments. We will release the code soon.

机器人导航扩散模型自回归长程预测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。