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
Related Articles
- EP ID EP752194
- DOI 10.47852/bonviewJDSIS3202744
- Views 13
- Downloads 0
3D-STCNN: Spatiotemporal Convolutional Neural Network Based on EEG 3D Features for Detecting Driving Fatigue
Fatigue driving has become one of the main causes of traffic accidents, and driving fatigue detection based on electroencephalogram (EEG) can effectively evaluate the driver's mental state and avoid the occurrence of tra...
Topological Data Analysis of COVID-19 Using Artificial Intelligence and Machine Learning Techniques in Big Datasets of Hausdorff Spaces
In this paper, we carry out an in-depth topological data analysis (TDA) of COVID-19 pandemic using artificial intelligence (AI) and Machine Learning (ML) techniques. We show the distribution patterns of pandemic all over...
Multiple Regression Model as Interpolation Through the Points of Weighted Means
A well-known property of the multiple linear regression is that its plane goes through the point of the mean values of all variables, and this feature can be used to find the model's intercept. This work shows that a re...
Intra-annual National Statistical Accounts Based on Machine Learning Algorithm
The methods used for forecasting financial series are based on the concept that a pattern can be identified in the data and distinguished from randomness by smoothing past values. This smoothing process eliminates random...
A Study of the Effects of the Shape Parameter and Type of Data Points Locations on the Accuracy of the Hermite-Based Symmetric Approach Using Positive Definite Radial Kernels
Theoretical approximation ideas served as the driving force behind the research. one can see that the shape parameter's behavior is driven by the kind of problem and the analytical standards that are applied. the primary...