Using deep learning to segment breast and fibroglandular tissue in MRI volumes

Purpose: Automated segmentation of breast and fibroglandular tissue (FGT) is required for various computer‐aided applications of breast MRI. Traditional image analysis and computer vision techniques, such atlas, template matching, or, edge and surface detection, have been applied to solve this task. However, applicability of these methods is usually limited by the characteristics of the images used in the study datasets, while breast MRI varies with respect to the different MRI protocols used, in addition to the variability in breast shapes. All this variability, in addition to various MRI artifacts, makes it a challenging task to develop a robust breast and FGT segmentation method using traditional approaches. Therefore, in this study, we investigated the use of a deep‐learning approach known as “U‐net.” Materials and methods: We used a dataset of 66 breast MRI's randomly selected from our scientific archive, which includes five different MRI acquisition protocols and breasts from four breast density categories in a balanced distribution. To prepare reference segmentations, we manually segmented breast and FGT for all images using an in‐house developed workstation. We experimented with the application of U‐net in two different ways for breast and FGT segmentation. In the first method, following the same pipeline used in traditional approaches, we trained two consecutive (2C) U‐nets: first for segmenting the breast in the whole MRI volume and the second for segmenting FGT inside the segmented breast. In the second method, we used a single 3‐class (3C) U‐net, which performs both tasks simultaneously by segmenting the volume into three regions: nonbreast, fat inside the breast, and FGT inside the breast. For comparison, we applied two existing and published methods to our dataset: an atlas‐based method and a sheetness‐based method. We used Dice Similarity Coefficient (DSC) to measure the performances of the automated methods, with respect to the manual segmentations. Additionally, we computed Pearson's correlation between the breast density values computed based on manual and automated segmentations. Results: The average DSC values for breast segmentation were 0.933, 0.944, 0.863, and 0.848 obtained from 3C U‐net, 2C U‐nets, atlas‐based method, and sheetness‐based method, respectively. The average DSC values for FGT segmentation obtained from 3C U‐net, 2C U‐nets, and atlas‐based methods were 0.850, 0.811, and 0.671, respectively. The correlation between breast density values based on 3C U‐net and manual segmentations was 0.974. This value was significantly higher than 0.957 as obtained from 2C U‐nets (P < 0.0001, Steiger's Z‐test with Bonferoni correction) and 0.938 as obtained from atlas‐based method (P = 0.0016). Conclusions: In conclusion, we applied a deep‐learning method, U‐net, for segmenting breast and FGT in MRI in a dataset that includes a variety of MRI protocols and breast densities. Our results showed that U‐net‐based methods significantly outperformed the existing algorithms and resulted in significantly more accurate breast density computation.

[1]  Jennifer A Harvey,et al.  Breast MR imaging artifacts: how to recognize and fix them. , 2007, Radiographics : a review publication of the Radiological Society of North America, Inc.

[2]  Marcin Andrychowicz,et al.  Learning to learn by gradient descent by gradient descent , 2016, NIPS.

[3]  Nico Karssemeijer,et al.  Segmentation of the Pectoral Muscle in Breast MRI Using Atlas-Based Approaches , 2012, MICCAI.

[4]  Min-Ying Su,et al.  Development of a quantitative method for analysis of breast density based on three-dimensional breast MRI. , 2008, Medical physics.

[5]  N. Boyd,et al.  Mammographic density and breast cancer risk: current understanding and future prospects , 2011, Breast Cancer Research.

[6]  Aaron Klein,et al.  Fast Bayesian Optimization of Machine Learning Hyperparameters on Large Datasets , 2016, AISTATS.

[7]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[8]  Jianhua Yao,et al.  Breast Tumor Analysis in Dynamic Contrast Enhanced MRI Using Texture Features and Wavelet Transform , 2009, IEEE Journal of Selected Topics in Signal Processing.

[9]  T. Ivanovska,et al.  A Level Set Based Framework for Quantitative Evaluation of Breast Tissue Density from MRI Data , 2014, PloS one.

[10]  Nico Karssemeijer,et al.  Volumetric Breast Density Estimation from Full-Field Digital Mammograms: A Validation Study , 2014, IEEE Transactions on Medical Imaging.

[11]  Martin J. Yaffe,et al.  MRI Volumetric Analysis of Breast Fibroglandular Tissue to Assess Risk of the Spared Nipple in BRCA1 and BRCA2 Mutation Carriers , 2014, Annals of Surgical Oncology.

[12]  Lei Wang,et al.  Fully automatic breast segmentation in 3D breast MRI , 2012, 2012 9th IEEE International Symposium on Biomedical Imaging (ISBI).

[13]  Anne L. Martel,et al.  Automatic atlas-based segmentation of the breast in MRI for 3D breast volume computation. , 2012, Medical physics.

[14]  Heinz-Otto Peitgen,et al.  Computer assistance for MR based diagnosis of breast cancer: Present and future challenges , 2007, Comput. Medical Imaging Graph..

[15]  F. Gao,et al.  Confirmation of double-peaked time distribution of mortality among Asian breast cancer patients in a population-based study , 2007, Breast Cancer Research.

[16]  Nico Karssemeijer,et al.  A fully automated system for quantification of background parenchymal enhancement in breast DCE-MRI , 2016, SPIE Medical Imaging.

[17]  Farzad Khalvati,et al.  Automated Segmentation of Breast in 3-D MR Images Using a Robust Atlas , 2015, IEEE Transactions on Medical Imaging.

[18]  V. McCormack,et al.  Breast Density and Parenchymal Patterns as Markers of Breast Cancer Risk: A Meta-analysis , 2006, Cancer Epidemiology Biomarkers & Prevention.

[19]  Nico Karssemeijer,et al.  Breast Segmentation and Density Estimation in Breast MRI: A Fully Automatic Framework , 2015, IEEE Journal of Biomedical and Health Informatics.

[20]  T. Sellers,et al.  Mammographic density, breast cancer risk and risk prediction , 2007, Breast Cancer Research.

[21]  Yoshua Bengio,et al.  Random Search for Hyper-Parameter Optimization , 2012, J. Mach. Learn. Res..

[22]  Brian B. Avants,et al.  N4ITK: Improved N3 Bias Correction , 2010, IEEE Transactions on Medical Imaging.

[23]  Anil K. Jain,et al.  A modified Hausdorff distance for object matching , 1994, Proceedings of 12th International Conference on Pattern Recognition.

[24]  Yoshua Bengio,et al.  Understanding the difficulty of training deep feedforward neural networks , 2010, AISTATS.

[25]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[26]  Celia MT Greenwood,et al.  A genome-wide linkage study of mammographic density, a risk factor for breast cancer , 2011, Breast Cancer Research.

[27]  Anne L. Martel,et al.  Breast segmentation in MRI using Poisson surface reconstruction initialized with random forest edge detection , 2016, SPIE Medical Imaging.

[28]  A. Bert,et al.  A fully automatic algorithm for segmentation of the breasts in DCE-MR images , 2010, 2010 Annual International Conference of the IEEE Engineering in Medicine and Biology.

[29]  J. H. Steiger Tests for comparing elements of a correlation matrix. , 1980 .

[30]  Nico Karssemeijer,et al.  Towards Accurate Segmentation of Fibroglandular Tissue in Breast MRI Using Fuzzy C-Means and Skin-Folds Removal , 2015, ICIAP.

[31]  Matthias Koenig,et al.  Automatic segmentation of relevant structures in DCE MR mammograms , 2007, SPIE Medical Imaging.

[32]  Nico Karssemeijer,et al.  Automated localization of breast cancer in DCE-MRI , 2015, Medical Image Anal..

[33]  Min-Ying Su,et al.  Template-based automatic breast segmentation on MRI by excluding the chest region. , 2013, Medical physics.

[34]  Ron Kikinis,et al.  Statistical validation of image segmentation quality based on a spatial overlap index. , 2004, Academic radiology.

[35]  Olga Chambers,et al.  Automated breast-region segmentation in the axial breast MR images , 2015, Comput. Biol. Medicine.

[36]  Shandong Wu,et al.  Automated fibroglandular tissue segmentation and volumetric density estimation in breast MRI using an atlas-aided fuzzy C-means method. , 2013, Medical physics.