English

Riemannian Flow Matching Policy for Robot Motion Learning

Robotics 2024-08-28 v2 Machine Learning

Abstract

We introduce Riemannian Flow Matching Policies (RFMP), a novel model for learning and synthesizing robot visuomotor policies. RFMP leverages the efficient training and inference capabilities of flow matching methods. By design, RFMP inherits the strengths of flow matching: the ability to encode high-dimensional multimodal distributions, commonly encountered in robotic tasks, and a very simple and fast inference process. We demonstrate the applicability of RFMP to both state-based and vision-conditioned robot motion policies. Notably, as the robot state resides on a Riemannian manifold, RFMP inherently incorporates geometric awareness, which is crucial for realistic robotic tasks. To evaluate RFMP, we conduct two proof-of-concept experiments, comparing its performance against Diffusion Policies. Although both approaches successfully learn the considered tasks, our results show that RFMP provides smoother action trajectories with significantly lower inference times.

Keywords

Cite

@article{arxiv.2403.10672,
  title  = {Riemannian Flow Matching Policy for Robot Motion Learning},
  author = {Max Braun and Noémie Jaquier and Leonel Rozo and Tamim Asfour},
  journal= {arXiv preprint arXiv:2403.10672},
  year   = {2024}
}

Comments

Accepted for publication at IROS'24. 8 pages, 5 figures, 4 tables

R2 v1 2026-06-28T15:22:23.622Z