AI-Powered Skin Lesion Diagnosis using Whale Optimization Algorithm Enhanced ResNet 50 for Cancer Prediction

基于鲸鱼优化算法的AI皮肤病变诊断;增强型ResNet 50用于癌症预测

阅读:1

Abstract

OBJECTIVE: The primary objective of this study is to enhance the accuracy and efficiency of binary skin lesion classification by optimizing the ResNet-50 convolutional neural network using the Whale Optimization Algorithm (WOA). This involves fine-tuning key hyperparameters such as learning rate, weights, and biases to improve predictive performance. METHODS: This study compares five CNN architectures: AlexNet, GoogleNet, VGG16, Resnet 50, and WOA-optimized Resnet 50. The dataset comprises 3,600 balanced images (224×244 resolution) of skin moles, evenly divided into 1,800 benign and 1,800 malignant cases.  The models were trained on an open-access dermoscopic dataset to categorize skin lesions.  WOA was applied to optimize Resnet 50's hyperparameters weight and bias learning rate. Model performance was analysed using accuracy, preci-sion, recall, F1 score, specificity, Matthews Correlation Coefficient (MCC), log loss, AUC-ROC, and infer-ence time. The confusion matrix was analyzed to assess misclassification rates.  Result: The WOA-optimized Resnet 50 outperformed all other models, achieving 98.29% accuracy, higher than standard Resnet 50 (90.13%), GoogleNet (87.1%), AlexNet (86.53%), and VGG16 (81.18%). It also demonstrated superior recall (99.31%), specificity (97.07%), and an AUC-ROC of 99.84%, indicating excellent classification capability. The MCC score (0.9657) confirmed strong predictive reliability. Addi-tionally, the optimized model achieved the lowest log loss (0.0512), ensuring high confidence in predictions. With an inference time of 0.1488 seconds, it was significantly faster than standard Resnet 50 (1.029 seconds), making it computationally efficient. The confusion matrix confirmed its reliability, showing min-imal false positives (7) and false negatives (2). CONCLUSION: WOA-optimized Resnet 50 significantly improves accuracy, recall, specificity, and computational efficiency for binary skin lesion classification. Compared to traditional deep learning models, it offers superior predictive performance while maintaining fast inference time. These findings suggest that WOA-enhanced deep learning can enhance dermatological diagnostics, aiding early detection and clinical decision-making. Future research may explore its application for multi-class skin lesion classification and real-time medical imaging systems.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。