Skip to main content

Convolutional neural network for breast cancer diagnosis using diffuse optical tomography

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.

Fig. 1
figure 1

Examples of diffuse optical tomography breast images. a 3D image of a benign breast tumor with a size of 3.0*1.3*1.3, b 2D image sliced from Fig. 1a, c 3D image of a malignant breast tumor with a size of 2.4*1.3*1.2, d 2D image sliced from Fig. 1c

Table 1 2D diffuse optical tomography image dataset and its division into training and test sets

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.

Fig. 2
figure 2

Architecture of the convolutional neural network

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).

Table 2 Augmented dataset and its division into training and test sets

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.

Table 3 Comparison summary of the convolutional neural network performance on original and augmented data
Fig. 3
figure 3

a Misclassification rate curve of training set and testing set during training, b Receiver operating characteristic (ROC) comparison of original data and augmented data, c ROC comparison of convolutional neural network (CNN) with different learning rate, d ROC of CNN evaluated by a 10-fold cross-validation

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

  1. 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

    Article  Google Scholar 

  2. 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

    Article  Google Scholar 

  3. 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

    Article  Google Scholar 

  4. 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

    Article  Google Scholar 

  5. 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

    Article  Google Scholar 

  6. 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

    Article  Google Scholar 

  7. 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

    Article  Google Scholar 

  8. 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

    Article  Google Scholar 

  9. 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

    Article  Google Scholar 

  10. 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

    Article  Google Scholar 

  11. 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

    Article  Google Scholar 

  12. 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

    Article  Google Scholar 

  13. 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

    Article  Google Scholar 

  14. 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

    Article  Google Scholar 

  15. 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

    Article  Google Scholar 

  16. 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

  17. 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

    Article  Google Scholar 

  18. 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

    Chapter  Google Scholar 

  19. 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

    Article  Google Scholar 

  20. 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

    Article  Google Scholar 

  21. 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

    Chapter  Google Scholar 

  22. 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

    Article  Google Scholar 

  23. 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

  24. 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

    Article  Google Scholar 

  25. 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

    Article  Google Scholar 

  26. 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

    Article  Google Scholar 

  27. 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

    Article  Google Scholar 

  28. 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

  29. 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

    Article  Google Scholar 

  30. 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

    Article  Google Scholar 

  31. 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

    Google Scholar 

  32. Jiang HB (2010) Diffuse optical tomography: principles and applications. CRC Press, Boca Raton, FL, United States. https://doi.org/10.1201/b10482

  33. Ioffe S, Szegedy C (2015) Batch normalization: accelerating deep network training by reducing internal covariate shift. arXiv preprint arXiv:1502.03167

Download references

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

Authors

Contributions

All authors read and approved the final manuscript.

Corresponding author

Correspondence to Huabei Jiang.

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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Received:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1186/s42492-019-0012-y

Keywords