» Articles » PMID: 37835902

DermAI 1.0: A Robust, Generalized, and Novel Attention-Enabled Ensemble-Based Transfer Learning Paradigm for Multiclass Classification of Skin Lesion Images

Abstract

Skin lesion classification plays a crucial role in dermatology, aiding in the early detection, diagnosis, and management of life-threatening malignant lesions. However, standalone transfer learning (TL) models failed to deliver optimal performance. In this study, we present an attention-enabled ensemble-based deep learning technique, a powerful, novel, and generalized method for extracting features for the classification of skin lesions. This technique holds significant promise in enhancing diagnostic accuracy by using seven pre-trained TL models for classification. Six ensemble-based DL (EBDL) models were created using stacking, softmax voting, and weighted average techniques. Furthermore, we investigated the attention mechanism as an effective paradigm and created seven attention-enabled transfer learning (aeTL) models before branching out to construct three attention-enabled ensemble-based DL (aeEBDL) models to create a reliable, adaptive, and generalized paradigm. The mean accuracy of the TL models is 95.30%, and the use of an ensemble-based paradigm increased it by 4.22%, to 99.52%. The aeTL models' performance was superior to the TL models in accuracy by 3.01%, and aeEBDL models outperformed aeTL models by 1.29%. Statistical tests show significant p-value and Kappa coefficient along with a 99.6% reliability index for the aeEBDL models. The approach is highly effective and generalized for the classification of skin lesions.

Citing Articles

Artificial intelligence in managing retinal disease-current concepts and relevant aspects for health care providers.

Riedl S, Birner K, Schmidt-Erfurth U Wien Med Wochenschr. 2025; .

PMID: 39992600 DOI: 10.1007/s10354-024-01069-1.


Development of an individualized dementia risk prediction model using deep learning survival analysis incorporating genetic and environmental factors.

Yuan S, Liu Q, Huang X, Tan S, Bai Z, Yu J Alzheimers Res Ther. 2024; 16(1):278.

PMID: 39736679 PMC: 11684124. DOI: 10.1186/s13195-024-01663-w.


Deep learning approach for cardiovascular disease risk stratification and survival analysis on a Canadian cohort.

Bhagawati M, Paul S, Mantella L, Johri A, Laird J, Singh I Int J Cardiovasc Imaging. 2024; 40(6):1283-1303.

PMID: 38678144 DOI: 10.1007/s10554-024-03100-3.


GeneAI 3.0: powerful, novel, generalized hybrid and ensemble deep learning frameworks for miRNA species classification of stationary patterns from nucleotides.

Singh J, Khanna N, Rout R, Singh N, Laird J, Singh I Sci Rep. 2024; 14(1):7154.

PMID: 38531923 PMC: 11344070. DOI: 10.1038/s41598-024-56786-9.

References
1.
Hosny K, Kassem M, Foaud M . Classification of skin lesions using transfer learning and augmentation with Alex-net. PLoS One. 2019; 14(5):e0217293. PMC: 6529006. DOI: 10.1371/journal.pone.0217293. View

2.
Rajendra Acharya U, Sree S, Krishnan M, Krishnananda N, Ranjan S, Umesh P . Automated classification of patients with coronary artery disease using grayscale features from left ventricle echocardiographic images. Comput Methods Programs Biomed. 2013; 112(3):624-32. DOI: 10.1016/j.cmpb.2013.07.012. View

3.
Shrivastava V, Londhe N, Sonawane R, Suri J . Computer-aided diagnosis of psoriasis skin images with HOS, texture and color features: A first comparative study of its kind. Comput Methods Programs Biomed. 2016; 126:98-109. DOI: 10.1016/j.cmpb.2015.11.013. View

4.
Narayanan D, Saladi R, Fox J . Ultraviolet radiation and skin cancer. Int J Dermatol. 2010; 49(9):978-86. DOI: 10.1111/j.1365-4632.2010.04474.x. View

5.
Suri J, Agarwal S, Pathak R, Ketireddy V, Columbu M, Saba L . COVLIAS 1.0: Lung Segmentation in COVID-19 Computed Tomography Scans Using Hybrid Deep Learning Artificial Intelligence Models. Diagnostics (Basel). 2021; 11(8). PMC: 8392426. DOI: 10.3390/diagnostics11081405. View