Uncertainty Estimation in Deep Neural Networks for Dermoscopic Image Classification
- Resource Type
- Conference
- Authors
- Combalia, Marc; Hueto, Ferran; Puig, Susana; Malvehy, Josep; Vilaplana, Veronica
- Source
- 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW) Computer Vision and Pattern Recognition Workshops (CVPRW),2020 IEEE/CVF Conference on. :3211-3220 Jun, 2020
- Subject
- Computing and Processing
Uncertainty
Lesions
Skin
Neural networks
Monte Carlo methods
Training
Task analysis
- Language
- ISSN
- 2160-7516
The high performance of machine learning algorithms for the task of skin lesion classification has been shown over the past few years. However, real-world implementations are still scarce. One of the reasons could be that most methods do not quantify the uncertainty in the predictions and are not able to detect data that is anomalous or significantly different from that used in training, which may lead to a lack of confidence in the automated diagnosis or errors in the interpretation of results. In this work, we explore the use of uncertainty estimation techniques and metrics for deep neural networks based on Monte-Carlo sampling and apply them to the problem of skin lesion classification on data from ISIC Challenges 2018 and 2019. Our results show that uncertainty metrics can be successfully used to detect difficult and out-of-distribution samples.