- Original Article
- Open access
- Published:
Convolutional neural network for breast cancer diagnosis using diffuse optical tomography
Visual Computing for Industry, Biomedicine, and Art volume 2, Article number: 1 (2019)
Abstract
We have developed a computer-aided diagnosis system based on a convolutional neural network that aims to classify breast mass lesions in optical tomographic images obtained using a diffuse optical tomography system, which is suitable for repeated measurements in mass screening. Sixty-three optical tomographic images were collected from women with dense breasts, and a dataset of 1260 2D gray scale images sliced from these 3D images was built. After image preprocessing and normalization, we tested the network on this dataset and obtained 0.80 specificity, 0.95 sensitivity, 90.2% accuracy, and 0.94 area under the receiver operating characteristic curve (AUC). Furthermore, a data augmentation method was implemented to alleviate the imbalance between benign and malignant samples in the dataset. The sensitivity, specificity, accuracy, and AUC of the classification on the augmented dataset were 0.88, 0.96, 93.3%, and 0.95, respectively.
Introduction
Breast cancer is the most common cancer among women. To reduce the mortality of breast cancer, early detection and an accurate diagnosis are important [1]. Thus, a promotable and sensitive detection technology with efficient statistical analysis methods are necessary for mass screening for breast cancer.
There are currently several clinical methods for breast cancer detection including X-ray mammography, magnetic resonance imaging (MRI), and ultrasound. X-ray mammography is the most commonly used method for breast detection; however, it may cause damage owing to ionization, making it unsuitable for repeated mass screening measurements [2, 3]. MRI can offer excellent images of breast tissue with higher sensitivities; however, MRI incurs high costs, low specificities, and is not very convenient, which greatly limits its application [4,5,6]. As a relatively inexpensive and nonionizing radiation method, ultrasound can differentiate between benign and malignant masses; however, it has a low sensitivity and is highly dependent on the skill of the technician [4, 7]. Other methods, such as photoacoustic imaging, are also emerging as diagnostic techniques for breast cancer detection [8,9,10]. Diffuse optical tomography (DOT) is an emerging method for breast tumor diagnosis that provides optical properties of breast tissue correlating to the tumor’s physiological signatures [11, 12]. As an imaging modality with endogenous contrast, DOT has the potential to overcome the limitations of the abovementioned modalities, particularly when screening dense breasts, in terms of safety, cost, portability, sensitivity, and specificity. DOT also allows for improved discrimination between malignant and benign lesions, making it a competitive mass screening method for breast cancer.
Thus far, several studies have applied computer-aided diagnosis (CAD) techniques to breast cancer detection; these techniques include the use of artificial neural networks [13,14,15], fuzzy logic [16, 17], Bayesian networks [18, 19], decision trees [20, 21], and k-means clustering [22, 23]. However, few researchers have implemented CAD methods with DOT to diagnose breast cancer. A study analyzed the absorption, scattering, and refractive index images obtained by a phase-contrast diffuse optical tomography system and demonstrated the sensitivity and specificity to be 0.82 and 0.92, respectively, with a support vector machine (SVM) classifier [24]. Another study proposed a semi-automatic detection method of malignant breast lesions in DOT images using logistic regression of three optically measured physiological parameters; this method had an average sensitivity and specificity of 0.89 and 0.89, respectively [25].
Recently, convolutional neural networks (CNNs) have been proven to work well in the differentiation between benign and malignant breast lesions [26,27,28,29,30]. Compared with traditional methods, CNNs reduce the steps involved in image feature extraction; alternatively, they feed image data directly into the network that can learn discriminative features automatically. CNN architecture is particularly adapted to take advantage of the 2D structure of the input image.
In this work, a dataset of DOT breast images with 63 biopsy-confirmed tumor-bearing dense breasts were used. Dense breasts are common among young women and cannot be examined by existing mammography imaging systems [31]. All images were collected from a DOT system and reconstructed with a finite element algorithm [32]. In our previous work, an SVM classifier was used to classify breast tumors and achieved an accuracy of 71.7%. This study develops an optimized CNN adapted to the characteristics of diffusion tomography images; this proposed CNN shows a high accuracy on the breast cancer dataset.
Methods
CNNs require a large amount of training data to achieve high accuracies. However, it is difficult to build a comprehensive dataset with a large number of breast tumor images. The collected 3D images are gray scale images with a size of 91 × 91 × 31. Figure 1 illustrates DOT images of a benign tumor-bearing breast and a malignant tumor-bearing breast. It is difficult to distinguish tumor and normal tissue from the images owing to the high breast density. We found that most tumors are distributed in the range of 6 to 25 on the Z-axis of the 3D image by matching the biopsy result to the image and chose to slice images from this range. A total of 20 slices in each 3D image was used for this dataset. The total dataset consisted of 1260 2D images, obtained from slicing the 63 3D images. Each 2D image is resized to a size of 32 × 32 and normalized for CNN input. The dataset was randomly divided into training and test groups that contained 75% and 25% of the data, respectively, as shown in Table 1.
A CNN-based CAD method was proposed to classify breast mass lesions. The architecture of our network is summarized in Fig. 2. It contains five learned layers including two convolutional, two batch normalization, and one fully-connected layer. The input of the network is a 32 × 32 Gy scale image. The first convolutional layer, C1, filters the 32 × 32 input gray scale image with six kernels of size 5 × 5 with a stride of 1 pixel. The second convolutional layer, C4, takes the normalized and pooled output of the first convolutional layer as its input and filters it with 12 kernels of size 5 × 5.
Each convolutional layer is followed by a batch normalization layer, which aims to stabilize the distributions of layer inputs by controlling the mean and variance [33]. It adds two extra parameters per activation to preserve the representation ability of the network. Because of batch normalization, the network can tolerate increased training rates and often does not require dropouts for regularization. The normalized results then pass through a sigmoid nonlinear function as an output for these layers.
Pooling layers perform local averaging and subsampling, thereby reducing the resolution of the feature map and the sensitivity of the output to shifts and distortions. Pooling layers have a filter size of 2 × 2 pixels and output the average value offour inputs in each local region.
A simple logistic classifier is used in the final layer of the network. Malignant and benign are represented with 2-bit one-hot encoding as (1,0) and (0,1) in the output layer. The mini-batch stochastic gradient descent momentum is used for training. We also adopted an early stopping strategy to monitor the test misclassification rate (MCR), where training is terminated if no progress is noted on the test set. Optimum results were obtained by tuning the hyper parameters through an extensive set of trial and error experiments; finally, we used a learning rate of 0.3, mini-batch size of 83, and maximum epochs of 500.
Owing to the imbalance between the number of benign and malignant samples in the dataset, we use two groups of sets for training. The first set is the original training set and includes 945 images, and the second set, which includes 1245 images, is the augmented version of the first group where the number of malignant images was doubled by flipping the images. The distribution of the augmented dataset is given in Table 2. Training images are used to train the network and the remaining ones are used for testing. The network predictions and the biopsy results were compared using a receiver operating characteristic (ROC) analysis. Differences in performance are evaluated by computing the area under the ROC curves (AUC).
Results
Our results on the DOT breast dataset are summarized in Table 3. Sensitivity and specificity are statistical indexes of the performance of a binary classification test; they measure the proportion of correctly identified positives and negatives. Our network achieves a test accuracy rate of 90.2%, a specificity of 0.80, and a sensitivity of 0.95 compared with the original data. More satisfactory results are obtained when the network is trained on the augmented data, with a test accuracy rate of 93.3%, a specificity of 0.88, and a sensitivity of 0.96; this is the best performance achieved in this study. The MCR curve of the training and test sets during training is shown in Fig. 3a. The test set was checked during training to monitor progress and to apply the early stopping strategy when a plateau was reached. The proposed network can be trained faster and requires fewer training steps (four times) to achieve the same accuracy compared with a simple CNN without batch normalization. The minimum test MCR was achieved after approximately 200 epochs during training. There was no obvious overfitting when data augmentation and batch normalization are used. The ROC curve comparing the performance of the network trained on the original and augmented data is shown in Fig. 3b. The AUC is 0.94 for the network trained on the original data and 0.95 for one trained on the augmented data.
The network is also trained in five groups of learning rates to obtain the best performance. The learning rate defines the speed at which the weight is updated on each epoch in the neural network. A network may learn fast but may be unstable and exhibit a very high learning rate. Conversely, a network that learns slowly may be easily susceptible local minimum trapping. Comparing the ROC curves in Fig. 3c, the suitable learning rate is determined to be 0.3.
During the study, we repeated the experiment several times and found that the test results show subtle differences. Thus, we trained and evaluated the CNN using a 10-fold cross-validation for robust testing. The ROC curve obtained is shown in Fig. 3d and the average AUC is found to be 0.93 (±0.03). This gives a sensitivity of 0.79 and a specificity of 0.97. It indicates that the CNN model is robust and is reliable for breast tumor classification. The average test accuracy rate reaches 91% with a standard deviation of 1.66%; this is a promising result achieved with little sample data.
Conclusions
Our results in this work show that it is possible to achieve a satisfactory result on breast cancer diagnosis using a CNN trained with DOT breast images. A DOT breast dataset is built; it includes 63 patient samples with malignant or benign tumors, for a total of 1260 2D gray scale images. Although it is an arduous task to detect a dense breast with a small set of data, the proposed 8-layer CNN has a good capability of classifying image patterns while achieving a specificity and sensitivity of 0.88 and 0.96, respectively. This technique has the potential to assist radiologists in diagnosing breast cancer and improving the diagnostic rate that furthermore promotes mass screening for breast cancer.
In future studies, more samples will be collected to improve generalization as the accuracy can be further improved with a larger dataset. It is important to carefully select a better convolutional network architecture and optimization algorithm. We would like to train a 3D CNN on 3D data, where spatial structures can provide information that may be missing or is far less obvious in 2D images. Unsupervised pre-training is also being considered for our diagnosis method in the future.
The images used in this study are reconstructed with optical absorption distribution of tissue; however, DOT not only provides the optical absorption coefficients but can also determine the optical scattering coefficients. As a functional optical imaging system, DOT can quantify the tumor hemoglobin concentration and blood oxygen saturation; these are directly related to tumor angiogenesis. The focus of our next research study will be how to make full use of this information.
References
Siegel RL, Miller KD, Jemal A (2017) Cancer statistics, 2017. CA Cancer J Clin 67(1):7–30. https://doi.org/10.3322/caac.21387
de González AB, Darby S (2004) Risk of cancer from diagnostic X-rays. Lancet 363(9424):1910. https://doi.org/10.1016/S0140-6736(04)16374-5
Ronckers CM, Erdmann CA, Land CE (2004) Radiation and breast cancer: a review of current evidence. Breast Cancer Res 7(1):21–32. https://doi.org/10.1186/bcr970
Elmore JG, Armstrong K, Lehman CD, Fletcher SW (2005) Screening for breast cancer. JAMA 293(10):1245–1256. https://doi.org/10.1001/jama.293.10.1245
Ikeda DM, Baker DR, Daniel BL (2000) Magnetic resonance imaging of breast cancer: clinical indications and breast MRI reporting system. J Magn Reson Imaging 12(6):975–983. https://doi.org/10.1002/1522-2586(200012)12:6<975::AID-JMRI24>3.0.CO;2-Y
Orel SG (2000) MR imaging of the breast. Radiol Clin N Am 38(4):899–913. https://doi.org/10.1016/S0033-8389(05)70208-6
Berg WA, Blume JD, Cormack JB, Mendelson EB, Lehrer D, Böhm-Vélez M et al (2008) Combined screening with ultrasound and mammography vs mammography alone in women at elevated risk of breast cancer. JAMA 299(18):2151–2163. https://doi.org/10.1001/jama.299.18.2151
Kruger RA, Kuzmiak CM, Lam RB, Reinecke DR, Del Rio SP, Steed D (2013) Dedicated 3D photoacoustic breast imaging. Med Phys 40(11):113301. https://doi.org/10.1118/1.4824317
Chen Q, Guo H, Jin T, Qi WZ, Xie HK, Xi L (2018) Ultracompact high-resolution photoacoustic microscopy. Opt Lett 43(7):1615–1618. https://doi.org/10.1364/OL.43.001615
Qin W, Jin T, Guo H, Xi L (2018) Large-field-of-view optical resolution photoacoustic microscopy. Opt Express 26(4):4271–4278. https://doi.org/10.1364/OE.26.004271
Jiang HB, Xu Y, Iftimia N, Eggert J, Klove K, Baron L et al (2001) Three-dimensional optical tomographic imaging of breast in a human subject. IEEE Trans Med Imaging 20(12):1334–1340. https://doi.org/10.1109/42.974928
Xi L, Li XQ, Yao L, Grobmyer S, Jiang HB (2012) Design and evaluation of a hybrid photoacoustic tomography and diffuse optical tomography system for breast cancer detection. Med Phys 39(5):2584–2594. https://doi.org/10.1118/1.3703598
Cheng HD, Cui MY (2004) Mass lesion detection with a fuzzy neural network. Pattern Recogn 37(6):1189–1200. https://doi.org/10.1016/j.patcog.2003.11.002
Fogel DB, Wasson EC, Boughton EM, Porto VW (1998) Evolving artificial neural networks for screening features from mammograms. Artif Intell Med 14(3):317–326. https://doi.org/10.1016/S0933-3657(98)00040-2
Floyd CE Jr, Lo JY, Yun AJ, Sullivan DC, Kornguth PJ (2015) Prediction of breast cancer malignancy using an artificial neural network. Cancer 74(11):2944–2948. https://doi.org/10.1002/1097-0142(19941201)74:11<2944::AID-CNCR2820741109>3.0.CO;2-F
Bhattacharya M, Das A (2007) Fuzzy logic based segmentation of microcalcification in breast using digital mammograms considering multiresolution. In: proceedings of the international machine vision and image processing conference, IEEE, Kildare, 5-7 September 2007, pp 98-105. https://doi.org/10.1109/IMVIP.2007.33
Kovalerchuk B, Triantaphyllou E, Ruiz JF, Clayton J (1997) Fuzzy logic in computer-aided breast cancer diagnosis: analysis of lobulation. Artif Intell Med 11(1):75–85. https://doi.org/10.1016/S0933-3657(97)00021-3
Zheng B, Chang YH, Wang XH, Good WF, Gur D (1999) Application of a Bayesian belief network in a computer-assisted diagnosis scheme for mass detection. In: Proceedings of SPIE 3661, medical imaging 1999: image processing. SPIE, San Diego, CA, United States, pp 1553–1561. https://doi.org/10.1117/12.348558
Viton JL, Rasigni M, Rasigni G, Llebaria A (1996) Method for characterizing masses in digital mammograms. Opt Eng 35(12):3453–3459. https://doi.org/10.1117/1.601107
Zhen L, Chan AK (2001) An artificial intelligent algorithm for tumor detection in screening mammogram. IEEE Trans Med Imaging 20(7):559–567. https://doi.org/10.1109/42.932741
Li LH, Qian W, Clarke LP, Clark RA, Thomas JA (1999) Improving mass detection by adaptive and multiscale processing in digitized mammograms. In: Proceedings of SPIE 3661, medical imaging 1999: image processing. SPIE, San Diego, CA, United States, pp 490–498. https://doi.org/10.1117/12.348604
Patel BC, Sinha GR (2010) An adaptive K-means clustering algorithm for breast image segmentation. Int J Comput Appl 10(4):35–38. https://doi.org/10.5120/1467-1982
Zhao HP, Li LH, Xu WD, Zhang J (2010) A novel clustering method based on k-means with region growing for micro-calcifications in mammographic images. In: proceedings of 2010 international conference on computer and information application, IEEE, Tianjin, China, 3-5 December 2010, pp 1-4. https://doi.org/10.1109/ICCIA.2010.6141521
Wang JZ, Liang XP, Zhang QZ, Fajardo LL, Jiang HB (2008) Automated breast cancer classification using near-infrared optical tomographic images. J Biomed Opt 13(4):044001. https://doi.org/10.1117/1.2956662
Busch DR, Guo WS, Choe R, Durduran T, Feldman MD, Mies C et al (2010) Computer aided automatic detection of malignant lesions in diffuse optical mammography. Med Phys 37(4):1840–1849. https://doi.org/10.1118/1.3314075
Peng W, Mayorga RV, Hussein EMA (2016) An automated confirmatory system for analysis of mammograms. Comput Methods Prog Biomed 125:134–144. https://doi.org/10.1016/j.cmpb.2015.09.019
Arevalo J, González FA, Ramos-Pollán R, Oliveira JL, Lopez MAG (2016) Representation learning for mammography mass lesion classification with convolutional neural networks. Comput Methods Prog Biomed 127:248–257. https://doi.org/10.1016/j.cmpb.2015.12.014
Carneiro G, Nascimento J, Bradley AP (2015) Unregistered multiview mammogram analysis with pre-trained deep learning models. In: proceedings of the 18th international conference on medical image computing and computer-assisted intervention. Munich, Germany, 5-9 October 2015, pp 652-660. https://doi.org/10.1007/978-3-319-24574-4_78
Jiao ZC, Gao XB, Wang Y, Li J (2016) A deep feature based framework for breast masses classification. Neurocomputing 197:221–231. https://doi.org/10.1016/j.neucom.2016.02.060
Chougrad H, Zouaki H, Alheyane O (2018) Deep convolutional neural networks for breast cancer screening. Comput Methods Prog Biomed 157:19–30. https://doi.org/10.1016/j.cmpb.2018.01.011
Joy JE, Penhoet EE, Petitti DB (2005) Benefits and limitations of mammography. In: Joy JE, Penhoet EE, Petitti DB (eds) Saving women's lives: strategies for improving breast cancer detection and diagnosis. National Academies Press, Washington, DC, United States
Jiang HB (2010) Diffuse optical tomography: principles and applications. CRC Press, Boca Raton, FL, United States. https://doi.org/10.1201/b10482
Ioffe S, Szegedy C (2015) Batch normalization: accelerating deep network training by reducing internal covariate shift. arXiv preprint arXiv:1502.03167
Acknowledgements
This research was supported by the University of Electronic Science and Technology of China.
Funding
China Postdoctoral Science Foundation (No. 2018M633347).
Availability of data and materials
The datasets generated and/or analyzed during the current study are not publicly available due to personal privacy but are available from the corresponding author on reasonable request.
Author information
Authors and Affiliations
Contributions
All authors read and approved the final manuscript.
Corresponding author
Ethics declarations
Competing interests
The authors declare that they have no competing interests.
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Open Access This article is distributed under the terms of the Creative Commons Attribution 4.0 International License (http://creativecommons.org/licenses/by/4.0/), which permits unrestricted use, distribution, and reproduction in any medium, provided you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made.
About this article
Cite this article
Xu, Q., Wang, X. & Jiang, H. Convolutional neural network for breast cancer diagnosis using diffuse optical tomography. Vis. Comput. Ind. Biomed. Art 2, 1 (2019). https://doi.org/10.1186/s42492-019-0012-y
Received:
Accepted:
Published:
DOI: https://doi.org/10.1186/s42492-019-0012-y