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.
Copyrights © 2025