Y-Net: Joint Segmentation and Classification for Diagnosis of Breast Biopsy Images
In this paper, we introduce a conceptually simple network for generating discriminative tissue-level segmentation masks for the purpose of breast cancer diagnosis. Our method efficiently segments different types of tissues in breast biopsy images while simultaneously predicting a discriminative map for identifying important areas in an image. Our network, Y-Net, extends and generalizes U-Net by adding a parallel branch for discriminative map generation and by supporting convolutional block modularity, which allows the user to adjust network efficiency without altering the network topology. Y-Net delivers state-of-the-art segmentation accuracy while learning \(6.6\times \) fewer parameters than its closest competitors. The addition of descriptive power from Y-Net’s discriminative segmentation masks improve diagnostic classification accuracy by 7% over state-of-the-art methods for diagnostic classification. Source code is available at: https://sacmehta.github.io/YNet.
Research reported in this publication was supported by the National Cancer Institute awards R01 CA172343, R01 CA140560, and RO1 CA200690. We would also like to thank NVIDIA Corporation for donating the Titan X Pascal GPU used for this research.
- 4.Fine, R.E.: Diagnostic Techniques, 2nd edn. B. C. Decker Inc, Ontario (2006)Google Scholar
- 5.Geçer, B.: Detection and classification of breast cancer in whole slide histopathology images using deep convolutional networks. Ph.D. thesis, Bilkent University (2016)Google Scholar
- 6.He, K., et al.: Deep residual learning for image recognition. In: CVPR (2016)Google Scholar
- 7.He, K., et al.: Mask R-CNN. In: ICCV (2017)Google Scholar
- 8.Hou, L., et al.: Patch-based convolutional neural network for whole slide tissue image classification. In: CVPR (2016)Google Scholar
- 10.Mehta, S., et al.: Learning to segment breast biopsy whole slide images. In: WACV (2018)Google Scholar
- 11.Ronneberger, O., Fischer, P., Brox, T.: U-net: Convolutional networks for biomedical image segmentation. In: MICCAI (2015)Google Scholar
- 12.Zhao, H., et al.: Pyramid scene parsing network. In: CVPR (2017)Google Scholar