用小波变换与新型优化算法提升皮肤癌诊断准确率。
Enhancing Skin Cancer Diagnosis (SCD) Using Late Discrete Wavelet Transform (DWT) and New Swarm-Based Optimizers
- 结合小波变换与自注意力机制提取图像特征。
- 在ISIC数据集上达98.11%准确率,较旧方法提升至少1%。
- 适合医学影像分析、AI辅助诊断研究者参考。
皮肤癌是威胁生命的主要癌症之一,早期诊断至关重要。近年来深度学习在皮肤癌检测中展现出潜力,但模型效率与准确率仍有提升空间。本文提出一种新方法:先用DenseNet-121、Inception、Xception和MobileNet等预训练网络提取图像层次特征,再通过离散小波变换(DWT)层捕捉高低频成分,并引入自注意力模块学习特征间全局依赖关系。随后采用三种新型群智能优化算法——改进的银狼优化器(IGWO)、改进的狐优化算法(FOX)和改进的猩猩群优化器(MGTO)——优化网络权重与神经元数量。实验结果表明,该方法显著提升诊断性能,在ISIC-2016数据集上使用MobileNet+Wavelet+FOX组合达到98.11%准确率,在ISIC-2017数据集上使用Inception+Wavelet+MGTO组合达97.95%,相较其他方法至少提升1%。
原文摘要 · Abstract (English)
Skin cancer (SC) stands out as one of the most life-threatening forms of cancer, with its danger amplified if not diagnosed and treated promptly. Early intervention is critical, as it allows for more effective treatment approaches. In recent years, Deep Learning (DL) has emerged as a powerful tool in the early detection and skin cancer diagnosis (SCD). Although the DL seems promising for the diagnosis of skin cancer, still ample scope exists for improving model efficiency and accuracy. This paper proposes a novel approach to skin cancer detection, utilizing optimization techniques in conjunction with pre-trained networks and wavelet transformations. First, normalized images will undergo pre-trained networks such as Densenet-121, Inception, Xception, and MobileNet to extract hierarchical features from input images. After feature extraction, the feature maps are passed through a Discrete Wavelet Transform (DWT) layer to capture low and high-frequency components. Then the self-attention module is integrated to learn global dependencies between features and focus on the most relevant parts of the feature maps. The number of neurons and optimization of the weight vectors are performed using three new swarm-based optimization techniques, such as Modified Gorilla Troops Optimizer (MGTO), Improved Gray Wolf Optimization (IGWO), and Fox optimization algorithm. Evaluation results demonstrate that optimizing weight vectors using optimization algorithms can enhance diagnostic accuracy and make it a highly effective approach for SCD. The proposed method demonstrates substantial improvements in accuracy, achieving top rates of 98.11% with the MobileNet + Wavelet + FOX and DenseNet + Wavelet + Fox combination on the ISIC-2016 dataset and 97.95% with the Inception + Wavelet + MGTO combination on the ISIC-2017 dataset, which improves accuracy by at least 1% compared to other methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。