Advertisement

Stroke Lesion Segmentation with 2D Novel CNN Pipeline and Novel Loss Function

  • Pengbo LiuEmail author
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11383)

Abstract

Recently, CT perfusion (CTP) has been used to triage ischemic stroke patients in the early stage, because of its speed, availability, and lack of contraindications. But CTP data alone, even with the generated perfusion maps is not enough to describe the precise location of infarct core or penumbra. Considering the good performance demonstrated on Diffusion Weighted Imaging (DWI), We propose a CTP data analysis technique using Generative Adversarial Networks (GAN) [2] to generate DWI, and segment the regions of ischemic stroke lesion on top of the generated DWI based on convolutional neutral network (CNN) and a novel loss function. Specifically, our CNN structure consists of a generator, a discriminator and a segmentator. The generator synthesizes DWI from CT, which generates a high quality representation for subsequent segmentator. Meanwhile the discriminator competes with the generator to identify whether its input DWI is real or generated. And we propose a novel segmentation loss function that contains a weighted cross-entropy loss and generalized dice loss [1] to balance the positive and negative loss in the training phase. And the weighted cross entropy loss can highlight the area of stroke lesion to enhance the structural contrast. We also introduce other techniques in our network, like GN [13], Maxout [14] in the proposed network. Data augmentation is also used in the training phase. In our experiments, an average dice coefficient of 60.65% is achieved with four-fold cross-validation. From the results of our experiments, this novel network combined with the proposed loss function achieved a better performance in CTP data analysis than traditional methods using CTP perfusion parameters.

Keywords

Generation Segmentation Cross-entropy loss Generalized dice loss 

References

  1. 1.
    Sudre, C.H., Li, W., Vercauteren, T., et al.: Generalised dice overlap as a deep learning loss function for highly unbalanced segmentations (2017)CrossRefGoogle Scholar
  2. 2.
    Goodfellow, I., et al.: Generative adversarial nets. In: Advances in Neural Information Processing Systems (2014)Google Scholar
  3. 3.
    Yu, Y., Han, Q., Ding, X., et al.: Defining core and penumbra in ischemic stroke: a voxel- and volume-based analysis of whole brain CT perfusion. Sci. Rep. 6, 20932 (2015)CrossRefGoogle Scholar
  4. 4.
    The top 10 causes of death (2017). http://www.who.int/mediacentre/factsheets/fs310/en/. Accessed 30 June 2017
  5. 5.
    Mouridsen, K., Hansen, M.B., Østergaard, L., Jespersen, S.N.: Reliable estimation of capillary transit time distributions using DSC-MRI. J. Cereb. Blood Flow Metab. 34, 1511–1521 (2014).  https://doi.org/10.1038/jcbfm.2014.111CrossRefGoogle Scholar
  6. 6.
    Nielsen, A., Hansen, M.B., Tietze, A., et al.: Prediction of tissue outcome and assessment of treatment effect in acute ischemic stroke using deep learning. Stroke 49, 1349–1401 (2018)CrossRefGoogle Scholar
  7. 7.
    Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. In: International Conference on Neural Information Processing Systems, pp. 1097–1105. Curran Associates Inc. (2012)Google Scholar
  8. 8.
    Ren, S., He, K., Girshick, R., et al.: Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2017)CrossRefGoogle Scholar
  9. 9.
    Long, J., Shelhamer, E., Darrell, T.: Fully convolutional networks for semantic segmentation. IEEE Trans. Pattern Anal. Mach. Intell. PP(99), 1 (2014)Google Scholar
  10. 10.
    Ciresan, D., Giusti, A., Gambardella, L.M., Schmidhuber, J.: Deep neural networks segment neuronal membranes in electron microscopy images. In: Advances in Neural Information Processing Systems, pp. 2843–2851 (2012)Google Scholar
  11. 11.
    Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015).  https://doi.org/10.1007/978-3-319-24574-4_28CrossRefGoogle Scholar
  12. 12.
    Glorot, X., Bordes, A., Bengio, Y.: Deep sparse rectifier neural networks. In: International Conference on Artificial Intelligence and Statistics, pp. 315–323 (2011)Google Scholar
  13. 13.
    Wu, Y., He, K.: Group normalization (2018)CrossRefGoogle Scholar
  14. 14.
    Goodfellow, I.J., Warde-Farley, D., Mirza, M., et al.: Maxout networks. arXiv preprint arXiv:1302.4389 (2013)
  15. 15.
    Mao, X., Li, Q., Xie, H., et al.: Least squares generative adversarial networks (2016)Google Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  1. 1.Beijing University of TechnologyBeijingChina

Personalised recommendations