-
SkinCON: Towards Consensus for the Uncertainty of Skin Cancer Sub-typing through Distribution Regularized Adaptive Predictive Sets (DRAPS)
-
Zhihang Ren and Yunqi Li and Xinyu Li and Xinrong Xie and Erik P. Duhaime and Kathy Fang and Tapabrata Chakraborti and Yunhui Guo and Stella X. Yuand David Whitney
-
International Conference on Medical Image Computing and Computer Assisted Intervention, Marrakesh, Morocco, 6-10 October 2024
-
Paper
|
Poster
|
Code
-
Abstract
-
Deep learning has been widely utilized in medical diagnosis. Convolutional neural networks and transformers can achieve high predictive accuracy, which can be on par with or even exceed human performance. However, uncertainty quantification remains an unresolved issue, impeding the deployment of deep learning models in practical settings. Conformal analysis can, in principle, estimate the uncertainty of each diagnostic prediction, but doing so effectively requires extensive human annotations to characterize the underlying empirical distributions. This has been challenging in the past because instance-level class distribution data has been unavailable: Collecting massive ground truth labels is already challenging, and obtaining the class distribution of each instance is even more difficult. Here, we provide a large skin cancer instance-level class distribution dataset, SkinCON, that contains 25,331 skin cancer images from the ISIC 2019 challenge dataset. SkinCON is built upon over 937,167 diagnostic judgments from 10,509 participants. Using SkinCON, we propose the distribution regularized adaptive predictive sets (DRAPS) method for skin cancer diagnosis. We also provide a new evaluation metric based on SkinCON. Experiment results show the quality of our proposed DRAPS method and the uncertainty variation with respect to patient age and sex from health equity and fairness perspective.
-
Keywords
-
image-based medical diagnosis, skin cancer subtyping
|