Hyperbolic Wasserstein Distance for Shape Indexing

IEEE Trans Pattern Anal Mach Intell. 2020 Jun;42(6):1362-1376. doi: 10.1109/TPAMI.2019.2898400. Epub 2019 Feb 8.

Abstract

Shape space is an active research topic in computer vision and medical imaging fields. The distance defined in a shape space may provide a simple and refined index to represent a unique shape. This work studies the Wasserstein space and proposes a novel framework to compute the Wasserstein distance between general topological surfaces by integrating hyperbolic Ricci flow, hyperbolic harmonic map, and hyperbolic power Voronoi diagram algorithms. The resulting hyperbolic Wasserstein distance can intrinsically measure the similarity between general topological surfaces. Our proposed algorithms are theoretically rigorous and practically efficient. It has the potential to be a powerful tool for 3D shape indexing research. We tested our algorithm with human face classification and Alzheimer's disease (AD) progression tracking studies. Experimental results demonstrated that our work may provide a succinct and effective shape index.

Publication types

  • Research Support, N.I.H., Extramural
  • Research Support, Non-U.S. Gov't
  • Research Support, U.S. Gov't, Non-P.H.S.

MeSH terms

  • Algorithms*
  • Brain / diagnostic imaging
  • Face / anatomy & histology
  • Face / diagnostic imaging
  • Humans
  • Image Processing, Computer-Assisted / methods
  • Imaging, Three-Dimensional / classification*
  • Imaging, Three-Dimensional / methods*
  • Pattern Recognition, Automated / methods