A Convolutional Neural Network Based on Soft Attention Mechanism and Multi-Scale Fusion for Skin Cancer Classification

被引:5
作者
Bao, Qiwei [1 ]
Han, Hua [1 ]
Huang, Li [1 ]
Muzahid, A. A. M. [1 ]
机构
[1] Shanghai Univ Engn Sci, Sch Elect & Elect Engn, Shanghai 201620, Peoples R China
基金
上海市自然科学基金; 国家重点研发计划;
关键词
Deep learning; CNNs; soft attention mechanism; multi-scale fusion; MODEL;
D O I
10.1142/S0218001423560244
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The seven most common skin diseases are melanocytic nevus, melanoma, benign keratosis, basal cell carcinoma, actinic keratosis, vascular lesions, and dermatofibroma. Among them, melanoma has been identified as one of the deadliest cancers based on medical studies and research. The current trend in disease detection revolves around the use of machine learning and deep learning models. Regardless of the model used, the crucial aspect is achieving accurate classification for these diseases. With the emergence of powerful convolutional neural networks (CNNs), significant progress has been made in classification of skin cancer lesions in recent years. However, various challenges hinder the development of practical and effective solutions. First, due to the specific nature of skin cancer lesion images, the current deep neural network architectures and training strategies have poor adaptability to medical images. They are also prone to gradient vanishing issues during network iteration, which hinders the construction of high-performance deep learning models that leverage distinctive characteristics of skin lesion images. Second, there exists a discordance between skin lesion images and deep learning network structures. To address these issues, this study introduces a soft attention mechanism to enhance adaptability to skin cancer lesion images and improve the extraction of informative features from medical images. Additionally, a novel multi-scale fusion convolutional neural network model is proposed to overcome the mismatch between deep learning CNN architectures and skin lesion images. This model autonomously extracts appearance features from raw dermatological medical images. Comparisons with other popular techniques demonstrate the effectiveness of the proposed model, which can achieve an accuracy of 93.9% on HAM10000 dataset. There is ongoing research to overcome the remaining challenges and further enhance the performance of skin cancer classification algorithms.
引用
收藏
页数:23
相关论文
共 61 条
[31]   A survey on deep learning in medical image analysis [J].
Litjens, Geert ;
Kooi, Thijs ;
Bejnordi, Babak Ehteshami ;
Setio, Arnaud Arindra Adiyoso ;
Ciompi, Francesco ;
Ghafoorian, Mohsen ;
van der Laak, Jeroen A. W. M. ;
van Ginneken, Bram ;
Sanchez, Clara I. .
MEDICAL IMAGE ANALYSIS, 2017, 42 :60-88
[32]   SKIN LESION CLASSIFICATION FROM DERMOSCOPIC IMAGES USING DEEP LEARNING TECHNIQUES [J].
Lopez, Adria Romero ;
Giro-i-Nieto, Xavier ;
Burdick, Jack ;
Marques, Oge .
2017 13TH IASTED INTERNATIONAL CONFERENCE ON BIOMEDICAL ENGINEERING (BIOMED), 2017, :49-54
[33]   A New Date-Balanced Method Based on Adaptive Asymmetric and Diversity Regularization in Person Re-Identification [J].
Ma, Wenjin ;
Han, Hua ;
Kong, Yong ;
Zhang, Yujin .
INTERNATIONAL JOURNAL OF PATTERN RECOGNITION AND ARTIFICIAL INTELLIGENCE, 2020, 34 (09)
[34]   Transfer learning using a multi-scale and multi-network ensemble for skin lesion classification [J].
Mahbod, Amirreza ;
Schaefer, Gerald ;
Wang, Chunliang ;
Dorffner, Georg ;
Ecker, Rupert ;
Ellinger, Isabella .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 193
[35]  
Menegola A, 2017, I S BIOMED IMAGING, P297, DOI 10.1109/ISBI.2017.7950523
[36]   Diagnosis of skin cancer using machine learning techniques [J].
Murugan, A. ;
Nair, S. Anu H. ;
Preethi, A. Angelin Peace ;
Kumar, K. P. Sanal .
MICROPROCESSORS AND MICROSYSTEMS, 2021, 81
[37]  
Le DNT, 2020, Arxiv, DOI arXiv:2009.05977
[38]  
Namozov A, 2018, I C INF COMM TECH CO, P417, DOI 10.1109/ICTC.2018.8539451
[39]   Novel Ensembling Methods for Dermatological Image Classification [J].
Nyiri, Tamas ;
Kiss, Attila .
THEORY AND PRACTICE OF NATURAL COMPUTING (TPNC 2018), 2018, 11324 :438-448
[40]   Classification of Skin Lesions Using Weighted Majority Voting Ensemble Deep Learning [J].
Okuboyejo, Damilola A. A. ;
Olugbara, Oludayo O. O. .
ALGORITHMS, 2022, 15 (12)