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 条
[1]  
A Ameri, 2020, J Biomed Phys Eng, V10, P801, DOI 10.31661/jbpe.v0i0.2004-1107
[2]  
Abdulrahman Asma Abdulelah, 2021, Journal of Physics: Conference Series, V1879, DOI 10.1088/1742-6596/1879/2/022118
[3]  
Akter M., 2022, 2022 IEEE INT C BIG, P5404, DOI [10.1109/BigData55660.2022.10020302, DOI 10.1109/BIGDATA55660.2022.10020302]
[4]   An Efficient Deep Learning-Based Skin Cancer Classifier for an Imbalanced Dataset [J].
Alam, Talha Mahboob ;
Shaukat, Kamran ;
Khan, Waseem Ahmad ;
Hameed, Ibrahim A. ;
Abd Almuqren, Latifah ;
Raza, Muhammad Ahsan ;
Aslam, Memoona ;
Luo, Suhuai .
DIAGNOSTICS, 2022, 12 (09)
[5]   An enhanced technique of skin cancer classification using deep convolutional neural network with transfer learning models [J].
Ali, Md Shahin ;
Miah, Md Sipon ;
Haque, Jahurul ;
Rahman, Md Mahbubur ;
Islam, Md Khairul .
MACHINE LEARNING WITH APPLICATIONS, 2021, 5
[6]   Multi-class Skin Disease Classification Using Transfer Learning Model [J].
Anand, Vatsala ;
Gupta, Sheifali ;
Koundal, Deepika ;
Nayak, Soumya Ranjan ;
Nayak, Janmenjoy ;
Vimal, S. .
INTERNATIONAL JOURNAL ON ARTIFICIAL INTELLIGENCE TOOLS, 2022, 31 (02)
[7]   Machine Learning and Deep Learning Algorithms for Skin Cancer Classification from Dermoscopic Images [J].
Bechelli, Solene ;
Delhommelle, Jerome .
BIOENGINEERING-BASEL, 2022, 9 (03)
[8]  
Codella N, 2019, Arxiv, DOI arXiv:1902.03368
[9]   Deep-learning-based, computer-aided classifier developed with a small dataset of clinical images surpasses board-certified dermatologists in skin tumour diagnosis [J].
Fujisawa, Y. ;
Otomo, Y. ;
Ogata, Y. ;
Nakamura, Y. ;
Fujita, R. ;
Ishitsuka, Y. ;
Watanabe, R. ;
Okiyama, N. ;
Ohara, K. ;
Fujimoto, M. .
BRITISH JOURNAL OF DERMATOLOGY, 2019, 180 (02) :373-381
[10]   Deep convolutional neural network VGG-16 model for differential diagnosing of papillary thyroid carcinomas in cytological images: a pilot study [J].
Guari, Qing ;
Wang, Yunjun ;
Ping, Bo ;
Li, Duanshu ;
Du, Jiajun ;
Qin, Yu ;
Lu, Hongtao ;
Wan, Xiaochun ;
Xiang, Jun .
JOURNAL OF CANCER, 2019, 10 (20) :4876-4882