Rao PMM, Singh SK, Khamparia A, Bhushan B, Podder P. Multi-class Breast Cancer Classification using Ensemble of Pretrained models and Transfer Learning.
Curr Med Imaging 2021;
18:409-416. [PMID:
33602102 DOI:
10.2174/1573405617666210218101418]
[Citation(s) in RCA: 1] [Impact Index Per Article: 0.3] [Reference Citation Analysis] [What about the content of this article? (0)] [Affiliation(s)] [Abstract] [Key Words] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 09/30/2020] [Revised: 12/21/2020] [Accepted: 01/06/2021] [Indexed: 02/08/2023]
Abstract
AIMS
Early detection of breast cancer has reduced many deaths. Earlier CAD systems used to be the second opinion for radiologists and clinicians. Machine learning and deep learning has brought tremendous changes in medical diagnosis and imagining.
BACKGROUND
Breast cancer is the most commonly occurring cancer in the women and it is the second most common cancer overall. According to the 2018 statistics, there were over 2million cases all over the world. Belgium and Luxembourg have the highest rate of cancer.
OBJECTIVE
Proposed a method for breast cancer detection using Ensemble learning. 2-class and 8-class classification is performed.
METHOD
To deal with imbalance classification the authors have proposed an ensemble of pretrained models.
RESULT
98.5% training accuracy and 89% of test accuracy are achieved on 8-class classification. And 99.1% and 98% train and test accuracy are achieved on 2 class classification.
CONCLUSION
it is found that there are high misclassifications in class DC when compared to the other classes, this is due to the imbalance in the dataset. In future, one can increase the size of the datasets or use different methods. In implement this research work, authors have used 2 Nvidia Tesla V100 GPU's in google cloud platform.
Collapse