Research Area: Machine Learning
Breast cancer is among the leading cause of mortality among women in developing as well as under-developing countries. The detection and classification of breast cancer in the early stages of its development may allow patients to have proper treatment. In this article, we proposed a novel deep learning framework for the detection and classification of breast cancer in breast cytology images using the concept of transfer learning. In general, deep learning architectures are modeled to be problem specific and is performed in isolation. Contrary to classical learning paradigms, which develop and yield in isolation, transfer learning is aimed to utilize the gained knowledge during the solution of one problem into another related problem. In the proposed framework, features from images are extracted using pre-trained CNN architectures, namely, GoogLeNet, Visual Geometry Group Network (VGGNet) and Residual Networks (ResNet), which are fed into a fully connected layer for classification of malignant and benign cells using average pooling classification. To evaluate the performance of the proposed framework, experiments are performed on standard benchmark data sets. It has been observed that the proposed framework outclass all the other deep learning architectures in terms of accuracy in detection and classification of breast tumor in cytology images.
Keywords:
Deep Learning
Detection
Classification
Breast Cancer
Transfer Learning
Machine Learning
Deep Learning
Author(s) Name: SanaUllah Khan, Naveed Islam, Zahoor Jan, Ikram Ud Din, Joel J.P.C. Rodrigues
Journal name: Pattern Recognition Letters
Conferrence name:
Publisher name: Elsevier
DOI: 10.1016/j.patrec.2019.03.022
Volume Information: Volume 125, 1 July 2019, Pages 1-6
