Advertisement

Automatic Brain Tumor Segmentation with Contour Aware Residual Network and Adversarial Training

  • Hao-Yu YangEmail author
  • Junlin Yang
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11384)

Abstract

Localizing brain tumor and identifying different subtypes of tissues play a crucial role in treatment assessment and management of gliomas. In this paper, we present a contour-aware 3D convolution neural network (CNN) with adversarial training for segmenting gliomas and an ensemble of models for overall survival prediction. For the segmentation task, contour loss and adversarial loss are added as an auxiliary information in addition to the pixel-wise classification loss to ensure the segmentation results mimic the contours of the ground truth annotation. We employed both random-forest-based and neural-network-based regression scores for predicting overall survival time. Hand-crafted imaging feature incorporated with the non-imaging feature is employed. The proposed method was evaluated on the BraTS 2018 dataset and achieved competitive results for both segmentation and survival prediction tasks. We demonstrate that raw segmentation results can be improved by incorporating extra constraints in contours and adversarial training.

Keywords

Neural networks Adversarial training Contour aware 

References

  1. 1.
    Bakas, S., et al.: Advancing the cancer genome atlas glioma MRI collections with expert segmentation labels and radiomic features. Nature Sci. Data 4, 170117 (2017)CrossRefGoogle Scholar
  2. 2.
    Bakas, S., et al.: Segmentation labels and radiomic features for the pre-operative scans of the TCGA-GBM collection (2017)Google Scholar
  3. 3.
    Bakas, S., et al.: Segmentation labels and radiomic features for the pre-operative scans of the TCGA-LGG collection. In: The Cancer Imaging Archive (2017)Google Scholar
  4. 4.
    Bakas, S., et al.: Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challenge. ArXiv e-prints, November 2018Google Scholar
  5. 5.
    Chen, H., Qi, X., Yu, L., Heng, P.-A.: DCAN: deep contour-aware networks for accurate gland segmentation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2487–2496 (2016)Google Scholar
  6. 6.
    Cheng, J.-X., Zhang, X., Liu, B.-L.: Health-related quality of life in patients with high-grade glioma. Neuro-oncology 11(1), 41–50 (2009)CrossRefGoogle Scholar
  7. 7.
    Goodfellow, I., et al.: Generative adversarial nets. In: Ghahramani, Z., Welling, M., Cortes, C., Lawrence, N.D., Weinberger, K.Q. (eds.) Advances in Neural Information Processing Systems 27, pp. 2672–2680. Curran Associates Inc (2014)Google Scholar
  8. 8.
    He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. CoRR abs/1512.03385 (2015)Google Scholar
  9. 9.
    Kass, M., Witkin, A., Terzopoulos, D.: Snakes: active contour models. Int. J. Comput. Vis. 1(4), 321–331 (1988)CrossRefGoogle Scholar
  10. 10.
    Li, X., Chen, H., Qi, X., Dou, Q., Fu, C., Heng, P.: H-DenseUNet: Hybrid densely connected UNet for liver and liver tumor segmentation from CT volumes. CoRR abs/1709.07330 (2017)Google Scholar
  11. 11.
    Luc, P., Couprie, C., Chintala, S., Verbeek, J.: Semantic segmentation using adversarial networks. CoRR abs/1611.08408 (2016)Google Scholar
  12. 12.
    Menze, B.H., et al.: The multimodal brain tumor image segmentation benchmark (brats). IEEE Trans. Med. Imaging 34(10), 1993–2024 (2015)CrossRefGoogle Scholar
  13. 13.
    Nyúl, L.G., Udupa, J.K., Zhang, X.: New variants of a method of MRI scale standardization. IEEE Trans. Med. Imaging 19(2), 143–150 (2000)CrossRefGoogle Scholar
  14. 14.
    Quan, T.M., Hildebrand, D.G.C., Jeong, W.: Fusionnet: A deep fully residual convolutional neural network for image segmentation in connectomics. CoRR abs/1612.05360 (2016)Google Scholar
  15. 15.
    Tustison, N.J., et al.: N4ITK: Improved N3 bias correction. IEEE Trans. Med. Imaging 29(6), 1310–1320 (2010)CrossRefGoogle Scholar
  16. 16.
    Wen, P.Y., Kesari, S.: Malignant gliomas in adults. New Engl, J. Med. 359(5), 492–507 (2008)CrossRefGoogle Scholar
  17. 17.
    Yushkevich, P.A., et al.: User-guided 3D active contour segmentation of anatomical structures: significantly improved efficiency and reliability. Neuroimage 31(3), 1116–1128 (2006)CrossRefGoogle Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  1. 1.Yale UniversityNew HavenUSA
  2. 2.Cura Cloud CooperationSeattleUSA

Personalised recommendations