Identifying Melanoma Images using EfficientNet Ensemble: Winning Solution to the SIIM-ISIC Melanoma Classification Challenge

Abstract

We present our winning solution to the SIIM-ISIC Melanoma Classification Challenge. It is an ensemble of convolutions neural network (CNN) models with different backbones and input sizes, most of which are image-only models while a few of them used image-level and patient-level metadata. The keys to our winning are: (1) stable validation scheme (2) good choice of model target (3) carefully tuned pipeline and (4) ensembling with very diverse models. The winning submission scored 0.9600 AUC on cross validation and 0.9490 AUC on private leaderboard.

arXiv comment6 pages, 2 figures

References in corpus (3)

• EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks
• Improved Regularization of Convolutional Neural Networks with Cutout
• SGDR: Stochastic Gradient Descent with Warm Restarts

Cited by in corpus (5)

• Dermatologist-like explainable AI enhances trust and confidence in diagnosing melanoma
• A survey, review, and future trends of skin lesion segmentation and classification
• Clinical Melanoma Diagnosis with Artificial Intelligence: Insights from a Prospective Multicenter Study
• Chest X-ray Image Phase Features for Improved Diagnosis of COVID-19 Using Convolutional Neural Network
• Skin Deep Unlearning: Artefact and Instrument Debiasing in the Context of Melanoma Classification