Enchancing Robustness in Skin Lesion Detection: A Benchmark of 32 Models on a Novel Dataset Including Healthy Skin Images
Source abstract
Abstract
The rising incidence of skin cancer necessitates the development of automated, reliable diagnostic tools to support clinicians. While deep learning models, particularly from the YOLO family, have shown promise, their application in real-world scenarios is limited by challenges such as class imbalance and the inability to process images of healthy skin, leading to potential false positives. This study presents a comprehensive comparative analysis of 32 object detection models, primarily from the YOLO architecture (v5–v12) and RT-DETR, to identify the most effective solution for skin lesion detection. We curated a novel, balanced dataset of 10,000 images based on the ISIC archive, comprising 10 distinct lesion classes (benign and malignant). Crucially, we introduced a dedicated ‘background’ class containing 1000 images of clear skin, a novelty designed to enhance model robustness in clinical practice. Models were systematically evaluated and filtered based on performance metrics (mAP, Recall) and complexity. Through a multi-stage evaluation, the YOLOv9c model was identified as the superior architecture, achieving a mAP@50 of 72.5% and a Recall of 71.6% across all classes. The model demonstrated strong performance, especially considering the dataset’s complexity with 10 classes and background images. Our research establishes a new benchmark for skin lesion detection. We demonstrate that including a ‘background’ class is a critical step towards creating clinically viable tools. The YOLOv9c model emerges as a powerful and efficient solution. To foster further research, our curated 10-class dataset with background images will be made publicly available.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Cutaneous Melanoma Detection and Management · Nonmelanoma Skin Cancer Studies · Face recognition and analysis
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Faculty of Electrical Engineering and Computer Science, Lublin University of Technology, Nadbystrzycka 38A, 20-618 Lublin, Poland
Funding
the Ministry of Education and Science—Poland
References (36)
Show source-provided references
1. The global burden of skin cancer: A longitudinal analysis from the Global Burden of Disease Study, 1990–201710.1016/j.jdin.2020.10.013 ↗
2. World Health Organization (2025, October 01). The Effect of Occupational Exposure to Solar Ultraviolet Radiation on Malignant Skin Melanoma and Non-Melanoma Skin Cancer: A Systematic Review and Meta-Analysis from the WHO/ILO Joint Estimates of the Work-Related Burden of Disease and Injury, Available online: https://iris.who.int/handle/10665/350569.
3. Improving outcomes in patients with melanoma: Strategies to ensure an early diagnosis
4. Cutaneous Malignant Melanoma: A Review of Early Diagnosis and Management10.14740/wjon1349 ↗
5. Waseh, S., and Lee, J.B. (2023). Advances in melanoma: Epidemiology, diagnosis, and prognosis. Front. Med., 10.10.3389/fmed.2023.1268479 ↗
6. Skin Cancer diagnosis with Yolo Deep Neural Network10.1016/j.procs.2023.03.083 ↗
7. AlSadhan, N.A., Alamri, S.A., Ben Ismail, M.M., and Bchir, O. (2024). Skin Cancer Recognition Using Unified Deep Convolutional Neural Networks. Cancers, 16.10.3390/cancers16071246 ↗
8. Stefan, Ć., and Nikola, S. (2024, January 20–24). Application of the YOLO algorithm for Medical Purposes in the Detection of Skin Cancer. Proceedings of the 10th International Scientific Conference Technics, Informatics and Education-TIE 2024, Čačak, Serbia.
9. Riyadi, M.A., Ayuningtias, A., and Isnanto, R.R. (2024, January 26–27). Detection and Classification of Skin Cancer Using YOLOv8n. Proceedings of the 2024 11th International Conference on Electrical Engineering, Computer Science and Informatics (EECSI), Yogyakarta, Indonesia.10.1109/EECSI63442.2024.10776505 ↗
10. Real-Time Skin Disease Detection and Classification Using YOLOv8 Object Detection for Healthcare Diagnosis
11. Elshahawy, M., Elnemr, A., Oproescu, M., Schiopu, A.G., Elgarayhi, A., Elmogy, M.M., and Sallah, M. (2023). Early Melanoma Detection Based on a Hybrid YOLOv5 and ResNet Technique. Diagnostics, 13.10.20944/preprints202307.1970.v1 ↗
12. Improved Skin Lesion Segmentation in Dermoscopic Images Using Object Detection and Semantic Segmentation10.2147/CCID.S518751 ↗
13. Gül, S., Cetinel, G., Aydin, B.M., Akgün, D., and Öztaş Kara, R. (2025). YOLOSAMIC: A Hybrid Approach to Skin Cancer Segmentation with the Segment Anything Model and YOLOv8. Diagnostics, 15.10.3390/diagnostics15040479 ↗
14. Automatic Classification of Melanoma Skin Cancer with Deep Convolutional Neural Networks10.3390/ai3020029 ↗
15. SkinLiTE: Lightweight Supervised Contrastive Learning Model for Enhanced Skin Lesion Detection and Disease Typification in Dermoscopic Images10.2174/0115734056313837240612065845 ↗
16. Automatic Detection and Categorization of Skin Lesions for Early Diagnosis of Skin Cancer Using YOLO-v3-DCNN Architecture10.5566/ias.2773 ↗
17. Abdullah, K., Muhammad, S., Nauman, K., Ayman, Y., and Qaisar, A. (2025). CAD-Skin: A Hybrid Convolutional Neural Network–Autoencoder Framework for Precise Detection and Classification of Skin Lesions and Cancer. Bioengineering, 12.10.3390/bioengineering12040326 ↗
18. A Geometric algebra-enhanced network for skin lesion detection with diagnostic prior10.1007/s11227-024-06833-z ↗
19. Rashid, J., Ishfaq, M., Ali, G., Saeed, M.R., Hussain, M., Alkhalifah, T., Alturise, F., and Samand, N. (2022). Skin Cancer Disease Detection Using Transfer Learning Technique. Appl. Sci., 12.10.3390/app12115714 ↗
20. Hybrid Skin Lesion Detection Integrating CNN and XGBoost for Accurate Diagnosis
21. Region-of-Interest Based Transfer Learning Assisted Framework for Skin Cancer Detection10.1109/ACCESS.2020.3014701 ↗
22. Joseph, R., Santosh, D., Ross, G., and Ali, F. (July, January 26). You Only Look Once: Unified, Real-Time Object Detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.
23. Object Detection with Discriminatively Trained Part-Based Models10.1109/TPAMI.2009.167 ↗
24. Rahima, K., and Muhammad, H. (2024). What is YOLOv5: A deep look into the internal features of the popular object detector. arXiv.
25. Reis, D., Kupec, J., Hong, J., and Daoudi, A. (2023). Real-Time Flying Object Detection with YOLOv8. arXiv.
26. Wang, C.-Y., Yeh, I.-H., and Liao, H.-Y. (October, January 29). YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information. Proceedings of the 18th European Conference on Computer Vision (ECCV), Milan, Italy.
27. Wang, A., Chen, H., Liu, L., Chen, K., Lin, Z., Han, J., and Ding, G. (2024). YOLOv10: Real-Time End-to-End Object Detection. arXiv.
28. Khanam, R., and Hussain, M. (2024). YOLOv11: An Overview of the Key Architectural Enhancements. arXiv.
29. Ye, Q., and Doermann, D. (2025). YOLOv12: Attention-Centric Real-Time Object Detectors. arXiv.
30. Joseph, R., and Ali, F. (2017, January 21–26). YOLO9000: Better, Faster, Stronger. Proceedings of the 30th IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.
31. Joseph, R., and Ali, F. (2018). YOLOv3: An Incremental Improvement. arXiv.
32. Bochkovskiy, A., Wang, C.-Y., and Liao, H. (2020). YOLOv4: Optimal Speed and Accuracy of Object Detection. arXiv.
33. Li, C., Li, L., Jiang, H., Weng, K., Geng, Y., Li, L., Ke, Z., Li, Q., Cheng, M., and Nie, W. (2022). YOLOv6: A Single-Stage Object Detection Framework for Industrial Applications. arXiv.
34. Wang, C.-Y., Bochkovskiy, A., and Liao, H.-Y.M. (2023, January 17–24). YOLOv7: Trainable Bag-of-Freebies Sets New State-of-the-Art for Real-Time Object Detectors. Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Vancouver, BC, Canada.10.1109/CVPR52729.2023.00721 ↗
35. Lv, W., Xu, S., Zhao, Y., Wang, G., Wei, J., Cui, C., Du, Y., Dang, Q., and Liu, Y. (2023). DETRs Beat YOLOs on Real-time Object Detection. arXiv.
36. (2025, December 05). Ultralytics Homepage. Available online: https://docs.ultralytics.com/models/.