PUMA: Deep Metric Imitation Learning for Stable Motion Primitives

Rodrigo Pérez-Dattari*, Cosimo Della Santina, Jens Kober

*Corresponding author for this work

Research output: Contribution to journalArticleScientificpeer-review

13 Downloads (Pure)

Abstract

Imitation learning (IL) facilitates intuitive robotic programming. However, ensuring the reliability of learned behaviors remains a challenge. In the context of reaching motions, a robot should consistently reach its goal, regardless of its initial conditions. To meet this requirement, IL methods often employ specialized function approximators that guarantee this property by construction. Although effective, these approaches come with some limitations: 1) they are typically restricted in the range of motions they can model, resulting in suboptimal IL capabilities, and 2) they require explicit extensions to account for the geometry of motions that consider orientations. To address these challenges, we introduce a novel stability loss function that does not constrain the function approximator's architecture and enables learning policies that yield accurate results. Furthermore, it is not restricted to a specific state space geometry; therefore, it can easily incorporate the geometry of the robot's state space. Proof of the stability properties induced by this loss is provided and the method is empirically validated in various settings. These settings include Euclidean and non-Euclidean state spaces, as well as first-order and second-order motions, both in simulation and with real robots. More details about the experimental results can be found at https://youtu.be/ZWKLGntCI6w.

Original languageEnglish
Article number2400144
Number of pages20
JournalAdvanced Intelligent Systems
DOIs
Publication statusPublished - 2024

Keywords

  • deep metric learning
  • deep neural networks
  • dynamical systems
  • imitation learning
  • motion primitives
  • non-Euclidean geometries

Fingerprint

Dive into the research topics of 'PUMA: Deep Metric Imitation Learning for Stable Motion Primitives'. Together they form a unique fingerprint.

Cite this