Dermoscopic Image Classification Method Using an Ensemble of Fine-Tuned Convolutional Neural Networks

Sensors

Lisheng Wei¹  Xin Shen²  Shaoyu Tang² 

[1] Anhui Key Laboratory of Electric Drive and Control, Anhui Polytechnic University, Wuhu 241002, China;School of Electrical Engineering, Anhui Polytechnic University, Wuhu 241000, China;

关键词: dermoscopic image;    classification;    deep learning;    transfer learning;    fine-tuning;    ensemble learning;   

DOI  :  10.3390/s22114147

来源: DOAJ

【 摘 要 】

Aiming at the problems of large intra-class differences, small inter-class differences, low contrast, and small and unbalanced datasets in dermoscopic images, this paper proposes a dermoscopic image classification method based on an ensemble of fine-tuned convolutional neural networks. By reconstructing the fully connected layers of the three pretrained models of Xception, ResNet50, and Vgg-16 and then performing transfer learning and fine-tuning the three pretrained models with the ISIC 2016 Challenge official skin dataset, we integrated the outputs of the three base models using a weighted fusion ensemble strategy in order to obtain a final prediction result able to distinguish whether a dermoscopic image indicates malignancy. The experimental results show that the accuracy of the ensemble model is 86.91%, the precision is 85.67%, the recall is 84.03%, and the F1-score is 84.84%, with these four evaluation metrics being better than those of the three basic models and better than some classical methods, proving the effectiveness and feasibility of the proposed method.

【 授权许可】

Unknown