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.

Matching Parameterized Shapes by Nonparametric Belief Propagation

Regularization Based Iterative Point Match Weighting for Accurate Rigid Transformation Estimation

Contour matching based on belief propagation

3D Shape Retrieval Based on Laplace Operator and Joint Bayesian Model

Progressive Probabilistic Graph Matching With Local Consistency Regularization

Belief propagation on Riemannian manifold for stereo matching

3D Shape Registration Using Spectral Graph Embedding and Probabilistic Matching

Nonrigid Shape Recovery by Gaussian Process Regression

Shape Sparse Correspondence for Nearly-Isometric Meshes Based on Green's Function Representation

Toward a Real-Time Belief Propagation Stereo Reconstruction for Computers, Robots, and Beyond

Deformation-Guided Unsupervised Non-Rigid Shape Matching

Reduced Set Density Estimator For Object Segmentation Based On Shape Probabilistic Representation

Multi-part shape matching by simultaneous partial functional correspondence

Dependent landmark drift: robust point set registration with a Gaussian mixture model and a statistical shape model

Integrating Efficient Optimal Transport and Functional Maps For Unsupervised Shape Correspondence Learning

Model-free Consensus Maximization for Non-Rigid Shapes

Optimal Mass Transport for Shape Matching and Comparison

DNBP: Differentiable Nonparametric Belief Propagation

Non-rigid 3D Shape Retrieval Using Multidimensional Scaling and Bag-of-Features

Extracting a functional representation from a dictionary for non-rigid shape matching

Similarity estimation based on sparse spectral correspondence