Two-Stage Deep Neural Network via Ensemble Learning for Melanoma Classification

被引:8
作者
Ding, Jiaqi [1 ]
Song, Jie [1 ]
Li, Jiawei [1 ]
Tang, Jijun [2 ]
Guo, Fei [3 ]
机构
[1] Tianjin Univ, Sch Comp Sci & Technol, Coll Intelligence & Comp, Tianjin, Peoples R China
[2] Chinese Acad Sci, Shenzhen Inst Adv Technol, Shenzhen, Peoples R China
[3] Cent South Univ, Sch Comp Sci & Engn, Changsha, Peoples R China
来源
FRONTIERS IN BIOENGINEERING AND BIOTECHNOLOGY | 2022年 / 9卷
基金
国家重点研发计划; 中国国家自然科学基金;
关键词
melanoma classification; ensemble learning; deep convolutional neural network; image segmentation; dermoscopy images; ABCD RULE; SEGMENTATION; IMAGES;
D O I
10.3389/fbioe.2021.758495
中图分类号
Q81 [生物工程学(生物技术)]; Q93 [微生物学];
学科分类号
071005 ; 0836 ; 090102 ; 100705 ;
摘要
Melanoma is a skin disease with a high fatality rate. Early diagnosis of melanoma can effectively increase the survival rate of patients. There are three types of dermoscopy images, malignant melanoma, benign nevis, and seborrheic keratosis, so using dermoscopy images to classify melanoma is an indispensable task in diagnosis. However, early melanoma classification works can only use the low-level information of images, so the melanoma cannot be classified efficiently; the recent deep learning methods mainly depend on a single network, although it can extract high-level features, the poor scale and type of the features limited the results of the classification. Therefore, we need an automatic classification method for melanoma, which can make full use of the rich and deep feature information of images for classification. In this study, we propose an ensemble method that can integrate different types of classification networks for melanoma classification. Specifically, we first use U-net to segment the lesion area of images to generate a lesion mask, thus resize images to focus on the lesion; then, we use five excellent classification models to classify dermoscopy images, and adding squeeze-excitation block (SE block) to models to emphasize the more informative features; finally, we use our proposed new ensemble network to integrate five different classification results. The experimental results prove the validity of our results. We test our method on the ISIC 2017 challenge dataset and obtain excellent results on multiple metrics; especially, we get 0.909 on accuracy. Our classification framework can provide an efficient and accurate way for melanoma classification using dermoscopy images, laying the foundation for early diagnosis and later treatment of melanoma.
引用
收藏
页数:12
相关论文
共 51 条
  • [11] Xception: Deep Learning with Depthwise Separable Convolutions
    Chollet, Francois
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 1800 - 1807
  • [12] Codella Noel, 2015, Machine Learning in Medical Imaging. 6th International Workshop, MLMI 2015, held in conjunction with MICCAI 2015. Proceedings: LNCS 9352, P118, DOI 10.1007/978-3-319-24888-2_15
  • [13] Codella NCF, 2018, I S BIOMED IMAGING, P168, DOI 10.1109/ISBI.2018.8363547
  • [14] Instance-aware Semantic Segmentation via Multi-task Network Cascades
    Dai, Jifeng
    He, Kaiming
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 3150 - 3158
  • [15] Datta S.K., 2021, SOFT ATTENTION IMPRO
  • [16] Diaz I. G., 2017, INT SKIN IM COLL ISI
  • [17] A Deep Learning Approach for Colonoscopy Pathology WSI Analysis: Accurate Segmentation and Classification
    Feng, Ruiwei
    Liu, Xuechen
    Chen, Jintai
    Chen, Danny Z.
    Gao, Honghao
    Wu, Jian
    [J]. IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2021, 25 (10) : 3700 - 3708
  • [18] Automated melanoma recognition
    Ganster, H
    Pinz, A
    Röhrer, R
    Wildling, E
    Binder, M
    Kittler, H
    [J]. IEEE TRANSACTIONS ON MEDICAL IMAGING, 2001, 20 (03) : 233 - 239
  • [19] Skin Lesion Classification Using CNNs With Patch-Based Attention and Diagnosis-Guided Loss Weighting
    Gessert, Nils
    Sentker, Thilo
    Madesta, Frederic
    Schmitz, Ruediger
    Kniep, Helge
    Baltruschat, Ivo
    Werner, Rene
    Schlaefer, Alexander
    [J]. IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2020, 67 (02) : 495 - 503
  • [20] DermaKNet: Incorporating the Knowledge of Dermatologists to Convolutional Neural Networks for Skin Lesion Diagnosis
    Gonzalez-Diaz, Ivan
    [J]. IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2019, 23 (02) : 547 - 559