Deep CNN ensembles and suggestive annotations for infant brain MRI segmentation

  • 2017-12-14 16:27:42
  • Jose Dolz, Christian Desrosiers, Li Wang, Jing Yuan, Dinggang Shen, Ismail Ben Ayed
  • 28

Abstract

Precise 3D segmentation of infant brain tissues is an essential step towardscomprehensive volumetric studies and quantitative analysis of early braindevelopement. However, computing such segmentations is very challenging,especially for the infant brain at round 6-month of age, due to the poor imagequality, among other difficulties inherent to infant brain MRI, e.g., theisointense contrast between white and gray matter and the severe partial volumeeffect due to small brain sizes. This study investigates the problem with anensemble of semi-dense fully convolutional neural networks (CNNs), whichemploys T1-weighted and T2-weighted MR images as input. We demonstrate that theensemble agreement is highly correlated with the segmentation errors.Therefore, our method provides measures that can guide local user corrections.To the best of our knowledge, this work is the first ensemble of 3D CNNs forsuggesting annotations within images. Furthermore, inspired by the very recentsuccess of dense networks, we propose a novel architecture, SemiDenseNet, whichconnects all convolutional layers directly to the end of the network. Ourarchitecture allows the efficient propagation of gradients during training,while limiting the number of parameters, requiring one order of magnitude lessparameters than popular medical image segmentation networks such as 3D U-Net.Another contribution of our work is the study of the impact that early or latefusions of multiple image modalities might have on the performances of deeparchitectures. We report evaluations of our method on the public data of theMICCAI iSEG-2017 Challenge on 6-month infant brain MRI segmentation, and showvery competitive results among 21 teams, ranking first or second in mostmetrics.

 

Quick Read (beta)

loading the full paper ...