Deep learning for cancer tumor classification using transfer learning and feature concatenation

Abdallah Mohamed Hassan, Mohamed Bakry El-Mashade, Ashraf Aboshosha

Abstract


Deep convolutional neural networks (CNNs) represent one of the
state-of-the-art methods for image classification in a variety of fields. Because the number of training dataset images in biomedical image classification is limited, transfer learning with CNNs is frequently applied. Breast cancer is one of most common types of cancer that causes death in women. Early detection and treatment of breast cancer are vital for improving survival rates. In this paper, we propose a deep neural network framework based on the transfer learning concept for detecting and classifying breast cancer histopathology images. In the proposed framework, we extract features from images using three pre-trained CNN architectures: VGG-16, ResNet50, and Inception-v3, and concatenate their extracted features, and then feed them into a fully connected (FC) layer to classify benign and malignant tumor cells in the histopathology images of the breast cancer. In comparison to the other CNN architectures that use a single CNN and many conventional classification methods, the proposed framework outperformed all other deep learning architectures and achieved an average accuracy of 98.76%.

Keywords


Breast cancer; Cancer tumor; Classification; Deep learning; Feature concatenation; Transfer learning

Full Text:

PDF


DOI: http://doi.org/10.11591/ijece.v12i6.pp6736-6743

Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

International Journal of Electrical and Computer Engineering (IJECE)
p-ISSN 2088-8708, e-ISSN 2722-2578