Abstract:Surface based 3D shape analysis plays a fundamental role in computer vision and medical imaging. This work proposes to use optimal mass transport map for shape matching and comparison, focusing on two important applications including surface registration and shape space. The computation of the optimal mass transport map is based on Monge-Brenier theory, in comparison to the conventional method based on Monge-Kantorovich theory, this method significantly improves the efficiency by reducing computational complexity from $O(n^2)$ to $O(n)$ . For surface registration problem, one commonly used approach is to use conformal map to convert the shapes into some canonical space. Although conformal mappings have small angle distortions, they may introduce large area distortions which are likely to cause numerical instability thus resulting failures of shape analysis. This work proposes to compose the conformal map with the optimal mass transport map to get the unique area-preserving map, which is intrinsic to the Riemannian metric, unique, and diffeomorphic. For shape space study, this work introduces a novel Riemannian framework, Conformal Wasserstein Shape Space, by combing conformal geometry and optimal mass transport theory. In our work, all metric surfaces with the disk topology are mapped to the unit planar disk by a conformal mapping, which pushes the area element on the surface to a probability measure on the disk. The optimal mass transport provides a map from the shape space of all topological disks with metrics to the Wasserstein space of the disk and the pullback Wasserstein metric equips the shape space with a Riemannian metric. We validate our work by numerous experiments and comparisons with prior approaches and the experimental results demonstrate the efficiency and efficacy of our proposed approach.

Alpha shapes and optimal transport on the sphere

Alpha shapes in kernel density estimation

Optimal Mass Transport for Shape Matching and Comparison

Topological Optimal Transport for Geometric Cycle Matching

On optimal transport of matrix-valued measures

Monge, Bregman and Occam: Interpretable Optimal Transport in High-Dimensions with Feature-Sparse Maps

Sliced Optimal Transport on the Sphere

Parallelly Sliced Optimal Transport on Spheres and on the Rotation Group

Shape Analysis with Hyperbolic Wasserstein Distance

On the Kähler Geometry of Certain Optimal Transport Problems

A Volumetric Approach to Monge's Optimal Transport on Surfaces

Linear Spherical Sliced Optimal Transport: A Fast Metric for Comparing Spherical Data

Optimal Transport on Discrete Domains

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

Optimal transport for Gaussian mixture models

Non‐Euclidean Sliced Optimal Transport Sampling

A new method for determining Wasserstein 1 optimal transport maps from Kantorovich potentials, with deep learning applications

Data representation with optimal transport

Simple Unbalanced Optimal Transport

The Shape of Data and Probability Measures

Tsallis Regularized Optimal Transport and Ecological Inference