Multimodal Learning 相关度: 7/10

Toward Reliable and Explainable Nail Disease Classification: Leveraging Adversarial Training and Grad-CAM Visualization

Farzia Hossain, Samanta Ghosh, Shahida Begum, B. M. Shahria Alam, Mohammad Tahmid Noor, Md Parvez Mia, Nishat Tasnim Niloy
arXiv: 2602.04820v1 发布: 2026-02-04 更新: 2026-02-04

AI 摘要

本文提出了一种基于深度学习的指甲疾病分类方法,利用对抗训练和Grad-CAM可视化提高模型的可靠性和可解释性。

主要贡献

  • 利用InceptionV3等CNN模型进行指甲疾病分类
  • 应用对抗训练增强模型鲁棒性
  • 使用SHAP解释模型预测结果

方法论

训练InceptionV3, DenseNet201, EfficientNetV2, ResNet50等模型,并使用对抗训练和SHAP进行优化和解释。

原文摘要

Human nail diseases are gradually observed over all age groups, especially among older individuals, often going ignored until they become severe. Early detection and accurate diagnosis of such conditions are important because they sometimes reveal our body's health problems. But it is challenging due to the inferred visual differences between disease types. This paper presents a machine learning-based model for automated classification of nail diseases based on a publicly available dataset, which contains 3,835 images scaling six categories. In 224x224 pixels, all images were resized to ensure consistency. To evaluate performance, four well-known CNN models-InceptionV3, DenseNet201, EfficientNetV2, and ResNet50 were trained and analyzed. Among these, InceptionV3 outperformed the others with an accuracy of 95.57%, while DenseNet201 came next with 94.79%. To make the model stronger and less likely to make mistakes on tricky or noisy images, we used adversarial training. To help understand how the model makes decisions, we used SHAP to highlight important features in the predictions. This system could be a helpful support for doctors, making nail disease diagnosis more accurate and faster.

标签

图像分类 深度学习 对抗训练 可解释性

arXiv 分类

cs.CV cs.AI cs.LG