arXiv:2606.21646cs.RO2026-06

用能量模型统一长程机器人规划,提升准确性与效率

Energy-based Compositional Diffusion Planning

论文配图:Energy-based Compositional Diffusion Planning
图 1 · 摘自论文原文
  • 将局部轨迹拼接转化为能量最小化问题,构建保守修正场
  • 在OGBench任务上达到当前最优成功率,推理速度接近启发式方法
  • 适合需要高可靠性长程规划的机器人研究者

组合扩散规划器通过短训练轨迹解决长时序机器人任务,但现有方法依赖启发式拼接,导致拼接更新通常是非保守场,不对应任何有效的全局轨迹对数密度函数。本文提出基于能量的组合扩散器(ECD),将全局轨迹建模为局部桥接势能之和的最小值。该能量视角定义了保守修正场,并包含启发式拼接忽略的边界反应项。为实现高效推理,引入基于马尔可夫的得分近似,通过单次块三对角求解计算反应项,保持规划时长线性时间复杂度。实验表明,ECD在一系列OGBench拼接任务上取得当前最优成功率达92.3%,同时几乎与启发式拼接方法相当的推理速度。代码已开源:https://github.com/GradientSpaces/ECD。

原文摘要 · Abstract (English)

Compositional diffusion planners aim to solve long-horizon robotic tasks using short training trajectories. Yet, current approaches often rely on the heuristic stitching of local predictions. We show that the resulting stitched update is generally a non-conservative field} that does not mathematically correspond to any valid global trajectory log-density function. We propose Energy-based Compositional Diffuser (ECD), a framework that formulates the global trajectory as the minimizer of the sum of local bridge potentials. This energy-based perspective defines a conservative correction field and contains a boundary reaction term that heuristic stitching omits. To enable efficient inference, we further introduce a Markov-based score approximation that computes the reaction term via a single block-tridiagonal solve, maintaining time complexity linear in the planning horizon. Empirically, ECD achieves state-of-the-art success rates on a range of OGBench stitching tasks, while nearly matching the inference speed of heuristic stitching methods. Code is available at https://github.com/GradientSpaces/ECD.

机器人规划扩散模型能量模型轨迹生成

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