Daily Arxiv

This page organizes papers related to artificial intelligence published around the world.
This page is summarized using Google Gemini and is operated on a non-profit basis.
The copyright of the paper belongs to the author and the relevant institution. When sharing, simply cite the source.

A Quad-Step Approach to Uncertainty-Aware Deep Learning for Skin Cancer Classification

Created by
  • Haebom

Author

Hamzeh Asgharnezhad, Pegah Tabarisaadi, Abbas Khosravi, Roohallah Alizadehsani, U. Rajendra Acharya

Outline

This paper presents a comprehensive evaluation of deep learning (DL)-based skin lesion classification with transfer learning and uncertainty quantification (UQ) using the HAM10000 dataset. We benchmarked several pre-trained feature extractors, including CLIP variants, ResNet50, DenseNet121, VGG16, and EfficientNet-V2-Large, combined with traditional classifiers such as SVM, XGBoost, and logistic regression. We explored various PCA settings (64, 128, 256, and 512), and achieved the best baseline performance when LAION CLIP ViT-H/14 and ViT-L/14 were used with PCA-256. In the UQ step, we applied Monte Carlo Dropout (MCD), Ensemble, and Ensemble Monte Carlo Dropout (EMCD) and evaluated using uncertainty-aware metrics (UAcc, USen, USpe, and UPre). An ensemble method using PCA-256 provided the best balance between accuracy and reliability. Further improvements were achieved through feature fusion of the best-performing extractors. Finally, a feature fusion-based model trained with the predictive entropy (PE) loss function was proposed, achieving superior performance to all previous configurations in both standard and uncertainty-aware evaluations, thereby advancing reliable DL-based skin cancer diagnosis.

Takeaways, Limitations

Takeaways:
We demonstrate that a deep learning model utilizing transfer learning and uncertainty quantification contributes to improving the accuracy of skin cancer diagnosis.
Comparative analysis of various feature extractors, classifiers, and PCA settings to suggest the optimal model configuration.
Feature fusion models utilizing the predictive entropy loss function demonstrate superior performance compared to existing models.
We demonstrate that model reliability can be improved through uncertainty quantification techniques.
Limitations:
Due to limitations of the HAM10000 dataset, further validation of the model's generalization performance is required.
Lack of model performance evaluation in real clinical environments.
Further improvements in uncertainty quantification techniques are needed.
There is a possibility of overfitting to certain datasets.
👍