This analysis applies transfer learning with ResNet18 on ultrasound images to enhance accuracy in breast cancer detection.
Early detection of breast cancer is crucial for improving patient survival rates. This study implements a Convolutional Neural Network (CNN) architecture based on ResNet18 using a transfer learning approach to classify breast ultrasound (USG) images into three categories: normal, benign, and malignant. The dataset, comprising 1,578 grayscale images collected from Baheya Hospital in Egypt, underwent preprocessing steps including image conversion, normalization, and augmentation. The ResNet18 model was fine-tuned using selective layer unfreezing to better adapt to the medical imaging domain. Evaluation was conducted using stratified 5-fold cross-validation and assessed with accuracy, precision, recall, F1-score, and AUC metrics. The best results were achieved by fine-tuning layer2, layer3, and the fully connected layer, yielding 95% accuracy, a macro F1-score of 0.93, and an AUC of 0.9906. The findings demonstrate that ResNet18, when properly fine-tuned with transfer learning, delivers high performance in breast cancer detection via ultrasound and holds strong potential as a reliable clinical decision-support tool.
No takes yet. Share an insight, caveat, or question.
Siden et al. (2025) studied this question.
Synapse has enriched 5 closely related papers on similar clinical questions. Consider them for comparative context: