ResNet for Histopathologic Cancer Detection, the Deeper, the Better?

Apply

ResNet for Histopathologic Cancer Detection, the Deeper, the Better?

Journal Title: Journal of Data Science and Intelligent Systems - Year 2024, Vol 2, Issue 4

Abstract

Histopathological image classification has become one of the most challenging tasks for researchers, due to the varied categories and detailed differences within diseases. In this study, we investigate the critical role of network depth in histopathological image classification, utilizing deep residual convolutional neural networks (ResNet). We evaluate the efficacy of two transfer learning strategies using ResNet with varying layers (18, 34, 50, 152) pretrained on ImageNet. Specifically, we analyze whether a deeper network or the fine-tuning of all layers in pre-trained ResNets enhances performance compared to freezing most layers and training only the last layer. Conducted on Kaggle's dataset of 220,025 labeled histopathology patches, our findings reveal that increasing the depth of ResNet does not guarantee better accuracy (ResNet-34 AUC: 0.992 vs. ResNet-152 AUC: 0.989). Instead, dataset-specific semantic features and the cost of training should guide model selection. Furthermore, deep ResNet outperforms traditional logistic regression (ResNet AUC: up to 0.992 vs. logistic regression AUC: 0.775), showcasing superior generalization and robustness. Notably, the strategy of freezing most layers doesn't improve the accuracy and efficiency of transfer learning and the performance of both transfer strategies depends largely on the types of data. Overall, both methods produce satisfactory results in comparison to models trained from scratch or conventional machine learning models.

Authors and Affiliations

Ziying Wang, Jinghong Gao, Hangyi Kan, Yang Huang, Furong Tang, Wen Li, Fenglong Yang

Keywords

EP ID EP752194
DOI 10.47852/bonviewJDSIS3202744
Views 6
Downloads 0