Skin lesion classification with ensembles of deep convolutional neural networks

被引:237
作者
Harangi, Balazs [1 ]
机构
[1] Univ Debrecen, Fac Informat, POB 400, H-4002 Debrecen, Hungary
关键词
Deep convolutional neural network; Ensemble-based system; Melanoma detection; Information fusion; MALIGNANT-MELANOMA; DERMOSCOPY IMAGES; ABCD; DIAGNOSIS; AREAS;
D O I
10.1016/j.jbi.2018.08.006
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Skin cancer is a major public health problem with over 123,000 newly diagnosed cases worldwide in every year. Melanoma is the deadliest form of skin cancer, responsible for over 9000 deaths in the United States each year. Thus, reliable automatic melanoma screening systems would provide a great help for clinicians to detect the malignant skin lesions as early as possible. In the last five years, the efficiency of deep learning-based methods increased dramatically and their performances seem to outperform conventional image processing methods in classification tasks. However, this type of machine learning-based approaches have a main drawback, namely they require thousands of labeled images per classes for their training. In this paper, we investigate how we can create an ensemble of deep convolutional neural networks to improve further their individual accuracies in the task of classifying dermoscopy images into the three classes melanoma, nevus, and seborrheic keratosis when we have no opportunity to train them on adequate number of annotated images. To achieve high classification accuracy, we fuse the outputs of the classification layers of four different deep neural network architectures. More specifically, we propose the aggregation of robust convolutional neural networks (CNNs) into one framework, where the final classification is achieved based on the weighted output of the member CNNs. For aggregation, we consider different fusion-based methods and select the best performing one for this problem. Our experimental results also prove that the creation of an ensemble of different neural networks is a meaningful approach, since each of the applied fusion strategies outperforms the individual networks regarding classification accuracy. The average area under the receiver operating characteristic curve has been found to be 0.891 for the 3-class classification task. For an objective evaluation of our approach, we have tested its performance on the official test database of the IEEE International Symposium on Biomedical Imaging (ISBI) 2017 challenge on Skin Lesion Analysis Towards Melanoma Detection dedicated to skin cancer recognition.
引用
收藏
页码:25 / 32
页数:8
相关论文
共 32 条
[1]   Melanoma recognition framework based on expert definition of ABCD for dermoscopic images [J].
Abbas, Qaisar ;
Celebi, M. Emre ;
Fondon Garcia, Irene ;
Ahmad, Waqar .
SKIN RESEARCH AND TECHNOLOGY, 2013, 19 (01) :E93-E102
[2]  
[Anonymous], LNCS
[3]  
[Anonymous], ARAGUAIA MED VISION
[4]  
[Anonymous], PROC CVPR IEEE
[5]  
[Anonymous], SKIN LESION DETECTIO
[6]  
[Anonymous], 2006, SEER CANC STAT REV 1
[7]  
[Anonymous], 2015, P 23 ACM INT C MULT
[8]   Border detection in dermoscopy images using statistical region merging [J].
Celebi, M. Emre ;
Kingravi, Hassan A. ;
Iyatomi, Hitoshi ;
Aslandogan, Y. Alp ;
Stoecker, William V. ;
Moss, Randy H. ;
Malters, Joseph M. ;
Grichnik, James M. ;
Marghoob, Ashfaq A. ;
Rabinovitz, Harold S. ;
Menzies, Scott W. .
SKIN RESEARCH AND TECHNOLOGY, 2008, 14 (03) :347-353
[9]   A methodological approach to the classification of dermoscopy images [J].
Celebi, M. Emre ;
Kingravi, Hassan A. ;
Uddin, Bakhtiyar ;
Lyatornid, Hitoshi ;
Aslandogan, Y. Alp ;
Stoecker, William V. ;
Moss, Randy H. .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2007, 31 (06) :362-373
[10]  
Codella N.C.F., 2017, SKIN LES AN MEL DET