Two-Stage Fine-Tuning of ResNet50 for High-Sensitivity Melanoma Detection on Dermoscopic Images
Description
Melanoma is the most dangerous form of skin cancer, and catching it early significantly improves patient outcomes. Five-year survival rates exceed 99% when detected at an early stage but fall sharply once the cancer spreads. This paper proposes and evaluates a two-stage fine-tuning approach for ResNet50 applied to binary melanoma classification on dermoscopic images. The core challenges addressed are class imbalance and suboptimal transfer learning. Random oversampling was used to achieve a 1:1 class balance on the HAM10000 dataset. Stage 1 trained only the classification head with the ResNet50 base frozen, while Stage 2 fine-tuned all layers jointly at a low learning rate of 1e-5 to prevent catastrophic forgetting. On an independent test set of 3,826 images, the model achieved an AUC-ROC of 0.9559, accuracy of 88.34%, sensitivity of 87.56%, and specificity of 89.13%. Code and a working Streamlit detection application are available at https://github.com/Aryanbhagat23/melanoma-detection
Files
Melanoma_Paper_Aryan_Bhagat.pdf
Files
(460.0 kB)
| Name | Size | Download all |
|---|---|---|
|
md5:5fdaac6d3cc116d32a876e3917a2da40
|
460.0 kB | Preview Download |
Additional details
Related works
- Is supplemented by
- Software: https://github.com/Aryanbhagat23/melanoma-detection (URL)
Software
- Repository URL
- https://github.com/Aryanbhagat23/melanoma-detection
- Programming language
- Python