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

• Published in: The journal of multimedia information system Vol. 3; no. 3; pp. 97 - 102
• Main Authors: Lee, Jin, Choi, Kwang Jong, Kim, Seong Jung, Oh, Ji Eun, Yoon, Woong Bae, Kim, Kwang Gi
• Format: Journal Article
• Language: Korean
• Published: 2016
• Subjects: Deep learning; Tensorflow; Breast mass; Classification
• ISSN: 2383-7632

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.