Skip to main content
Top

Hint

Swipe to navigate through the chapters of this book

2018 | OriginalPaper | Chapter

Ensemble of Multi-sized FCNs to Improve White Matter Lesion Segmentation

Authors : Zhewei Wang, Charles D. Smith, Jundong Liu

Published in: Machine Learning in Medical Imaging

Publisher: Springer International Publishing

share
SHARE

Abstract

In this paper, we develop a two-stage neural network solution for the challenging task of white-matter lesion segmentation. To cope with the vast variability in lesion sizes, we sample brain MR scans with patches at three different dimensions and feed them into separate fully convolutional neural networks (FCNs). In the second stage, we process large and small lesion separately, and use ensemble-nets to combine the segmentation results generated from the FCNs. A novel activation function is adopted in the ensemble-nets to improve the segmentation accuracy measured by Dice Similarity Coefficient. Experiments on MICCAI 2017 White Matter Hyperintensities (WMH) Segmentation Challenge data demonstrate that our two-stage-multi-sized FCN approach, as well as the new activation function, are effective in capturing white-matter lesions in MR images.
Literature
1.
go back to reference Anbeek, P., et al.: Automatic segmentation of different-sized white matter lesions by voxel probability estimation. Med. Image Anal. 8(3), 205–215 (2004) Anbeek, P., et al.: Automatic segmentation of different-sized white matter lesions by voxel probability estimation. Med. Image Anal. 8(3), 205–215 (2004)
3.
go back to reference Chen, Y., Shi, B., Wang, Z., Zhang, P., Smith, C.D., Liu, J.: Hippocampus segmentation through multi-view ensemble ConvNets. In: 14th IEEE, ISBI 2017, pp. 192–196 (2017) Chen, Y., Shi, B., Wang, Z., Zhang, P., Smith, C.D., Liu, J.: Hippocampus segmentation through multi-view ensemble ConvNets. In: 14th IEEE, ISBI 2017, pp. 192–196 (2017)
4.
go back to reference Çiçek, Ö., et al.: 3D U-Net: learning dense volumetric segmentation from sparse annotation. In: MICCAI, pp. 424–432. Springer, Berlin (2016) Çiçek, Ö., et al.: 3D U-Net: learning dense volumetric segmentation from sparse annotation. In: MICCAI, pp. 424–432. Springer, Berlin (2016)
5.
go back to reference Drozdzal, M., et al.: The importance of skip connections in biomedical image segmentation. In: Deep Learning and Data Labeling for Medical Applications, pp. 179–187. Springer, Chem (2016) Drozdzal, M., et al.: The importance of skip connections in biomedical image segmentation. In: Deep Learning and Data Labeling for Medical Applications, pp. 179–187. Springer, Chem (2016)
6.
go back to reference Ekanayake, J., et al.: Generalised wasserstein dice score for imbalanced multi-class segmentation using holistic convolutional networks. In: Third International Workshop, BrainLes, MICCAI (2017) Ekanayake, J., et al.: Generalised wasserstein dice score for imbalanced multi-class segmentation using holistic convolutional networks. In: Third International Workshop, BrainLes, MICCAI (2017)
7.
go back to reference Li, H., et al.: Fully convolutional network ensembles for white matter hyperintensities segmentation in MR images. arXiv preprint arXiv:​1802.​05203 (2018) Li, H., et al.: Fully convolutional network ensembles for white matter hyperintensities segmentation in MR images. arXiv preprint arXiv:​1802.​05203 (2018)
8.
go back to reference Liu, J., Smith, C.D., Chebrolu, H.: Automatic multiple sclerosis detection based on integrated square estimation. In: IEEE, CVPR Workshops, 20–25 June, 2009, pp. 31–38 (2009) Liu, J., Smith, C.D., Chebrolu, H.: Automatic multiple sclerosis detection based on integrated square estimation. In: IEEE, CVPR Workshops, 20–25 June, 2009, pp. 31–38 (2009)
9.
go back to reference Long, J., et al.: Fully convolutional networks for semantic segmentation. In: Proceedings of CVPR, pp. 3431–3440 (2015) Long, J., et al.: Fully convolutional networks for semantic segmentation. In: Proceedings of CVPR, pp. 3431–3440 (2015)
10.
go back to reference Milletari, F., et al.: V-net: fully convolutional neural networks for volumetric medical image segmentation. In: 3D Vision (3DV). pp. 565–571. IEEE (2016) Milletari, F., et al.: V-net: fully convolutional neural networks for volumetric medical image segmentation. In: 3D Vision (3DV). pp. 565–571. IEEE (2016)
11.
go back to reference Ronneberger, O., Fischer, P., Brox, T.: U-net: convolutional networks for biomedical image segmentation. In: MICCAI, pp. 234–241. Springer, Berlin (2015) Ronneberger, O., Fischer, P., Brox, T.: U-net: convolutional networks for biomedical image segmentation. In: MICCAI, pp. 234–241. Springer, Berlin (2015)
12.
go back to reference Souplet, J.C., et al.: An automatic segmentation of T2-FLAIR multiple sclerosis lesions. In: The MIDAS Journal-MS Lesion Segmentation (MICCAI 2008 Workshop). Citeseer (2008) Souplet, J.C., et al.: An automatic segmentation of T2-FLAIR multiple sclerosis lesions. In: The MIDAS Journal-MS Lesion Segmentation (MICCAI 2008 Workshop). Citeseer (2008)
13.
go back to reference Sudre, C.H., et al.: Generalised dice overlap as a deep learning loss function for highly unbalanced segmentations. In: Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, pp. 240–248. Springer, Berlin (2017) Sudre, C.H., et al.: Generalised dice overlap as a deep learning loss function for highly unbalanced segmentations. In: Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support, pp. 240–248. Springer, Berlin (2017)
14.
go back to reference Warfield, S., Tomas-Fernandez, X.: Lesion segmentation. In: Toga, A.W. (ed.) Brain Mapping, pp. 323–332. Academic Press, Waltham (2015) CrossRef Warfield, S., Tomas-Fernandez, X.: Lesion segmentation. In: Toga, A.W. (ed.) Brain Mapping, pp. 323–332. Academic Press, Waltham (2015) CrossRef
Metadata
Title
Ensemble of Multi-sized FCNs to Improve White Matter Lesion Segmentation
Authors
Zhewei Wang
Charles D. Smith
Jundong Liu
Copyright Year
2018
DOI
https://doi.org/10.1007/978-3-030-00919-9_26

Premium Partner