Performance analysis of seven Convolutional Neural Networks (CNNs) with transfer learning for Invasive Ductal Carcinoma (IDC) grading in breast histopathological images
Document Type
Article
Publication Date
11-10-2022
Abstract
Computer-aided Invasive Ductal Carcinoma (IDC) grading classification systems based on deep learning have shown that deep learning may achieve reliable accuracy in IDC grade classification using histopathology images. However, there is a dearth of comprehensive performance comparisons of Convolutional Neural Network (CNN) designs on IDC in the literature. As such, we would like to conduct a comparison analysis of the performance of seven selected CNN models: EfficientNetB0, EfficientNetV2B0, EfficientNetV2B0-21k, ResNetV1-50, ResNetV2-50, MobileNetV1, and MobileNetV2 with transfer learning. To implement each pre-trained CNN architecture, we deployed the corresponded feature vector available from the TensorFlowHub, integrating it with dropout and dense layers to form a complete CNN model. Our findings indicated that the EfficientNetV2B0-21k (0.72B Floating-Point Operations and 7.1 M parameters) outperformed other CNN models in the IDC grading task. Nevertheless, we discovered that practically all selected CNN models perform well in the IDC grading task, with an average balanced accuracy of 0.936 +/- 0.0189 on the cross-validation set and 0.9308 +/- 0.0211on the test set.
Keywords
Histologic grade, Cancer, Classification
Divisions
biomedengine
Funders
Universiti Tunku Abdul Rahman Research Fund (Grant No: FRGS/1/2019/ICT04/UTAR/02/1, 8073/Y01 & IPSR/RMC/UTARRF/2022-C1/H01)
Publication Title
Scientific Reports
Volume
12
Issue
1
Publisher
Nature Portfolio
Publisher Location
HEIDELBERGER PLATZ 3, BERLIN, 14197, GERMANY