arXiv:2412.19948cs.RO2024-12被引 58

用扩散模型学习机器人运动规划先验,提升复杂场景下的规划速度与平滑性。

Motion Planning Diffusion: Learning and Adapting Robot Motion Planning with Diffusion Models

  • 基于扩散模型学习轨迹分布先验,结合成本函数梯度生成路径。
  • 采用B样条参数化降低维度,生成平滑且可高频插值的轨迹。
  • 适用于仿真与真实机械臂任务,支持从人类示范中学习。

优化型机器人运动规划算法的性能高度依赖初始解,通常通过采样类规划器获取无碰撞路径。然而,这类方法在高维复杂场景中速度慢且结果不平滑。若能学习已有规划问题的轨迹分布作为先验,可显著加速新类似问题的求解。现有工作尝试将此先验用于初始化或贝叶斯优化。本文提出运动规划扩散模型(MPD),利用扩散模型学习轨迹分布先验。该生成模型擅长捕捉多模态数据,并具备梯度引导特性,适合梯度驱动的规划。给定规划问题后,构建代价函数,在去噪过程中结合学习到的先验与代价梯度采样后验分布。为减少参数量并保证平滑性,采用线性运动基元对轨迹进行低维表示,特别是使用B样条曲线。该参数化方式确保轨迹平滑、可高频插值,且所需参数少于密集路径点表示。我们在2D简单场景及7自由度机械臂的复杂任务中验证了方法效果。此外,还利用真实世界的拾取放置任务中的人类示范数据进行训练。

原文摘要 · Abstract (English)

The performance of optimization-based robot motion planning algorithms is highly dependent on the initial solutions, commonly obtained by running a sampling-based planner to obtain a collision-free path. However, these methods can be slow in high-dimensional and complex scenes and produce non-smooth solutions. Given previously solved path-planning problems, it is highly desirable to learn their distribution and use it as a prior for new similar problems. Several works propose utilizing this prior to bootstrap the motion planning problem, either by sampling initial solutions from it, or using its distribution in a maximum-a-posterior formulation for trajectory optimization. In this work, we introduce Motion Planning Diffusion (MPD), an algorithm that learns trajectory distribution priors with diffusion models. These generative models have shown increasing success in encoding multimodal data and have desirable properties for gradient-based motion planning, such as cost guidance. Given a motion planning problem, we construct a cost function and sample from the posterior distribution using the learned prior combined with the cost function gradients during the denoising process. Instead of learning the prior on all trajectory waypoints, we propose learning a lower-dimensional representation of a trajectory using linear motion primitives, particularly B-spline curves. This parametrization guarantees that the generated trajectory is smooth, can be interpolated at higher frequencies, and needs fewer parameters than a dense waypoint representation. We demonstrate the results of our method ranging from simple 2D to more complex tasks using a 7-dof robot arm manipulator. In addition to learning from simulated data, we also use human demonstrations on a real-world pick-and-place task.

运动规划扩散模型机器人轨迹生成

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