End-to-End Decoupled Training: A Robust Deep Learning Method for Long-Tailed Classification of Dermoscopic Images for Skin Lesion Classification

• DOI: 10.3390/electronics11203275
• Publication Date: 2022-10-12T09:31:18Z
• Authors: Arthur Cartel Foa...; Rabah Iguernaissi; Jean-Luc Damoiseaux; Abdellatif Moudafi; Djamal Merad
• Keywords: skin lesion detection; computer-aided diagnosis; long-tailed distribution; deep learning; 0202 electrical engineering, electronic engineering, information engineering; 02 engineering and technology; 3. Good health

Due to its increasing incidence, skin cancer, and especially melanoma, is considered a major public health issue. Manually detecting skin lesions (SL) from dermoscopy images is a difficult and time-consuming process. Thus, researchers designed computer-aided diagnosis (CAD) systems to assist dermatologists in the early detection of skin cancer. Moreover, SL detection naturally exhibits a long-tailed distribution due to the complex patient-level conditions and the existence of rare diseases. Very limited research for handling this issue exists on SL detection. In this paper, we propose an end-to-end decoupled training for the long-tailed skin lesion classification task. Specifically, we initialized the training of a network with a novel loss function Lf able to guide the model to a better representation of the features. Then, we fine-tuned the pretrained networks with a weighted variant of Lf helping to improve the robustness of the network to class imbalance. We evaluated our model on the ISIC 2018 public dataset against existing methods for handling class imbalance and existing approaches for SL detection. The results demonstrated the superiority of our framework, outperforming all compared methods by a minimum margin of 2% with a single model.