Spatially Localized Atlas Network Tiles Enables 3D Whole Brain Segmentation from Limited Data

06/01/2018 ∙ by Yuankai Huo, et al. ∙ 0

Whole brain segmentation on a structural magnetic resonance imaging (MRI) is essential in non-invasive investigation for neuroanatomy. Historically, multi-atlas segmentation (MAS) has been regarded as the de facto standard method for whole brain segmentation. Recently, deep neural network approaches have been applied to whole brain segmentation by learning random patches or 2D slices. Yet, few previous efforts have been made on detailed whole brain segmentation using 3D networks due to the following challenges: (1) fitting entire whole brain volume into 3D networks is restricted by the current GPU memory, and (2) the large number of targeting labels (e.g., > 100 labels) with limited number of training 3D volumes (e.g., < 50 scans). In this paper, we propose the spatially localized atlas network tiles (SLANT) method to distribute multiple independent 3D fully convolutional networks to cover overlapped sub-spaces in a standard atlas space. This strategy simplifies the whole brain learning task to localized sub-tasks, which was enabled by combing canonical registration and label fusion techniques with deep learning. To address the second challenge, auxiliary labels on 5111 initially unlabeled scans were created by MAS for pre-training. From empirical validation, the state-of-the-art MAS method achieved mean Dice value of 0.76, 0.71, and 0.68, while the proposed method achieved 0.78, 0.73, and 0.71 on three validation cohorts. Moreover, the computational time reduced from > 30 hours using MAS to 15 minutes using the proposed method. The source code is available online https://github.com/MASILab/SLANT_train_seg

READ FULL TEXT VIEW PDF
POST COMMENT

Comments

There are no comments yet.

Authors

page 2

page 3

page 5

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

1 Introduction

Historically, multi-atlas segmentation (MAS) has been regarded as the de facto standard method on detailed whole brain segmentation ( 100 anatomical regions) due to its high accuracy. Moreover, MAS only demands a small number of manually labeled examples (atlases)[1]

. Recently, deep convolutional neural networks (DCNN) have been applied to whole brain segmentation. To address the challenges of training a network on a small number of manually traced brains, patch-based DCNN methods have been proposed. de Brébisson et al.,

[2] proposed to learn 2D and 3D patches as well as spatial information, which was extended to include 2.5D patches by BrainSegNet [10]. Recently, DeepNAT [13] was proposed to perform hierarchical multi-task learning on 3D patches. Li et al., [8] introduced the 3D patch-based HC Net for high resolution segmentation. From another perspective, Roy et al., [12] proposed to use 2D fully convolutional network (FCN) to learn slice-wise image features by using auxiliary labels on initially unlabeled data. Although detailed cortical parcellations were not performed, Roy et al., revealed a promising direction on how to use initially unlabeled data to leverage training. With a large number of auxiliary labels, it is appealing to perform 3D FCN (e.g., 3D U-Net [3]) on whole brain segmentation since it typically yields higher spatial consistency than 2D or patch-based methods. However, directly applying 3D FCN to whole brain segmentation (e.g., 1mm isotropic resolution) is restricted by the current graphics processing unit (GPU) memory. A common solution is to down sample the inputs, yet, the accuracy can be sacrificed.

In this paper, we propose the spatially localized atlas network tiles (SLANT) method for detailed whole brain segmentation (133 labels under BrainCOLOR protocol [6]) by combining canonical medical image processing techniques with deep learning. SLANT distributes a set of independent 3D networks (“network tiles”) to cover overlapped sub-spaces in a standard MNI atlas space [5].

Figure 1: The proposed SLANT-27 (27 network tiles) method is presented, which combines canonical medical image processing methods (registration, harmonization, label fusion) with 3D network tiles. 3D U-Net is used as each tile, whose deconvolutional channel numbers are modified to 133. The tiles are spatially overlapped in MNI space.

Then, majority vote label fusion was used to obtain final whole brain segmentation from the overlapped sub-spaces. To leverage learning performance on 133 labels with only 45 manually traced training data, auxiliary labels on 5111 initially unlabeled scans were created from non-local spatial STAPLE (NLSS) MAS [1] for pre-training inspired by [12].

2 Methods

2.0.1 Registration and Intensity Harmonization:

Affine registration [11] was employed to register all training and testing scans to MNI 305 space [5] (Fig. 1

). Then, N4 bias field correction was deployed to reduce bias. To further harmonize the intensities on large-scale MRI, we introduced a regression-based intensity normalization method. First, we defined a gray-scale MRI volume (with N voxels) as a vector

.

was demeaned and normalized by standard deviation (std) to

. The intensities were harmonized by a pre-trained linear regression model on “sorted intensity”. The sorted intensity vector

was calculated from , where ”sort” rearrange intensities from largest to smallest, and ”” was a prior mask learned from a union operation from all atlases. To train the linear regression, mean sorted intensity vector was obtained by averaging from all atlases. . The coefficients were fitted between (from ) and , and intensity normalized image is obtained from fitted and : , and .

Figure 2: SLANT-8 covered eight non-overlapped sub-spaces in MNI, while SLANT-27 covered 27 overlapped sub-spaces in MNI. Middle coronal slices from all 27 sub-spaces were visualized (lower left panel). The number of overlays, as well as sub-spaces’ overlays, were showed (upper right panel). The incorrect labels (red arrow) in one sub-space were corrected in final segmentation by performing majority vote label fusion.

2.0.2 Network Tiles:

After affine registration, all training brains were mapped to the same MNI atlas space ( voxels with 1 mm isotropic resolution). We employed 3D U-Net as a network tiles to cover entire MNI space with/without overlaps (Fig. 2). To be compatible with 133 labels, the number of channels of deconvolutional layers in each 3D U-Net were defined as 133 (Fig. 1). Each sub-space was presented by one coordinate and sub-space size , as

(1)

As showed in Fig. 2, SLANT-8 covered the MNI space using eight U-Nets by covering non-overlapped sub-spaces. To improve spatial consistency at boundaries, SLANT-27 covered overlapped sub-spaces.

Label Fusion: For SLANT-27, whose sub-spaces were overlapped, the label fusion method were employed to get a single segmentation from overlapped sub-spaces. Briefly, the k segmentations from network tiles were fused to achieve final segmentation in MNI space by performing majority vote:

(2)

where represents possible labels for a given voxel . if , while , otherwise. Then, the was registered to the original space by conducting another affine registration [11]. When using SLANT-8, whose sub-spaces were not overlapped, the native concatenation was applied rather than performing the label fusion.

Figure 3: This figure demonstrates the major components of different segmentation methods. “(45)” indicated the 45 OASIS manually traced images were used in training, while “(5111)” indicated the 5111 auxiliary label images were used in training.

2.0.3 Boost Learning on Unlabeled Data:

Similar to [12], the auxiliary labels on large-scale initially unlabeled MRI scans were obtained by performing existing segmentation tools. Briefly, MAS using hierarchical non-local spatial staple (NLSS) label fusion [1] was performed on 5111 multi-site scans. Next, the large-scale auxiliary labels were used for pre-training. Then, the small-scale manually labeled training data were used for fine-tuning the network.

3 Experiments

Training Cohort: 45 T1-weighted (T1w) MRI scans from Open Access Series on Imaging Studies (OASIS) dataset [9] were manually labeled to 133 labels according to BrainCOLOR protocol [6]. 5111 multi-site T1w MRI scans for auxiliary labels were achieved from night different projects (described in [6]).

Testing Cohort 1: Five withheld T1w MRI scans from OASIS dataset with manual segmentation (BrainCOLOR protocol) were used for validation, which evaluates the performance of different methods on the same site testing data.

Testing Cohort 2: One T1 MRI scan from colin27 cohort [4] with manual segmentation (BrainCOLOR protocol) was used for testing. This cohort evaluates the performance of different methods on a widely used standard template.

Testing Cohort 3: 13 T1 MRI scans from Child and Adolescent Neuro Development Initiative (CANDI) [7] were used for testing. This cohort evaluates the performance of different methods on an independent population, whose age range (5-15 yrs.) was not covered by OASIS training cohort (18-96 yrs.).

Figure 4: Qualitative results of manual segmentation, MAS methods, patch-based DCNN method, U-Net approaches and proposed SLANT methods.

3.0.1 Experimental Design:

The experimental design is presented in Fig. 3. First, two state-of-the-art multi-atlas label fusion methods, joint label fusion (JLF) [14] and non-local spatial staple (NLSS) [1], were used as baseline methods. The parameters were set the same as the papers’, which were optimized for whole brain segmentation. Next, patch-based network [2] and naive 3D U-Net [3] methods were used using their open-source implementations. By using affine registration as preprocessing, “Reg.+U-Net” was trained using 45 manually labeled scans and 5111 auxiliary labeled scans. Then, the proposed SLANT methods were evaluated on covering eight non-overlapped sub-spaces (“SLANT-8”) and 27 overlapped sub-spaces (“SLANT-27”), trained by 5111 auxiliary labeled scans. Last, 45 manually labeled scans were used to fine-tune the SLANT networks.

For all 3D U-Net in baseline methods and SLANT networks, we used the same parameters with 3D batch size = 1, input resolution =

, input channel = 1, output channel = 133, optimizer = “Adam”, learning rate = 0.0001. The deep networks can fit into an NVIDIA Titan GPU with 12 GB memory. For all the training using 5111 scans, 6 epochs were trained (

24 training hours); while for all the training using 45 scans, 1000 epochs were trained to ensure the similar training batches as 5111 scans. For the fine-tuning using 45 scans, 30 epochs were trained.

Results reported in this paper were from the epoch with best overall performance on OASIS cohort for each method, so that colin27 and CANDI were independent testing cohorts as external validation.

Figure 5: From quantitative results, The propsed SLANT-27 using 5111 auxiliary labels for pretraining and fine-tuned (“FT”) by 45 manual labels achieved highest median Dice similarity coefficient (DSC) values. The SLANT-27 was used as the reference method (“REF”) in statistical analysis. The significant difference to REF was marked with “*”.

4 Result

The qualitative and quantitative results have been shown in Fig. 4 and 5. In Fig. 5, ”45” indicated the 45 OASIS manually traced images were used in training, while ”5111” indicated the 5111 auxiliary label images were used in training. The mean Dice similarity coefficient (DSC) values on 132 anatomical labels (excluding background) between automatic segmentation methods with manual segmentation in original image space were showed as boxplots in Fig. 5. From the results, the affine registration (Reg. + U-Net) significantly leveraged the U-Net performance (compared with Naive U-Net). For the same network (Reg. + U-Net), results using 5111 auxiliary labeled scans achieved better performance than using 45 manual labeled scans. From Table 1, the proposed SLANT-27 method with fine-tuning achieved superior performance on mean DSC across testing cohorts. All claims of statistical significance in this paper have been calculated using the Wilcoxon signed rank test for p 0.05.

5 Conclusion and Discussion

In this study, we developed the SLANT method to combine the canonical medical image processing approaches with localized 3D FCN networks in MNI space. For the same network (Reg. + U-Net), results from 5111 auxiliary labeled scans achieved better performance than the results from 45 manual labeled scans. From Fig. 4 and 5, and Table 1, we demonstrate that our proposed strategy successfully takes advantages of historical efforts (registration, harmonization, and label-fusion) and consistently yields superior performance. Moreover, the proposed method requires 15 minutes, compared with 30 hours by MAS. Note that the 3D U-Net in the proposed SLANT can be replaced by other 3D segmentation networks, which might yield better performance.

Methods
Training
Scan #
OASIS Dataset Colin27 CANDI Dataset
mean std median DSC mean std
median
JLF [14] 45 0.746 0.009 0.746 0.646 0.590 0.033 0.585
NLSS [1] 45 0.760 0.012 0.756 0.712 0.677 0.029 0.680
Patch DCNN [2] 45 0.702 0.011 0.701 0.012 0.409 0.038 0.422
Naive U-Net [3] 45 0.606 0.012 0.605 0.000 0.375 0.043 0.380
Reg. + U-Net 45 0.706 0.009 0.711 0.621 0.514 0.081 0.536
Reg, + U-Net 5111 0.726 0.012 0.722 0.695 0.669 0.023 0.671
SLANT-8 5111 0.753 0.011 0.750 0.717 0.694 0.024 0.694
SLANT-8+FT 5111+45 0.768 0.011 0.763 0.726 0.704 0.025 0.705
SLANT-27 5111 0.759 0.011 0.754 0.721 0.694 0.024 0.697
SLANT-27+FT 5111+45 0.776 0.012 0.775 0.732 0.711 0.023 0.712
Table 1: Mean, std and median DSC values on three validation cohorts

5.0.1 Acknowledgments:

This research was supported by NSF CAREER 1452485, NIH R01EB017230, R21EY024036, R21NS064534, R01EB006136, R03EB012461, R01NS095291, Intramural Research Program, National Institute on Aging, NIH.

References

  • [1]

    Asman, A.J., Landman, B.A.: Hierarchical performance estimation in the statistical label fusion framework. Medical image analysis

    18(7), 1070–1081 (2014)
  • [2] de Brébisson, A., Montana, G.: Deep neural networks for anatomical brain segmentation. arXiv preprint arXiv:1502.02445 (2015)
  • [3] Çiçek, Ö., Abdulkadir, A., Lienkamp, S.S., Brox, T., Ronneberger, O.: 3d u-net: learning dense volumetric segmentation from sparse annotation. In: International Conference on Medical Image Computing and Computer-Assisted Intervention. pp. 424–432. Springer (2016)
  • [4] Collins, D.L., Zijdenbos, A.P., Kollokian, V., Sled, J.G., Kabani, N.J., Holmes, C.J., Evans, A.C.: Design and construction of a realistic digital brain phantom. IEEE transactions on medical imaging 17(3), 463–468 (1998)
  • [5] Evans, A.C., Collins, D.L., Mills, S., Brown, E., Kelly, R., Peters, T.M.: 3d statistical neuroanatomical models from 305 mri volumes. In: Nuclear Science Symposium and Medical Imaging Conference, 1993., 1993 IEEE Conference Record. pp. 1813–1817. IEEE (1993)
  • [6] Huo, Y., Aboud, K., Kang, H., Cutting, L.E., Landman, B.A.: Mapping lifetime brain volumetry with covariate-adjusted restricted cubic spline regression from cross-sectional multi-site mri. In: International Conference on Medical Image Computing and Computer-Assisted Intervention. pp. 81–88. Springer (2016)
  • [7] Kennedy, D.N., Haselgrove, C., Hodge, S.M., Rane, P.S., Makris, N., Frazier, J.A.: Candishare: a resource for pediatric neuroimaging data (2012)
  • [8] Li, W., Wang, G., Fidon, L., Ourselin, S., Cardoso, M.J., Vercauteren, T.: On the compactness, efficiency, and representation of 3d convolutional networks: brain parcellation as a pretext task. In: International Conference on Information Processing in Medical Imaging. pp. 348–360. Springer (2017)
  • [9] Marcus, D.S., Wang, T.H., Parker, J., Csernansky, J.G., Morris, J.C., Buckner, R.L.: Open access series of imaging studies (oasis): cross-sectional mri data in young, middle aged, nondemented, and demented older adults. Journal of cognitive neuroscience 19(9), 1498–1507 (2007)
  • [10] Mehta, R., Majumdar, A., Sivaswamy, J.: Brainsegnet: a convolutional neural network architecture for automated segmentation of human brain structures. Journal of Medical Imaging 4(2), 024003 (2017)
  • [11] Ourselin, S., Roche, A., Subsol, G., Pennec, X., Ayache, N.: Reconstructing a 3d structure from serial histological sections. Image and vision computing 19(1-2), 25–31 (2001)
  • [12] Roy, A.G., Conjeti, S., Sheet, D., Katouzian, A., Navab, N., Wachinger, C.: Error corrective boosting for learning fully convolutional networks with limited data. In: International Conference on Medical Image Computing and Computer-Assisted Intervention. pp. 231–239. Springer (2017)
  • [13] Wachinger, C., Reuter, M., Klein, T.: Deepnat: Deep convolutional neural network for segmenting neuroanatomy. NeuroImage (2017)
  • [14] Wang, H., Yushkevich, P.: Multi-atlas segmentation with joint label fusion and corrective learning—an open source implementation. Frontiers in neuroinformatics 7,  27 (2013)