Accurate melanoma detection is essential for early-stage dermatological diagnosis, as it significantly improves treatment outcomes and patient survival rates. This study proposes a robust stacked ensemble framework that combines transfer learning (TL) with particle swarm optimization (PSO) to enhance melanoma classification performance. Initially, four deep transfer learning models: VGG16, ResNet50, DenseNet121, and InceptionV3 are trained on the Melanoma Skin Cancer Dataset consisting of 10,605 dermoscopic images to generate probability scores for melanoma and benign classes. These probability outputs are used to construct feature vectors for the subsequent stacked ensemble stage. The generated feature vectors are then provided as inputs to multiple meta-classifiers, whose hyperparameters are optimized using the PSO algorithm. PSO explores the hyperparameter search space iteratively, improving candidate solutions based on validation accuracy while utilizing both personal best and global best evaluations. Through this process, a total of eleven ensemble transfer learning (ETL) models are developed. Experimental results demonstrate that ETL11, which integrates VGG16, ResNet50, DenseNet121, and InceptionV3 with Gradient Boosting (GB), achieves the highest accuracy of 92.1
更多
查看译文
关键词
Melanoma detection,Deep transfer learning,Particle Swarm Optimization (PSO),Stacked ensemble learning (ETL),Explainable AI (XAI),SHAP