Skin lesion classification using ensembles of multi -resolution EfficientNets with meta data

被引:169
作者
Gessert, Nils [1 ,2 ]
Nielsen, Maximilian [2 ,3 ]
Shaikh, Mohsin [2 ,3 ]
Werner, Rene [2 ,3 ]
Schlaefer, Alexander [1 ,2 ]
机构
[1] Hamburg Univ Technol, Inst Med Technol, Hamburg, Germany
[2] Forschungszentrum Med Tech Hamburg, DAISYlab, Hamburg, Germany
[3] Univ Med Ctr Hamburg Eppendorf, Inst Computat Neurosci, Hamburg, Germany
关键词
Deep Learning; Multi-class skin lesion classification; Convolutional neural networks;
D O I
10.1016/j.mex.2020.100864
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
In this paper, we describe our method for the ISIC 2019 Skin Lesion Classification Challenge. The challenge comes with two tasks. For task 1, skin lesions have to be classified based on dermoscopic images. For task 2, dermoscopic images and additional patient meta data are used. Our deep learning-based method achieved first place for both tasks. The are several problems we address with our method. First, there is an unknown class in the test set which we cover with a data-driven approach. Second, there is a severe class imbalance that we address with loss balancing. Third, there are images with different resolutions which motivates two different cropping strategies and multi-crop evaluation. Last, there is patient meta data available which we incorporate with a dense neural network branch. We address skin lesion classification with an ensemble of deep learning models including EfficientNets, SENet, and ResNeXt WSL, selected by a search strategy. We rely on multiple model input resolutions and employ two cropping strategies for training. We counter severe class imbalance with a loss balancing approach. We predict an additional, unknown class with a data-driven approach and we make use of patient meta data with an additional input branch. (C) 2020 The Author(s). Published by Elsevier B.V.
引用
收藏
页数:8
相关论文
共 15 条
[1]  
[Anonymous], 2018, PROC CVPR IEEE, DOI [DOI 10.1109/CVPR.2018.00745, DOI 10.1109/TPAMI.2019.2913372]
[2]  
Codella NCF, 2018, I S BIOMED IMAGING, P168, DOI 10.1109/ISBI.2018.8363547
[3]  
Combalia M., 2019, BCN20000: Dermoscopic Lesions in the Wild
[4]   AutoAugment: Learning Augmentation Strategies from Data [J].
Cubuk, Ekin D. ;
Zoph, Barret ;
Mane, Dandelion ;
Vasudevan, Vijay ;
Le, Quoc V. .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :113-123
[5]  
DeVries Terrance, 2017, Improved regularization of convolutional neural networks with cutout
[6]  
Gessert N., 2018, Skin Lesion Diagnosis using Ensembles, Unscaled MultiCrop Evaluation and Loss Weighting
[7]   Left Ventricle Quantification Using Direct Regression with Segmentation Regularization and Ensembles of Pretrained 2D and 3D CNNs [J].
Gessert, Nils ;
Schlaefer, Alexander .
STATISTICAL ATLASES AND COMPUTATIONAL MODELS OF THE HEART: MULTI-SEQUENCE CMR SEGMENTATION, CRT-EPIGGY AND LV FULL QUANTIFICATION CHALLENGES, 2020, 12009 :375-383
[8]   Skin Lesion Classification Using CNNs With Patch-Based Attention and Diagnosis-Guided Loss Weighting [J].
Gessert, Nils ;
Sentker, Thilo ;
Madesta, Frederic ;
Schmitz, Ruediger ;
Kniep, Helge ;
Baltruschat, Ivo ;
Werner, Rene ;
Schlaefer, Alexander .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2020, 67 (02) :495-503
[9]   Seven-Point Checklist and Skin Lesion Classification Using Multitask Multimodal Neural Nets [J].
Kawahara, Jeremy ;
Daneshvar, Sara ;
Argenziano, Giuseppe ;
Hamarneh, Ghassan .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2019, 23 (02) :538-546
[10]   Focal Loss for Dense Object Detection [J].
Lin, Tsung-Yi ;
Goyal, Priya ;
Girshick, Ross ;
He, Kaiming ;
Dollar, Piotr .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :2999-3007