Skin Lesion Classification Using Collective Intelligence of Multiple Neural Networks

被引:47
作者
Popescu, Dan [1 ]
El-khatib, Mohamed [1 ]
Ichim, Loretta [1 ]
机构
[1] Univ Politehn Bucuresti, Fac Automat Control & Comp, Bucharest 060042, Romania
关键词
skin lesions classification; convolutional neural networks; data augmentation; residual blocks; dense blocks; inception module; multi-networks system; data fusion; decision weight; collective intelligence; CANCER;
D O I
10.3390/s22124399
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Skin lesion detection and analysis are very important because skin cancer must be found in its early stages and treated immediately. Once installed in the body, skin cancer can easily spread to other body parts. Early detection would represent a very important aspect since, by ensuring correct treatment, it could be curable. Thus, by taking all these issues into consideration, there is a need for highly accurate computer-aided systems to assist medical staff in the early detection of malignant skin lesions. In this paper, we propose a skin lesion classification system based on deep learning techniques and collective intelligence, which involves multiple convolutional neural networks, trained on the HAM10000 dataset, which is able to predict seven skin lesions including melanoma. The convolutional neural networks experimentally chosen, considering their performances, to implement the collective intelligence-based system for this purpose are: AlexNet, GoogLeNet, GoogLeNet-Places365, MobileNet-V2, Xception, ResNet-50, ResNet-101, InceptionResNet-V2 and DenseNet201. We then analyzed the performances of each of the above-mentioned convolutional neural networks to obtain a weight matrix whose elements are weights associated with neural networks and classes of lesions. Based on this matrix, a new decision matrix was used to build the multi-network ensemble system (Collective Intelligence-based System), combining each of individual neural network decision into a decision fusion module (Collective Decision Block). This module would then have the responsibility to take a final and more accurate decision related to the prediction based on the associated weights of each network output. The validation accuracy of the proposed system is about 3 percent better than that of the best performing individual network.
引用
收藏
页数:22
相关论文
共 27 条
[11]   Dermatologist-level classification of skin cancer with deep neural networks [J].
Esteva, Andre ;
Kuprel, Brett ;
Novoa, Roberto A. ;
Ko, Justin ;
Swetter, Susan M. ;
Blau, Helen M. ;
Thrun, Sebastian .
NATURE, 2017, 542 (7639) :115-+
[12]   Classification for Dermoscopy Images Using Convolutional Neural Networks Based on the Ensemble of Individual Advantage and Group Decision [J].
Gong, An ;
Yao, Xinjie ;
Lin, Wei .
IEEE ACCESS, 2020, 8 :155337-155351
[14]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[15]  
Howard A. G., 2017, arXiv
[16]   A New Architecture of Densely Connected Convolutional Networks for Pan-Sharpening [J].
Huang, Wei ;
Feng, Jingjing ;
Wang, Hua ;
Sun, Le .
ISPRS INTERNATIONAL JOURNAL OF GEO-INFORMATION, 2020, 9 (04)
[17]   Melanoma Detection Using an Objective System Based on Multiple Connected Neural Networks [J].
Ichim, Loretta ;
Popescu, Dan .
IEEE ACCESS, 2020, 8 :179189-179202
[18]   Deep Facial Diagnosis: Deep Transfer Learning From Face Recognition to Facial Diagnosis [J].
Jin, Bo ;
Cruz, Leandro ;
Goncalves, Nuno .
IEEE ACCESS, 2020, 8 :123649-123661
[19]   ImageNet Classification with Deep Convolutional Neural Networks [J].
Krizhevsky, Alex ;
Sutskever, Ilya ;
Hinton, Geoffrey E. .
COMMUNICATIONS OF THE ACM, 2017, 60 (06) :84-90
[20]  
Pomponiu V, 2016, IEEE IMAGE PROC, P2623, DOI 10.1109/ICIP.2016.7532834