Work

Aggregating Image Segmentation Predictions with Probabilistic Risk Control Guarantees

Year: 2025
Type: article
Abstract: In this work, we introduce a framework to combine arbitrary image segmentation algorithms from different agents under data privacy constraints to produce an aggregated prediction set satisfying finite-sample risk control guarantees. We leverage distribution-free uncertainty quantification techniques in order to aggregate deep neural networks for image segmentation tasks. Our method can be applied in settings to merge the predictions of multiple agents with arbitrarily dependent prediction sets. Moreover, we perform experiments in medical imaging tasks to illustrate our proposed framework. Our results show that the framework reduced the empirical false positive rate by 50% without compromising the false negative rate, with respect to the false positive rate of any of the constituent models in the aggregated prediction algorithm.
Source: Mathematics
Authors: Joaquin Alvarez, Edgar Roman-Rangel
Language: English
Open Access status: gold
Estimated APC paid: $3,137
OpenAlex ID: W4410623671
FWCI:
Cites: 30
Cited by:
Related to: 10