Current browse context:
eess.IV
Change to browse by:
References & Citations
Electrical Engineering and Systems Science > Image and Video Processing
Title: Segmentation-Consistent Probabilistic Lesion Counting
(Submitted on 11 Apr 2022 (v1), last revised 12 May 2022 (this version, v2))
Abstract: Lesion counts are important indicators of disease severity, patient prognosis, and treatment efficacy, yet counting as a task in medical imaging is often overlooked in favor of segmentation. This work introduces a novel continuously differentiable function that maps lesion segmentation predictions to lesion count probability distributions in a consistent manner. The proposed end-to-end approach--which consists of voxel clustering, lesion-level voxel probability aggregation, and Poisson-binomial counting--is non-parametric and thus offers a robust and consistent way to augment lesion segmentation models with post hoc counting capabilities. Experiments on Gadolinium-enhancing lesion counting demonstrate that our method outputs accurate and well-calibrated count distributions that capture meaningful uncertainty information. They also reveal that our model is suitable for multi-task learning of lesion segmentation, is efficient in low data regimes, and is robust to adversarial attacks.
Submission history
From: Chelsea Myers-Colet [view email][v1] Mon, 11 Apr 2022 17:26:49 GMT (1378kb,D)
[v2] Thu, 12 May 2022 13:04:37 GMT (1377kb,D)
Link back to: arXiv, form interface, contact.