DermAI 1.0: A Robust, Generalized, and Novel Attention-Enabled Ensemble-Based Transfer Learning Paradigm for Multiclass Classification of Skin Lesion Images
Overview
Authors
Affiliations
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.
Riedl S, Birner K, Schmidt-Erfurth U Wien Med Wochenschr. 2025; .
PMID: 39992600 DOI: 10.1007/s10354-024-01069-1.
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.
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.
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.