Exploring Instance-Level Uncertainty for Medical Detection
The ability of deep learning to predict with uncertainty is recognized as key for its adoption in clinical routines. Moreover, performance gain has been enabled by modelling uncertainty according to empirical evidence. While previous work has widely discussed the uncertainty estimation in segmentation and classification tasks, its application on bounding-box-based detection has been limited, mainly due to the challenge of bounding box aligning. In this work, we explore to augment a 2.5D detection CNN with two different bounding-box-level (or instance-level) uncertainty estimates, i.e., predictive variance and Monte Carlo (MC) sample variance. Experiments are conducted for lung nodule detection on LUNA16 dataset, a task where significant semantic ambiguities can exist between nodules and non-nodules. Results show that our method improves the evaluating score from 84.57% to 88.86% by utilizing a combination of both types of variances. Moreover, we show the generated uncertainty enables superior operating points compared to using the probability threshold only, and can further boost the performance to 89.52%. Example nodule detections are visualized to further illustrate the advantages of our method.
Code (0)
등록된 구현이 없습니다.
Tasks
Lung Nodule DetectionSimilar Papers 제목 키워드 기반
Exploring Structure-Wise Uncertainty for 3D Medical Image Segmentation
When applying a Deep Learning model to medical images, it is crucial to estimate the model uncertainty. Voxel-wise uncertainty is a useful visual marker for human experts and could be used to improve the model's voxel-wi…
Image SegmentationMedical Image SegmentationOut of Distribution (OOD) DetectionSegmentation+2Multiple instance active learning for object detection
Despite the substantial progress of active learning for image recognition, there still lacks an instance-level active learning method specified for object detection. In this paper, we propose Multiple Instance Active Obj…
Active LearningActive Object DetectionMultiple Instance LearningObject+2Exploring Aleatoric Uncertainty in Object Detection via Vision Foundation Models
Datasets collected from the open world unavoidably suffer from various forms of randomness or noiseness, leading to the ubiquity of aleatoric (data) uncertainty. Quantifying such uncertainty is particularly pivotal for o…
Objectobject-detectionObject DetectionExploring Uncertainty Measures in Deep Networks for Multiple Sclerosis Lesion Detection and Segmentation
Deep learning (DL) networks have recently been shown to outperform other segmentation methods on various public, medical-image challenge datasets [3,11,16], especially for large pathologies. However, in the context of di…
Lesion DetectionLesion SegmentationPrognosisSegmentationInterpretability of Uncertainty: Exploring Cortical Lesion Segmentation in Multiple Sclerosis
Uncertainty quantification (UQ) has become critical for evaluating the reliability of artificial intelligence systems, especially in medical image segmentation. This study addresses the interpretability of instance-wise …
Image SegmentationLesion SegmentationMedical Image SegmentationSegmentation+2