A transversal approach for patch-based label fusion via matrix completion

Recently, multi-atlas patch-based label fusion has received an increasing interest in the medical image segmentation field. After warping the anatomical labels from the atlas images to the target image by registration, label fusion is the key step to determine the latent label for each target image point. Two popular types of patch-based label fusion approaches are (1) reconstruction-based approaches that compute the target labels as a weighted average of atlas labels, where the weights are derived by reconstructing the target image patch using the atlas image patches; and (2) classification-based approaches that determine the target label as a mapping of the target image patch, where the mapping function is often learned using the atlas image patches and their corresponding labels. Both approaches have their advantages and limitations. In this paper, we propose a novel patch-based label fusion method to combine the above two types of approaches via matrix completion (and hence, we call it transversal). As we will show, our method overcomes the individual limitations of both reconstruction-based and classification-based approaches. Since the labeling confidences may vary across the target image points, we further propose a sequential labeling framework that first labels the highly confident points and then gradually labels more challenging points in an iterative manner, guided by the label information determined in the previous iterations. We demonstrate the performance of our novel label fusion method in segmenting the hippocampus in the ADNI dataset, subcortical and limbic structures in the LONI dataset, and mid-brain structures in the SATA dataset. We achieve more accurate segmentation results than both reconstruction-based and classification-based approaches. Our label fusion method is also ranked 1st in the online SATA Multi-Atlas Segmentation Challenge.

[1]  William M. Wells,et al.  Simultaneous truth and performance level estimation (STAPLE): an algorithm for the validation of image segmentation , 2004, IEEE Transactions on Medical Imaging.

[2]  Tom Vercauteren,et al.  Diffeomorphic demons: Efficient non-parametric image registration , 2009, NeuroImage.

[3]  Dinggang Shen,et al.  Development of Cortical Anatomical Properties from Early Childhood to Early Adulthood , 2022 .

[4]  Arno Klein,et al.  Evaluation of 14 nonlinear deformation algorithms applied to human brain MRI registration , 2009, NeuroImage.

[5]  Daniel Rueckert,et al.  Automatic anatomical brain MRI segmentation combining label propagation and decision fusion , 2006, NeuroImage.

[6]  N. Schuff,et al.  Comparison of automated and manual MRI volumetry of hippocampus in normal aging and dementia , 2002, Journal of magnetic resonance imaging : JMRI.

[7]  Colin Studholme,et al.  A Supervised Patch-Based Approach for Human Brain Labeling , 2011, IEEE Transactions on Medical Imaging.

[8]  Yaozong Gao,et al.  Learning to Rank Atlases for Multiple-Atlas Segmentation , 2014, IEEE Transactions on Medical Imaging.

[9]  Daoqiang Zhang,et al.  Confidence-Guided Sequential Label Fusion for Multi-atlas Based Segmentation , 2011, MICCAI.

[10]  Torsten Rohlfing,et al.  Evaluation of atlas selection strategies for atlas-based image segmentation with application to confocal microscopy images of bee brains , 2004, NeuroImage.

[11]  Bennett A Landman,et al.  Non-local statistical label fusion for multi-atlas segmentation , 2013, Medical Image Anal..

[12]  Brian B. Avants,et al.  A learning-based wrapper method to correct systematic errors in automatic image segmentation: Consistently improved performance in hippocampus, cortex and brain segmentation , 2011, NeuroImage.

[13]  Dinggang Shen,et al.  Iterative multi-atlas-based multi-image segmentation with tree-based registration , 2012, NeuroImage.

[14]  Dinggang Shen,et al.  Novel Multi-Atlas Segmentation by Matrix Completion , 2014, MLMI.

[15]  Torsten Rohlfing,et al.  Performance-based classifier combination in atlas-based image segmentation using expectation-maximization parameter estimation , 2004, IEEE Transactions on Medical Imaging.

[16]  Robert D. Nowak,et al.  Transduction with Matrix Completion: Three Birds with One Stone , 2010, NIPS.

[17]  Dinggang Shen,et al.  A computational growth model for measuring dynamic cortical development in the first year of life. , 2012, Cerebral cortex.

[18]  J. Gilmore,et al.  Mapping Longitudinal Development of Local Cortical Gyrification in Infants from Birth to 2 Years of Age , 2014, The Journal of Neuroscience.

[19]  Dinggang Shen,et al.  SharpMean: Groupwise registration guided by sharp mean image and tree-based registration , 2011, NeuroImage.

[20]  Francesc Serratosa,et al.  Smooth point-set registration using neighboring constraints , 2012, Pattern Recognit. Lett..

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

[22]  Max A. Viergever,et al.  Multi-Atlas-Based Segmentation With Local Decision Fusion—Application to Cardiac and Aortic Segmentation in CT Scans , 2009, IEEE Transactions on Medical Imaging.

[23]  Daniel Rueckert,et al.  Multi-atlas based segmentation of brain images: Atlas selection and its effect on accuracy , 2009, NeuroImage.

[24]  D. Louis Collins,et al.  Towards accurate, automatic segmentation of the hippocampus and amygdala from MRI by augmenting ANIMAL with a template library and label fusion , 2010, NeuroImage.

[25]  Emmanuel J. Candès,et al.  Exact Matrix Completion via Convex Optimization , 2009, Found. Comput. Math..

[26]  Paul A. Yushkevich,et al.  Optimal Weights for Multi-atlas Label Fusion , 2011, IPMI.

[27]  Ben Glocker,et al.  Atlas Encoding by Randomized Forests for Efficient Label Propagation , 2013, MICCAI.

[28]  Arthur W. Toga,et al.  Construction of a 3D probabilistic atlas of human cortical structures , 2008, NeuroImage.

[29]  Stephen M. Smith,et al.  Improved Optimization for the Robust and Accurate Linear Registration and Motion Correction of Brain Images , 2002, NeuroImage.

[30]  Kaiming Li,et al.  Gyral Folding Pattern Analysis via Surface Profiling , 2009, MICCAI.

[31]  Sébastien Ourselin,et al.  STEPS: Similarity and Truth Estimation for Propagated Segmentations and its application to hippocampal segmentation and brain parcelation , 2013, Medical Image Anal..

[32]  Alexandre Bernardino,et al.  Matrix Completion for Multi-label Image Classification , 2011, NIPS.

[33]  Dinggang Shen,et al.  journal homepage: www.elsevier.com/locate/ynimg , 2022 .

[34]  Cameron S. Carter,et al.  Optimum template selection for atlas-based segmentation , 2007, NeuroImage.

[35]  Brian B. Avants,et al.  N4ITK: Improved N3 Bias Correction , 2010, IEEE Transactions on Medical Imaging.

[36]  Tianzi Jiang,et al.  Local label learning (LLL) for subcortical structure segmentation: Application to hippocampus segmentation , 2014, Human brain mapping.

[37]  Paul A. Yushkevich,et al.  Multi-Atlas Segmentation with Joint Label Fusion , 2013, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[38]  Carlos Ortiz-de-Solorzano,et al.  Combination Strategies in Multi-Atlas Image Segmentation: Application to Brain MR Data , 2009, IEEE Transactions on Medical Imaging.

[39]  Lei Guo,et al.  Gyral folding pattern analysis via surface profiling , 2010, NeuroImage.

[40]  Michael Brady,et al.  Improved Optimization for the Robust and Accurate Linear Registration and Motion Correction of Brain Images , 2002, NeuroImage.

[41]  Dinggang Shen,et al.  HAMMER: hierarchical attribute matching mechanism for elastic registration , 2002, IEEE Transactions on Medical Imaging.

[42]  Shuiwang Ji,et al.  SLEP: Sparse Learning with Efficient Projections , 2011 .

[43]  D. Louis Collins,et al.  Patch-based segmentation using expert priors: Application to hippocampus and ventricle segmentation , 2011, NeuroImage.

[44]  Wei Li,et al.  Automatic hippocampus segmentation of 7.0Tesla MR images by combining multiple atlases and auto-context models , 2013, NeuroImage.

[45]  Francesc Serratosa,et al.  A new graph matching method for point-set correspondence using the EM algorithm and Softassign , 2012, Comput. Vis. Image Underst..

[46]  Torsten Rohlfing,et al.  Quo Vadis, Atlas-Based Segmentation? , 2005 .

[47]  Yaozong Gao,et al.  Learning of Atlas Forest Hierarchy for Automatic Labeling of MR Brain Images , 2014, MLMI.