用AI分析皮疹图诊断猴痘,准确率达97.5%。
An Explainable Nature-Inspired Framework for Monkeypox Diagnosis: Xception Features Combined with NGBoost and African Vultures Optimization Algorithm
- 用Xception提取特征,PCA降维,NGBoost分类
- 结合非洲秃鹫优化算法调参,准确率97.53%、F1值97.72%
- 支持可视化解释,适合医疗辅助诊断场景
猴痘在全球范围内的传播,尤其在非流行地区,引发了重大公共卫生关注。早期精准诊断对疾病管理和控制至关重要。为此,本研究提出一种基于深度学习的自动化猴痘检测框架,利用迁移学习、降维与先进机器学习技术,从皮肤病变图像中识别猴痘。采用新构建的猴痘皮肤病变数据集(MSLD),包含猴痘、水痘和麻疹图像,用于模型训练与评估。框架首先使用Xception进行深层特征提取,再通过主成分分析(PCA)降维,最后由自然梯度提升(NGBoost)完成分类。为优化模型性能与泛化能力,引入非洲秃鹫优化算法(AVOA)进行超参数调优,高效探索参数空间。结果表明,所提出的AVOA-NGBoost模型达到领先水平:准确率97.53%,F1-score 97.72%,AUC 97.47%。同时,借助Grad-CAM与LIME技术增强模型可解释性,揭示决策依据并突出关键判别特征。该框架为医疗提供高精度、高效的诊断工具,尤其适用于资源有限环境中的早期筛查。
原文摘要 · Abstract (English)
The recent global spread of monkeypox, particularly in regions where it has not historically been prevalent, has raised significant public health concerns. Early and accurate diagnosis is critical for effective disease management and control. In response, this study proposes a novel deep learning-based framework for the automated detection of monkeypox from skin lesion images, leveraging the power of transfer learning, dimensionality reduction, and advanced machine learning techniques. We utilize the newly developed Monkeypox Skin Lesion Dataset (MSLD), which includes images of monkeypox, chickenpox, and measles, to train and evaluate our models. The proposed framework employs the Xception architecture for deep feature extraction, followed by Principal Component Analysis (PCA) for dimensionality reduction, and the Natural Gradient Boosting (NGBoost) algorithm for classification. To optimize the model's performance and generalization, we introduce the African Vultures Optimization Algorithm (AVOA) for hyperparameter tuning, ensuring efficient exploration of the parameter space. Our results demonstrate that the proposed AVOA-NGBoost model achieves state-of-the-art performance, with an accuracy of 97.53%, F1-score of 97.72% and an AUC of 97.47%. Additionally, we enhance model interpretability using Grad-CAM and LIME techniques, providing insights into the decision-making process and highlighting key features influencing classification. This framework offers a highly precise and efficient diagnostic tool, potentially aiding healthcare providers in early detection and diagnosis, particularly in resource-constrained environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。