Principal geodesic analysis (PGA) is a generalization of principal component analysis (PCA) for dimensionality reduction of data on a Riemannian manifold. Currently PGA is defined as a geometric fit to the data, rather than as a probabilistic model. Inspired by probabilistic PCA, we present a latent variable model for PGA that provides a probabilistic framework for factor analysis on manifolds. To compute maximum likelihood estimates of the parameters in our model, we develop a Monte Carlo Expectation Maximization algorithm, where the expectation is approximated by Hamiltonian Monte Carlo sampling of the latent variables. We demonstrate the ability of our method to recover the ground truth parameters in simulated sphere data, as well as its effectiveness in analyzing shape variability of a corpus callosum data set from human brain images.
Miaomiao Zhang (MIT)
Tom Fletcher (University of Utah)
More from the Same Authors
2017 Poster: Semi-supervised Learning with GANs: Manifold Invariance with Improved Inference »
Abhishek Kumar · Prasanna Sattigeri · Tom Fletcher
2014 Workshop: Riemannian geometry in machine learning, statistics and computer vision »
Minh Ha Quang · Vikas Sindhwani · Vittorio Murino · Michael Betancourt · Tom Fletcher · Richard I Hartley · Anuj Srivastava · Bart Vandereycken