A cross-center smoothness prior for variational Bayesian brain tissue segmentation

Suppose one is faced with the challenge of tissue segmentation in MR images, without annotators at their center to provide labeled training data. One option is to go to another medical center for a trained classifier. Sadly, tissue classifiers do not generalize well across centers due to voxel intensity shifts caused by center-specific acquisition protocols. However, certain aspects of segmentations, such as spatial smoothness, remain relatively consistent and can be learned separately. Here we present a smoothness prior that is fit to segmentations produced at another medical center. This informative prior is presented to an unsupervised Bayesian model. The model clusters the voxel intensities, such that it produces segmentations that are similarly smooth to those of the other medical center. In addition, the unsupervised Bayesian model is extended to a semi-supervised variant, which needs no visual interpretation of clusters into tissues.

[1]  Marleen de Bruijne,et al.  Transfer Learning Improves Supervised Image Segmentation Across Imaging Protocols , 2015, IEEE Trans. Medical Imaging.

[2]  Qiang Yang,et al.  A Survey on Transfer Learning , 2010, IEEE Transactions on Knowledge and Data Engineering.

[3]  A. Hofman,et al.  The Rotterdam Scan Study: design update 2016 and main findings , 2015, European Journal of Epidemiology.

[4]  Jun Liu,et al.  Image Segmentation Using a Local GMM in a Variational Framework , 2012, Journal of Mathematical Imaging and Vision.

[5]  Radford M. Neal.  Pattern Recognition and Machine Learning , 2007, Technometrics.

[6]  Amir Alansary,et al.  MRBrainS Challenge: Online Evaluation Framework for Brain Image Segmentation in 3T MRI Scans , 2015, Comput. Intell. Neurosci..

[7]  N. Nasios,et al.  Variational learning for Gaussian mixture models , 2006, IEEE Transactions on Systems, Man, and Cybernetics, Part B (Cybernetics).

[8]  Yoonseok Shin,et al.  Application of Boosting Regression Trees to Preliminary Cost Estimation in Building Construction Projects , 2015, Comput. Intell. Neurosci..

[9]  D. Louis Collins,et al.  Twenty New Digital Brain Phantoms for Creation of Validation Image Data Bases , 2006, IEEE Transactions on Medical Imaging.

[10]  Christopher D. Manning,et al.  Hierarchical Bayesian Domain Adaptation , 2009, NAACL.

[11]  Gerhard Winkler,et al.  Image Analysis, Random Fields and Markov Chain Monte Carlo Methods: A Mathematical Introduction , 2002 .

[12]  Rajat Raina,et al.  Constructing informative priors using transfer learning , 2006, ICML.

[13]  Manuel Jorge Cardoso,et al.  Variational inference for medical image segmentation , 2016, Comput. Vis. Image Underst..

[14]  Torsten Rohlfing,et al.  Image Similarity and Tissue Overlaps as Surrogates for Image Registration Accuracy: Widely Used but Unreliable , 2012, IEEE Transactions on Medical Imaging.

[15]  Karl J. Friston,et al.  Unified segmentation , 2005, NeuroImage.

[16]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[17]  Nikos Komodakis,et al.  Markov Random Field modeling, inference & learning in computer vision & image understanding: A survey , 2013, Comput. Vis. Image Underst..

[18]  Wouter M. Kouw,et al.  A review of single-source unsupervised domain adaptation , 2019, ArXiv.

[19]  Anthony N. Pettitt,et al.  Variational Bayes for estimating the parameters of a hidden Potts model , 2009, Stat. Comput..

[20]  David M. Blei,et al.  Variational Inference: A Review for Statisticians , 2016, ArXiv.

[21]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[22]  Michael I. Jordan,et al.  Variational inference for Dirichlet process mixtures , 2006 .

[23]  Stephen M. Smith,et al.  Segmentation of brain MR images through a hidden Markov random field model and the expectation-maximization algorithm , 2001, IEEE Transactions on Medical Imaging.

[24]  Wouter M. Kouw,et al.  A Review of Domain Adaptation without Target Labels , 2019, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[25]  Marco Loog,et al.  Implicitly Constrained Semi-supervised Linear Discriminant Analysis , 2014, 2014 22nd International Conference on Pattern Recognition.

[26]  H Benoit-Cattin,et al.  The SIMRI project: a versatile and interactive MRI simulator. , 2005, Journal of magnetic resonance.