Assisted deep learning framework for multi-class skin lesion classification considering a binary classification support

被引:44
作者
Harangi, Balazs [1 ]
Baran, Agnes [1 ]
Hajdu, Andras [1 ]
机构
[1] Univ Debrecen, Fac Informat, POB 400, H-4002 Debrecen, Hungary
关键词
Assisted learning; Deep learning; Ensemble learning; Skin lesion;
D O I
10.1016/j.bspc.2020.102041
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
In this paper, we propose a deep convolutional neural network framework to classify dermoscopy images into seven classes. With taking the advantage that these classes can be merged into two (healthy/diseased) ones we can train a part of the network regarding this binary task only. Then, the confidences regarding the binary classification are used to tune the multi-class confidence values provided by the other part of the network, since the binary task can be solved more accurately. For both the classification tasks we used GoogLeNet Inception-v3, however, any CNN architectures could be applied for these purposes. The whole network is trained in the usual way, and as our experimental results on the skin lesion image classification show, the accuracy of the multi-class problem has been remarkably raised (by 7% considering the balanced multi-class accuracy) via embedding the more reliable binary classification outcomes. (C) 2020 The Authors. Published by Elsevier Ltd.
引用
收藏
页数:7
相关论文
共 15 条
[1]   A Survey of Feature Extraction in Dermoscopy Image Analysis of Skin Cancer [J].
Barata, Catarina ;
Celebi, M. Emre ;
Marques, Jorge S. .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2019, 23 (03) :1096-1109
[2]  
Celebi M.E., 2018, DERMOSCOPY IMAGE ANA
[3]  
Codella N.C.F., 2017, ARXIV171005006 CORR
[4]   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-+
[6]   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
[7]   Emotion recognition by assisted learning with convolutional neural networks [J].
He, Xuanyu ;
Zhang, Wei .
NEUROCOMPUTING, 2018, 291 :187-194
[8]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269
[9]  
Quinlan J. R., 1986, Machine Learning, V1, P81, DOI 10.1007/BF00116251
[10]   ImageNet Large Scale Visual Recognition Challenge [J].
Russakovsky, Olga ;
Deng, Jia ;
Su, Hao ;
Krause, Jonathan ;
Satheesh, Sanjeev ;
Ma, Sean ;
Huang, Zhiheng ;
Karpathy, Andrej ;
Khosla, Aditya ;
Bernstein, Michael ;
Berg, Alexander C. ;
Fei-Fei, Li .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2015, 115 (03) :211-252