Mar, 2024

基于黎曼流匹配策略的机器人运动学习

TL;DR我们引入了一种名为 Riemannian Flow Matching Policies(RFMP)的新型模型,用于学习和合成机器人视觉运动策略。RFMP 利用了流匹配方法的高效训练和推理能力,具备编码机器人任务中常见的高维多模态分布和简单快速推理过程的优点。我们展示了 RFMP 在基于状态和基于视觉条件的机器人运动策略中的适用性。值得注意的是,由于机器人状态存在于 Riemann 流形上,RFMP 固有地融合了几何意识,这对于真实机器人任务至关重要。为了评估 RFMP,我们进行了两个概念验证实验,并将其性能与 Diffusion Policies 进行了比较。尽管两种方法都成功地学习了考虑的任务,但结果表明,RFMP 提供了更平滑的动作轨迹并具有显着较低的推理时间。