A Non-Invasive Interpretable Diagnosis of Melanoma Skin Cancer Using Deep Learning and Ensemble Stacking of Machine Learning Models

被引:38
作者
Alfi, Iftiaz A. [1 ]
Rahman, Md Mahfuzur [2 ,3 ]
Shorfuzzaman, Mohammad [4 ]
Nazir, Amril [5 ]
机构
[1] North South Univ, Dept Elect & Comp Engn, Dhaka 1229, Bangladesh
[2] King Fahd Univ Petr & Minerals, Coll Comp & Math, Dept Informat & Comp Sci, Dhahran 31261, Saudi Arabia
[3] King Fahd Univ Petr & Minerals, Interdisciplinary Res Ctr Intelligent Secure Syst, Dhahran 31261, Saudi Arabia
[4] Taif Univ, Coll Comp & Informat Technol, Dept Comp Sci, At Taif 21944, Saudi Arabia
[5] Zayed Univ, Coll Technol Innovat, Dept Informat Syst, Abu Dhabi Campus,POB 144534, Abu Dhabi, U Arab Emirates
关键词
skin cancer; diagnosis; machine learning; stacking model; deep learning; interpretability; melanoma; CLASSIFICATION;
D O I
10.3390/diagnostics12030726
中图分类号
R5 [内科学];
学科分类号
1002 ; 100201 ;
摘要
A skin lesion is a portion of skin that observes abnormal growth compared to other areas of the skin. The ISIC 2018 lesion dataset has seven classes. A miniature dataset version of it is also available with only two classes: malignant and benign. Malignant tumors are tumors that are cancerous, and benign tumors are non-cancerous. Malignant tumors have the ability to multiply and spread throughout the body at a much faster rate. The early detection of the cancerous skin lesion is crucial for the survival of the patient. Deep learning models and machine learning models play an essential role in the detection of skin lesions. Still, due to image occlusions and imbalanced datasets, the accuracies have been compromised so far. In this paper, we introduce an interpretable method for the non-invasive diagnosis of melanoma skin cancer using deep learning and ensemble stacking of machine learning models. The dataset used to train the classifier models contains balanced images of benign and malignant skin moles. Hand-crafted features are used to train the base models (logistic regression, SVM, random forest, KNN, and gradient boosting machine) of machine learning. The prediction of these base models was used to train level one model stacking using cross-validation on the training set. Deep learning models (MobileNet, Xception, ResNet50, ResNet50V2, and DenseNet121) were used for transfer learning, and were already pre-trained on ImageNet data. The classifier was evaluated for each model. The deep learning models were then ensembled with different combinations of models and assessed. Furthermore, shapely adaptive explanations are used to construct an interpretability approach that generates heatmaps to identify the parts of an image that are most suggestive of the illness. This allows dermatologists to understand the results of our model in a way that makes sense to them. For evaluation, we calculated the accuracy, Fl-score, Cohen's kappa, confusion matrix, and ROC curves and identified the best model for classifying skin lesions.
引用
收藏
页数:18
相关论文
共 29 条
[1]   Deep Neural Network for Fuzzy Automatic Melanoma Diagnosis [J].
Abbes, Wiem ;
Sellami, Dorra .
VISAPP: PROCEEDINGS OF THE 14TH INTERNATIONAL JOINT CONFERENCE ON COMPUTER VISION, IMAGING AND COMPUTER GRAPHICS THEORY AND APPLICATIONS, VOL 4, 2019, :47-56
[2]  
Abuzaghleh O., 2014, IEEE Long Isl. Syst. Appl. Technol. Conf. LISAT, P1, DOI [10.1109/LISAT.2014.6845199, DOI 10.1109/LISAT.2014.6845199]
[3]   Integrated design of deep features fusion for localization and classification of skin cancer [J].
Amin, Javeria ;
Sharif, Abida ;
Gul, Nadia ;
Anjum, Muhammad Almas ;
Nisar, Muhammad Wasif ;
Azam, Faisal ;
Bukhari, Syed Ahmad Chan .
PATTERN RECOGNITION LETTERS, 2020, 131 :63-70
[4]  
[Anonymous], 2018, ISIC 2018 CHALLENGE
[5]  
[Anonymous], 2013, Medicine
[6]   Computer-Aided Diagnosis of Skin Diseases Using Deep Neural Networks [J].
Bajwa, Muhammad Naseer ;
Muta, Kaoru ;
Malik, Muhammad Imran ;
Siddiqui, Shoaib Ahmed ;
Braun, Stephan Alexander ;
Homey, Bernhard ;
Dengel, Andreas ;
Ahmed, Sheraz .
APPLIED SCIENCES-BASEL, 2020, 10 (07)
[7]   Multi-Label classification of multi-modality skin lesion via hyper-connected convolutional neural network [J].
Bi, Lei ;
Feng, David Dagan ;
Fulham, Michael ;
Kim, Jinman .
PATTERN RECOGNITION, 2020, 107
[8]   Automatic Melanoma Detection via Multi-scale Lesion-biased Representation and Joint Reverse Classification [J].
Bi, Lei ;
Kim, Jinman ;
Ahn, Euijoon ;
Feng, Dagan ;
Fulham, Michael .
2016 IEEE 13TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2016, :1055-1058
[9]   Towards Automated Melanoma Detection with Deep Learning: Data Purification and Augmentation [J].
Bisla, Devansh ;
Choromanska, Anna ;
Berman, Russell S. ;
Stein, Jennifer A. ;
Polsky, David .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2019), 2019, :2720-2728
[10]  
Daghrir L., 2020, 2020 5 INT C ADV TEC, P1, DOI [DOI 10.1109/ATSIP49331.2020.9231544, 10.1109/ATSIP49331.2020.923154432.R.]