An Efficient Method for Breast Mass Classification Using Pre-Trained Deep Convolutional Networks

Ebtihal Al-Mansour, Muhammad Hussain, Hatim A. Aboalsamh and Fazal-e-Amin
Additional contact information
Ebtihal Al-Mansour: Department of Computer Science, CCIS, King Saud University, Riyadh 11451, Saudi Arabia
Muhammad Hussain: Department of Computer Science, CCIS, King Saud University, Riyadh 11451, Saudi Arabia
Hatim A. Aboalsamh: Department of Computer Science, CCIS, King Saud University, Riyadh 11451, Saudi Arabia
Fazal-e-Amin: Department of Software Engineering, CCIS, King Saud University, Riyadh 11451, Saudi Arabia

Mathematics, 2022, vol. 10, issue 14, 1-19

Abstract: Masses are the early indicators of breast cancer, and distinguishing between benign and malignant masses is a challenging problem. Many machine learning- and deep learning-based methods have been proposed to distinguish benign masses from malignant ones on mammograms. However, their performance is not satisfactory. Though deep learning has been shown to be effective in a variety of applications, it is challenging to apply it for mass classification since it requires a large dataset for training and the number of available annotated mammograms is limited. A common approach to overcome this issue is to employ a pre-trained model and fine-tune it on mammograms. Though this works well, it still involves fine-tuning a huge number of learnable parameters with a small number of annotated mammograms. To tackle the small set problem in the training or fine-tuning of CNN models, we introduce a new method, which uses a pre-trained CNN without any modifications as an end-to-end model for mass classification, without fine-tuning the learnable parameters. The training phase only identifies the neurons in the classification layer, which yield higher activation for each class, and later on uses the activation of these neurons to classify an unknown mass ROI. We evaluated the proposed approach using different CNN models on the public domain benchmark datasets, such as DDSM and INbreast. The results show that it outperforms the state-of-the-art deep learning-based methods.

Keywords: CNN; feature extraction; feature reduction; breast cancer; classification; deep learning (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2022
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/10/14/2539/pdf (application/pdf)
https://www.mdpi.com/2227-7390/10/14/2539/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:10:y:2022:i:14:p:2539-:d:868174

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().