Probabilistic inference of regularisation in non-rigid registration

A long-standing issue in non-rigid image registration is the choice of the level of regularisation. Regularisation is necessary to preserve the smoothness of the registration and penalise against unnecessary complexity. The vast majority of existing registration methods use a fixed level of regularisation, which is typically hand-tuned by a user to provide "nice" results. However, the optimal level of regularisation will depend on the data which is being processed; lower signal-to-noise ratios require higher regularisation to avoid registering image noise as well as features, and different pairs of images require registrations of varying complexity depending on their anatomical similarity. In this paper we present a probabilistic registration framework that infers the level of regularisation from the data. An additional benefit of this proposed probabilistic framework is that estimates of the registration uncertainty are obtained. This framework has been implemented using a free-form deformation transformation model, although it would be generically applicable to a range of transformation models. We demonstrate our registration framework on the application of inter-subject brain registration of healthy control subjects from the NIREP database. In our results we show that our framework appropriately adapts the level of regularisation in the presence of noise, and that inferring regularisation on an individual basis leads to a reduction in model over-fitting as measured by image folding while providing a similar level of overlap.

[1]  B. Thirion,et al.  DEALING WITH SPATIAL NORMALIZATION ERRORS IN fMRI GROUP INFERENCE USING HIERARCHICAL MODELING , 2008 .

[2]  Stephen M. Smith,et al.  A global optimisation method for robust affine registration of brain images , 2001, Medical Image Anal..

[3]  Michael I. Miller,et al.  Deformable templates using large deformation kinematics , 1996, IEEE Trans. Image Process..

[4]  L. R. Dice Measures of the Amount of Ecologic Association Between Species , 1945 .

[5]  R. A. Leibler,et al.  On Information and Sufficiency , 1951 .

[6]  Carsten Peterson,et al.  A Mean Field Theory Learning Algorithm for Neural Networks , 1987, Complex Syst..

[7]  R. T. Cox Probability, frequency and reasonable expectation , 1990 .

[8]  Eigil Samset,et al.  Bayesian Estimation of Deformation and Elastic Parameters in Non-rigid Registration , 2010, WBIR.

[9]  Oscar Firschein,et al.  Readings in computer vision: issues, problems, principles, and paradigms , 1987 .

[10]  Michael I. Jordan,et al.  An Introduction to Variational Methods for Graphical Models , 1999, Machine-mediated learning.

[11]  H. Heyer,et al.  Information and Sufficiency , 1982 .

[12]  William M. Wells,et al.  A Marginalized MAP Approach and EM Optimization for Pair-Wise Registration , 2007, IPMI.

[13]  Mark W. Woolrich,et al.  Variational bayes inference of spatial mixture models for segmentation , 2006, IEEE Transactions on Medical Imaging.

[14]  M. Jenkinson Non-linear registration aka Spatial normalisation , 2007 .

[15]  Daniel Rueckert,et al.  Nonrigid registration using free-form deformations: application to breast MR images , 1999, IEEE Transactions on Medical Imaging.

[16]  John Ashburner,et al.  A fast diffeomorphic image registration algorithm , 2007, NeuroImage.

[17]  D. Hill,et al.  Non-rigid image registration: theory and practice. , 2004, The British journal of radiology.

[18]  William D. Penny,et al.  Variational Bayes for generalized autoregressive models , 2002, IEEE Trans. Signal Process..

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

[20]  Alain Trouvé,et al.  Computing Large Deformation Metric Mappings via Geodesic Flows of Diffeomorphisms , 2005, International Journal of Computer Vision.

[21]  Karl J. Friston,et al.  Diffeomorphic registration using geodesic shooting and Gauss–Newton optimisation , 2011, NeuroImage.

[22]  Tommi S. Jaakkola,et al.  Tutorial on variational approximation methods , 2000 .

[23]  Imran A. Pirwani,et al.  Introduction to the Non-rigid Image Registration Evaluation Project (NIREP) , 2006, WBIR.

[24]  Karl J. Friston,et al.  Tests for Distributed, Nonfocal Brain Activations , 1995, NeuroImage.

[25]  Marc L. Kessler,et al.  A Stochastic Approach to Estimate the UncertaintyInvolved in B-Spline Image Registration , 2009, IEEE Transactions on Medical Imaging.

[26]  Brian B. Avants,et al.  Symmetric diffeomorphic image registration with cross-correlation: Evaluating automated labeling of elderly and neurodegenerative brain , 2008, Medical Image Anal..

[27]  Nicholas Ayache,et al.  Symmetric Log-Domain Diffeomorphic Registration: A Demons-Based Approach , 2008, MICCAI.

[28]  M. Opper,et al.  Advanced mean field methods: theory and practice , 2001 .

[29]  Mark W. Woolrich,et al.  Longitudinal Brain MRI Analysis with Uncertain Registration , 2011, MICCAI.

[30]  Y. Amit,et al.  Towards a coherent statistical framework for dense deformable template estimation , 2007 .

[31]  Andrew P. Witkin,et al.  Scale-Space Filtering , 1983, IJCAI.

[32]  Mark W. Woolrich,et al.  Constrained linear basis sets for HRF modelling using Variational Bayes , 2004, NeuroImage.

[33]  Eigil Samset,et al.  Summarizing and Visualizing Uncertainty in Non-rigid Registration , 2010, MICCAI.

[34]  Jan Kybic,et al.  Bootstrap Resampling for Image Registration Uncertainty Estimation Without Ground Truth , 2010, IEEE Transactions on Image Processing.

[35]  Mert R. Sabuncu,et al.  Learning Task-Optimal Registration Cost Functions for Localizing Cytoarchitecture and Function in the Cerebral Cortex , 2010, IEEE Transactions on Medical Imaging.

[36]  D. Rubin,et al.  Maximum likelihood from incomplete data via the EM - algorithm plus discussions on the paper , 1977 .

[37]  J. Ashburner,et al.  Nonlinear spatial normalization using basis functions , 1999, Human brain mapping.

[38]  Haiying Liu,et al.  A Generic Framework for Non-rigid Registration Based on Non-uniform Multi-level Free-Form Deformations , 2001, MICCAI.

[39]  Mark W. Woolrich,et al.  Linked independent component analysis for multimodal data fusion , 2011, NeuroImage.