DOI QR코드

DOI QR Code

Breast Mass Classification using the Fundamental Deep Learning Approach: To build the optimal model applying various methods that influence the performance of CNN

  • Lee, Jin (Bachelor of Arts degree in biology from Taylor University) ;
  • Choi, Kwang Jong (Korea Christian International School) ;
  • Kim, Seong Jung (Biomedical Engineering Branch, Division of Precision Medicine and Cancer Informatics, National Cancer Center) ;
  • Oh, Ji Eun (Biomedical Engineering Branch, Division of Precision Medicine and Cancer Informatics, National Cancer Center) ;
  • Yoon, Woong Bae (Biomedical Engineering Branch, Division of Precision Medicine and Cancer Informatics, National Cancer Center) ;
  • Kim, Kwang Gi (Biomedical Engineering Branch, Division of Precision Medicine and Cancer Informatics, National Cancer Center)
  • Received : 2016.09.01
  • Accepted : 2016.10.10
  • Published : 2016.09.30

Abstract

Deep learning enables machines to have perception and can potentially outperform humans in the medical field. It can save a lot of time and reduce human error by detecting certain patterns from medical images without being trained. The main goal of this paper is to build the optimal model for breast mass classification by applying various methods that influence the performance of Convolutional Neural Network (CNN). Google's newly developed software library Tensorflow was used to build CNN and the mammogram dataset used in this study was obtained from 340 breast cancer cases. The best classification performance we achieved was an accuracy of 0.887, sensitivity of 0.903, and specificity of 0.869 for normal tissue versus malignant mass classification with augmented data, more convolutional filters, and ADAM optimizer. A limitation of this method, however, was that it only considered malignant masses which are relatively easier to classify than benign masses. Therefore, further studies are required in order to properly classify any given data for medical uses.

Keywords

References

  1. Lindsey A. Torre, Freddie Bray, Rebecca L. Siegel, Jacques Ferlay, Joannie Lortet-Tieulent and Ahmedin Jemal, "Global Cancer Statistics," A Cancer Journal for Clinicians, vol. 65, no. 2, pp. 87-108, 2015. https://doi.org/10.3322/caac.21262
  2. "American Cancer Society Recommendations for Early Breast Cancer Detection in Women without Breast Symptoms." American Cancer Society. N.p., 20 Oct. 2015. 14 July 2016.
  3. Arzav Jain and Daniel Levy, "DeepMammo - Breast Mass Classification Using Deep Convolutional Neural Networks," pp. 1-7, 15 July 2016.
  4. J.J. Fenton, L. Abraham, S. H. Taplin, B.M. Geller, P.A. Carney, C. D'Orsi, J. G. Elmore, W. E. Barlow, "Effectiveness of computer-aided detection in community mammography practice," J. Natl. Cancer Inst., vol. 103, pp. 1152-1161, 2011. https://doi.org/10.1093/jnci/djr206
  5. Thijs Kooi, Albert Gubern-Merida, Jan-Jurre Mordang, Ritse Mann, Ruud Pijnappel, Klaas Schuur, Ard Den Heeten and Nico Karssemeijer, "A Comparison Between a Deep Convolutional Neural Network and Radiologists for Classifying Regions of Interest in Mammography," Breast Imaging Lecture Notes in Computer Science, vol. 9699, pp. 51-56, 2016.
  6. Mordang, Jan-Jurre, Tim Janssen, Alessandro Bria, Thijs Kooi, Albert Gubern-Merida, and Nico Karssemeijer, "Automatic Microcalcification Detection in Multi-vendor Mammography Using Convolutional Neural Networks," Breast Imaging Lecture Notes in Computer Science, pp. 35-42, 2016.
  7. N. Srivastava, G. Hinton, A. Krizhevsky, I. Sutskever and R. Salakhutdinov, "Dropout: A Simple Way to Prevent Neural Networks from Overfitting," J. Mach. Learn. Res. 15, pp. 1929-1958, 2014
  8. Samer Hijazi, Rishi Kumar and Chris Rowen, "Using Convolutional Neural Networks for Image Recognition," Candence, pp.1-12, 2015.
  9. Diederik P. Kingma and Jimmy Lei Ba. "Adam: A Method for Stochastic Optimization," ICLR 2015, pp. 1-15, 2015.
  10. B. Sahiner, Heang-Ping Chan, N. Petrick, Datong Wei, M.A. Helvie, D.D. Adler, and M.M. Goodsitt, "Classification of Mass and Normal Breast Tissue: A Convolution Neural Network Classifier with Spatial Domain and Texture Images," IEEE Transactions on Medical Imaging, vol. 15, no.5, pp. 598-610, 1996. https://doi.org/10.1109/42.538937