Aarhus Universitets segl

Wrapped Gaussian Process Regression on Riemannian Manifolds

by Anton Mallasto and Aasa Feragen
CSGB Research Reports Number 3 (February 2018)
Gaussian process (GP) regression is a powerful tool in non-parametric regression providing uncertainty estimates. However, it is limited to data in vector spaces. In fields such as shape analysis and diffusion tensor imaging, the data often lies on a manifold, making GP regression non-viable, as the resulting predictive distribution does not live in the correct geometric space. We tackle the problem by defining wrapped Gaussian processes (WGPs) on Riemannian manifolds, using the probabilistic setting to generalize GP regression to the context of manifold-valued targets. The method is validated empirically on diffusion tensor imaging (DTI) data and in the Kendall shape space, endorsing WGP regression as an efficient and flexible tool for manifold-valued regression.
Format available: PDF (4 MB)