METHOD OF CLASSIFYING IMAGES OF MOLES FOR DETECTING MELANOMA
Keywords:
melanoma, dermoscopy, convolutional neural network, transfer learning, EfficientNet, image classification, Grad-CAMAbstract
The article presents method for automatic malignant melanoma detection based on dermoscopic images. The subsystem is built on the EfficientNet-B0 convolutional neural network for classifying skin lesions as benign or malignant, while the Grad-CAM method is used to visualize and interpret the model’s decisions.
References
Esteva, A., Kuprel, B., Novoa, R. A., Ko, J., Swetter, S. M., Blau, H. M., & Thrun, S. (2017). Dermatologist-level classification of skin cancer with deep neural networks. Nature, 542(7639), 115–118. https://doi.org/10.1038/nature21056
Tan, M., & Le, Q. V. (2019). EfficientNet: Rethinking model scaling for convolutional neural networks. Proceedings of the 36th International Conference on Machine Learning (ICML 2019), 6105–6114. https://doi.org/10.48550/arXiv.1905.11946
Mittal, B. (2024). Melanoma Cancer Image Dataset. Kaggle. https://www.kaggle.com/datasets/bhaveshmittal/melanoma-cancer-dataset/data
Selvaraju, R. R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., & Batra, D. (2017). Grad-CAM: Visual explanations from deep networks via gradient-based localization. Proceedings of the IEEE International Conference on Computer Vision (ICCV 2017), 618–626. https://doi.org/10.1109/ICCV.2017.74
Codella, N., Gutman, D., Celebi, M. E., Helba, B., Marchetti, M. A., Dusza, S., & Halpern, A. (2018). Skin lesion analysis toward melanoma detection: ISIC 2017 challenge. Proceedings of IEEE ISBI 2018, 168–172. https://doi.org/10.1109/ISBI.2018.8363547
