AI Agents 相关度: 9/10

Rectified Schrödinger Bridge Matching for Few-Step Visual Navigation

Wuyang Luan, Junhui Li, Weiguang Zhao, Wenjian Zhang, Tieru Wu, Rui Ma
arXiv: 2604.05673v1 发布: 2026-04-07 更新: 2026-04-07

AI 摘要

提出Rectified Schrödinger Bridge Matching (RSBM)算法,加速视觉导航中生成策略的推理速度。

主要贡献

  • 提出RSBM框架,利用熵正则化参数控制Schrödinger Bridges和Optimal Transport。
  • 证明速度场函数形式在整个ε-spectrum上的不变性(速度结构不变性)。
  • 证明降低ε线性降低条件速度方差,实现更稳定的粗步ODE积分。
  • 在仅3步积分的情况下,实现高成功率,提高实时机器人控制性能。

方法论

通过学习条件先验,缩短传输距离,并使用中间ε值平衡多模态覆盖和路径直线度,加速收敛。

原文摘要

Visual navigation is a core challenge in Embodied AI, requiring autonomous agents to translate high-dimensional sensory observations into continuous, long-horizon action trajectories. While generative policies based on diffusion models and Schrödinger Bridges (SB) effectively capture multimodal action distributions, they require dozens of integration steps due to high-variance stochastic transport, posing a critical barrier for real-time robotic control. We propose Rectified Schrödinger Bridge Matching (RSBM), a framework that exploits a shared velocity-field structure between standard Schrödinger Bridges ($\varepsilon=1$, maximum-entropy transport) and deterministic Optimal Transport ($\varepsilon\to 0$, as in Conditional Flow Matching), controlled by a single entropic regularization parameter $\varepsilon$. We prove two key results: (1) the conditional velocity field's functional form is invariant across the entire $\varepsilon$-spectrum (Velocity Structure Invariance), enabling a single network to serve all regularization strengths; and (2) reducing $\varepsilon$ linearly decreases the conditional velocity variance, enabling more stable coarse-step ODE integration. Anchored to a learned conditional prior that shortens transport distance, RSBM operates at an intermediate $\varepsilon$ that balances multimodal coverage and path straightness. Empirically, while standard bridges require $\geq 10$ steps to converge, RSBM achieves over 94% cosine similarity and 92% success rate in merely 3 integration steps -- without distillation or multi-stage training -- substantially narrowing the gap between high-fidelity generative policies and the low-latency demands of Embodied AI.

标签

视觉导航 Schrödinger Bridges 扩散模型 机器人控制 Optimal Transport

arXiv 分类

cs.RO cs.AI