Gaussian Process Morphable Models

Models of shape variations have become a central component for the automated analysis of images. An important class of shape models are point distribution models (PDMs). These models represent a class of shapes as a normal distribution of point variations, whose parameters are estimated from example shapes. Principal component analysis (PCA) is applied to obtain a low-dimensional representation of the shape variation in terms of the leading principal components. In this paper, we propose a generalization of PDMs, which we refer to as Gaussian Process Morphable Models (GPMMs). We model the shape variations with a Gaussian process, which we represent using the leading components of its Karhunen-Loève expansion. To compute the expansion, we make use of an approximation scheme based on the Nyström method. The resulting model can be seen as a continuous analog of a standard PDM. However, while for PDMs the shape variation is restricted to the linear span of the example data, with GPMMs we can define the shape variation using any Gaussian process. For example, we can build shape models that correspond to classical spline models and thus do not require any example data. Furthermore, Gaussian processes make it possible to combine different models. For example, a PDM can be extended with a spline model, to obtain a model that incorporates learned shape characteristics but is flexible enough to explain shapes that cannot be represented by the PDM. We introduce a simple algorithm for fitting a GPMM to a surface or image. This results in a non-rigid registration approach whose regularization properties are defined by a GPMM. We show how we can obtain different registration schemes, including methods for multi-scale or hybrid registration, by constructing an appropriate GPMM. As our approach strictly separates modeling from the fitting process, this is all achieved without changes to the fitting algorithm. To demonstrate the applicability and versatility of GPMMs, we perform a set of experiments in typical usage scenarios in medical image analysis and computer vision: The model-based segmentation of 3D forearm images and the building of a statistical model of the face. To complement the paper, we have made all our methods available as open source.

[1]  Christopher K. I. Williams,et al.  Gaussian regression and optimal finite dimensional linear models , 1997 .

[2]  Thomas Vetter,et al.  A Unified Approach to Shape Model Fitting and Non-rigid Registration , 2013, MLMI.

[3]  Michael Unser,et al.  Fast parametric elastic image registration , 2003, IEEE Trans. Image Process..

[4]  Jan Modersitzki,et al.  Combination of automatic non-rigid and landmark based registration: the best of both worlds , 2003, SPIE Medical Imaging.

[5]  James T. Kwok,et al.  Making Large-Scale Nyström Approximation Possible , 2010, ICML.

[6]  Nathan Halko,et al.  Finding Structure with Randomness: Probabilistic Algorithms for Constructing Approximate Matrix Decompositions , 2009, SIAM Rev..

[7]  Lawrence H. Staib,et al.  Boundary Finding with Prior Shape and Smoothness Models , 2000, IEEE Trans. Pattern Anal. Mach. Intell..

[8]  Yuesheng Xu,et al.  Refinement of Reproducing Kernels , 2009, J. Mach. Learn. Res..

[9]  Dinggang Shen,et al.  Hierarchical active shape models, using the wavelet transform , 2003, IEEE Transactions on Medical Imaging.

[10]  Philippe C. Cattin,et al.  Statismo - A framework for PCA based statistical models , 2012, The Insight Journal.

[11]  Eam Khwang Teoh,et al.  A Novel 3D Partitioned Active Shape Model for Segmentation of Brain MR Images , 2005, MICCAI.

[12]  U. Grenander,et al.  Computational anatomy: an emerging discipline , 1998 .

[13]  Bernhard Schölkopf,et al.  Object correspondence as a machine learning problem , 2005, ICML.

[14]  Alireza Khotanzad,et al.  Elastic body splines: a physics based approach to coordinate transformation in medical image matching , 1995, Proceedings Eighth IEEE Symposium on Computer-Based Medical Systems.

[15]  A. Berlinet,et al.  Reproducing kernel Hilbert spaces in probability and statistics , 2004 .

[16]  Martin Styner,et al.  Evaluation of 3D Correspondence Methods for Model Building , 2003, IPMI.

[17]  William E. Lorensen,et al.  Marching cubes: A high resolution 3D surface construction algorithm , 1987, SIGGRAPH.

[18]  Alejandro F. Frangi,et al.  Automatic Construction of 3D Statistical Deformation Models Using Non-rigid Registration , 2001, MICCAI.

[19]  Jürgen Weese,et al.  Landmark-based elastic registration using approximating thin-plate splines , 2001, IEEE Transactions on Medical Imaging.

[20]  Thomas Vetter,et al.  Spatially Varying Registration Using Gaussian Processes , 2014, MICCAI.

[21]  Ronen Basri,et al.  Feature Matching with Bounded Distortion , 2014, ACM Trans. Graph..

[22]  Timothy F. Cootes,et al.  Active Shape Models-Their Training and Application , 1995, Comput. Vis. Image Underst..

[23]  David Duvenaud,et al.  Automatic model construction with Gaussian processes , 2014 .

[24]  Michael R. Lyu,et al.  Nonrigid shape recovery by Gaussian process regression , 2009, 2009 IEEE Conference on Computer Vision and Pattern Recognition.

[25]  François-Xavier Vialard,et al.  Left-Invariant Metrics for Diffeomorphic Image Registration with Spatially-Varying Regularisation , 2013, MICCAI.

[26]  Ronen Basri,et al.  Wide Baseline Stereo Matching with Convex Bounded Distortion Constraints , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[27]  G. Wahba Spline models for observational data , 1990 .

[28]  Mikhail Belkin,et al.  On Learning with Integral Operators , 2010, J. Mach. Learn. Res..

[29]  Aaron F. Bobick,et al.  Multiscale 3-D Shape Representation and Segmentation Using Spherical Wavelets , 2007, IEEE Transactions on Medical Imaging.

[30]  Matthew Turk,et al.  A Morphable Model For The Synthesis Of 3D Faces , 1999, SIGGRAPH.

[31]  Stefanos Zafeiriou,et al.  A 3D Morphable Model Learnt from 10,000 Faces , 2016, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[32]  Hans-Christian Hege,et al.  Omnidirectional displacements for deformable surfaces , 2013, Medical Image Anal..

[33]  Josien P. W. Pluim,et al.  Evaluation of Optimization Methods for Nonrigid Medical Image Registration Using Mutual Information and B-Splines , 2007, IEEE Transactions on Image Processing.

[34]  Xavier Pennec,et al.  Sparse Multi-Scale Diffeomorphic Registration: The Kernel Bundle Framework , 2012, Journal of Mathematical Imaging and Vision.

[35]  Roland Opfer,et al.  Multiscale kernels , 2006, Adv. Comput. Math..

[36]  Michael I. Miller,et al.  Pattern Theory: From Representation to Inference , 2007 .

[37]  Nello Cristianini,et al.  Kernel Methods for Pattern Analysis , 2003, ICTAI.

[38]  J. Marron,et al.  Object Statistics on Curved Manifolds , 2017 .

[39]  Max A. Viergever,et al.  elastix: A Toolbox for Intensity-Based Medical Image Registration , 2010, IEEE Transactions on Medical Imaging.

[40]  François-Xavier Vialard,et al.  Mixture of Kernels and Iterated Semidirect Product of Diffeomorphisms Groups , 2011, Multiscale Model. Simul..

[41]  Nikos Paragios,et al.  Deformable Medical Image Registration: A Survey , 2013, IEEE Transactions on Medical Imaging.

[42]  Heng Tao Shen,et al.  Principal Component Analysis , 2009, Encyclopedia of Biometrics.

[43]  Andriy Myronenko,et al.  Point Set Registration: Coherent Point Drift , 2009, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[44]  Andrea Tagliasacchi,et al.  Modern techniques and applications for real-time non-rigid registration , 2016, SIGGRAPH ASIA Courses.

[45]  Thomas Vetter,et al.  A statistical deformation prior for non-rigid image and shape registration , 2008, 2008 IEEE Conference on Computer Vision and Pattern Recognition.

[46]  Patrick Bouthemy,et al.  Optical flow modeling and computation: A survey , 2015, Comput. Vis. Image Underst..

[47]  Mark Holden,et al.  A Review of Geometric Transformations for Nonrigid Body Registration , 2008, IEEE Transactions on Medical Imaging.

[48]  Thomas Vetter,et al.  Using Landmarks as a Deformation Prior for Hybrid Image Registration , 2011, DAGM-Symposium.

[49]  Ioannis A. Kakadiaris,et al.  PDM-ENLOR: Learning Ensemble of Local PDM-Based Regressions , 2013, 2013 IEEE Conference on Computer Vision and Pattern Recognition.

[50]  Timothy F. Cootes,et al.  Combining point distribution models with shape models based on finite element analysis , 1994, Image Vis. Comput..

[51]  Gary E. Christensen,et al.  Consistent landmark and intensity-based image registration , 2002, IEEE Transactions on Medical Imaging.

[52]  L. Younes Shapes and Diffeomorphisms , 2010 .

[53]  Hans-Peter Meinzer,et al.  Statistical shape models for 3D medical image segmentation: A review , 2009, Medical Image Anal..

[54]  U. Grenander,et al.  Structural Image Restoration through Deformable Templates , 1991 .

[55]  S. Umeyama,et al.  Least-Squares Estimation of Transformation Parameters Between Two Point Patterns , 1991, IEEE Trans. Pattern Anal. Mach. Intell..

[56]  Michael I. Miller,et al.  Gaussian Random Fields on Sub-Manifolds for Characterizing Brain Surfaces , 1997, IPMI.

[57]  Carl E. Rasmussen,et al.  Gaussian processes for machine learning , 2005, Adaptive computation and machine learning.

[58]  Sami Romdhani,et al.  A 3D Face Model for Pose and Illumination Invariant Face Recognition , 2009, 2009 Sixth IEEE International Conference on Advanced Video and Signal Based Surveillance.

[59]  Karl Rohr,et al.  Hybrid Spline-Based Multimodal Registration Using Local Measures for Joint Entropy and Mutual Information , 2009, MICCAI.