Controlling False Positives in Image Segmentation via Conformal Prediction
By: Luca Mossina, Corentin Friedrich
Potential Business Impact:
Helps doctors trust AI's medical image analysis.
Reliable semantic segmentation is essential for clinical decision making, yet deep models rarely provide explicit statistical guarantees on their errors. We introduce a simple post-hoc framework that constructs confidence masks with distribution-free, image-level control of false-positive predictions. Given any pretrained segmentation model, we define a nested family of shrunken masks obtained either by increasing the score threshold or by applying morphological erosion. A labeled calibration set is used to select a single shrink parameter via conformal prediction, ensuring that, for new images that are exchangeable with the calibration data, the proportion of false positives retained in the confidence mask stays below a user-specified tolerance with high probability. The method is model-agnostic, requires no retraining, and provides finite-sample guarantees regardless of the underlying predictor. Experiments on a polyp-segmentation benchmark demonstrate target-level empirical validity. Our framework enables practical, risk-aware segmentation in settings where over-segmentation can have clinical consequences. Code at https://github.com/deel-ai-papers/conseco.
Similar Papers
Statistical Management of the False Discovery Rate in Medical Instance Segmentation Based on Conformal Risk Control
CV and Pattern Recognition
Makes medical scans more accurate for doctors.
Conformal Prediction for Image Segmentation Using Morphological Prediction Sets
CV and Pattern Recognition
Shows how sure a computer is about its picture guesses.
Aerial Image Classification in Scarce and Unconstrained Environments via Conformal Prediction
Machine Learning (CS)
Shows how sure a computer is about its guesses.