English

A Wasserstein-type Distance for Gaussian Mixtures on Vector Bundles with Applications to Shape Analysis

Methodology 2023-11-29 v1

Abstract

This paper uses sample data to study the problem of comparing populations on finite-dimensional parallelizable Riemannian manifolds and more general trivial vector bundles. Utilizing triviality, our framework represents populations as mixtures of Gaussians on vector bundles and estimates the population parameters using a mode-based clustering algorithm. We derive a Wasserstein-type metric between Gaussian mixtures, adapted to the manifold geometry, in order to compare estimated distributions. Our contributions include an identifiability result for Gaussian mixtures on manifold domains and a convenient characterization of optimal couplings of Gaussian mixtures under the derived metric. We demonstrate these tools on some example domains, including the pre-shape space of planar closed curves, with applications to the shape space of triangles and populations of nanoparticles. In the nanoparticle application, we consider a sequence of populations of particle shapes arising from a manufacturing process, and utilize the Wasserstein-type distance to perform change-point detection.

Keywords

Cite

@article{arxiv.2311.16988,
  title  = {A Wasserstein-type Distance for Gaussian Mixtures on Vector Bundles with Applications to Shape Analysis},
  author = {Michael Wilson and Tom Needham and Chiwoo Park and Suprateek Kundu and Anuj Srivastava},
  journal= {arXiv preprint arXiv:2311.16988},
  year   = {2023}
}