Zeestraten, M.J.A., Havoutis, I., Silvério, J., Calinon, S. and Caldwell, D.G. (2017)
An Approach for Imitation Learning on Riemannian Manifolds
IEEE Robotics and Automation Letters (RA-L), 2:3, 1240-1247.


In imitation learning, multivariate Gaussians are widely used to encode robot behaviors. Such approaches do not provide the ability to properly represent end-effector orientation, as the distance metric in the space of orientations is not Euclidean. In this work we present an extension of common probabilistic learning from demonstration techniques to Riemannian manifolds. This generalization enables the encoding of joint distributions that include the robot pose. We show that Gaussian conditioning, Gaussian product and nonlinear regression can be achieved with this representation. The proposed approach is illustrated with examples on a 2-dimensional sphere, as well as with an example of regression between two robot end-effector poses, and by extending TP-GMM and GMR to Riemannian manifolds.

Bibtex reference

  author="Zeestraten, M. J. A. and Havoutis, I. and Silv\'erio, J. and Calinon, S. and Caldwell, D. G.",
  title="An Approach for Imitation Learning on {R}iemannian Manifolds",
  journal="{IEEE} Robotics and Automation Letters ({RA-L})",


Video credit: Ioannis Havoutis, Martijn Zeestraten

Source codes

Source codes related to this publication are available as part of PbDlib.

Go back to the list of publications