Melanoma Classification from Dermoscopy Images Using Ensemble of Convolutional Neural Networks

被引:25
作者
Raza, Rehan [1 ]
Zulfiqar, Fatima [1 ]
Tariq, Shehroz [1 ]
Anwar, Gull Bano [1 ]
Sargano, Allah Bux [1 ]
Habib, Zulfiqar [1 ]
机构
[1] COMSATS Univ Islamabad, Dept Comp Sci, Lahore 54000, Pakistan
基金
欧盟地平线“2020”;
关键词
deep learning; transfer learning; skin cancer; acral lentiginous melanoma; melanoma classification; ensemble learning; data augmentation; SKIN-CANCER;
D O I
10.3390/math10010026
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
Human skin is the most exposed part of the human body that needs constant protection and care from heat, light, dust, and direct exposure to other harmful radiation, such as UV rays. Skin cancer is one of the dangerous diseases found in humans. Melanoma is a form of skin cancer that begins in the cells (melanocytes) that control the pigment in human skin. Early detection and diagnosis of skin cancer, such as melanoma, is necessary to reduce the death rate due to skin cancer. In this paper, the classification of acral lentiginous melanoma, a type of melanoma with benign nevi, is being carried out. The proposed stacked ensemble method for melanoma classification uses different pre-trained models, such as Xception, Inceptionv3, InceptionResNet-V2, DenseNet121, and DenseNet201, by employing the concept of transfer learning and fine-tuning. The selection of pre-trained CNN architectures for transfer learning is based on models having the highest top-1 and top-5 accuracies on ImageNet. A novel stacked ensemble-based framework is presented to improve the generalizability and increase robustness by fusing fine-tuned pre-trained CNN models for acral lentiginous melanoma classification. The performance of the proposed method is evaluated by experimenting on a Figshare benchmark dataset. The impact of applying different augmentation techniques has also been analyzed through extensive experimentations. The results confirm that the proposed method outperforms state-of-the-art techniques and achieves an accuracy of 97.93%.
引用
收藏
页数:15
相关论文
共 53 条
[1]   Deep Convolutional Neural Network (DCNN) for Skin Cancer Classification [J].
Aburaed, Nour ;
Panthakkan, Alavikunhu ;
Al-Saad, Mina ;
Amin, Saad Ali ;
Mansoor, Wathiq .
2020 27TH IEEE INTERNATIONAL CONFERENCE ON ELECTRONICS, CIRCUITS AND SYSTEMS (ICECS), 2020,
[2]   Deep Learning-Based System for Automatic Melanoma Detection [J].
Adegun, Adekanmi A. ;
Viriri, Serestina .
IEEE ACCESS, 2020, 8 :7160-7172
[3]   Diagnosis of Malignant Melanoma of Skin Cancer Types [J].
Alasadi, Abbas H. Hassin ;
Alsafy, Baidaa M. .
INTERNATIONAL JOURNAL OF INTERACTIVE MULTIMEDIA AND ARTIFICIAL INTELLIGENCE, 2017, 4 (05) :44-49
[4]  
[Anonymous], 2014, COMPUT RES REPOSITOR
[5]  
Ashraf R., 2020, PROC IEEE 23 INT MUL, P1
[6]  
Barata C, 2019, I S BIOMED IMAGING, P841, DOI [10.1109/ISBI.2019.8759561, 10.1109/isbi.2019.8759561]
[7]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[8]   Deep learning ensembles for melanoma recognition in dermoscopy images [J].
Codella, N. C. F. ;
Nguyen, Q. -B. ;
Pankanti, S. ;
Gutman, D. A. ;
Helba, B. ;
Halpern, A. C. ;
Smith, J. R. .
IBM JOURNAL OF RESEARCH AND DEVELOPMENT, 2017, 61 (4-5)
[9]  
Combalia M., 2019, BCN20000: Dermoscopic Lesions in the Wild
[10]  
DAGHRIR J, 2020, 2020 5 INT C ADV TEC, P1, DOI DOI 10.1109/ATSIP49331.2020.9231544