Region-aware error triage for medical image segmentation with groupwise conformal risk control
Organizations: NIMISHES Lab, Dhaka, Bangladesh · Faculty of Electronic Engineering and Technology, Universiti Malaysia Perlis, 02600 Arau, Perlis, Malaysia
Abstract
Segmentation masks are increasingly reviewed rather than redrawn, which requires knowing where a mask is likely wrong. Learned error maps rank pixels by risk, but the number of pixels to warn is chosen ad hoc, and calibrating one warning threshold over all error pixels may leave errors far from the predicted contour under-flagged. We propose RACE-Triage, a post-hoc error triage framework that leaves the segmentation mask unchanged. A risk network is trained on out-of-fold predictions from backbone-independent inputs. The predicted contour partitions each image into a boundary band and a non-boundary region, and conformal risk control is applied separately in each region to bound the expected fraction of missed errors among cases with errors there. On ISIC 2018 dermoscopy and BUSI breast ultrasound (where patient identifiers are unavailable, so independence across images is assumed), pooled control met its overall target but missed on average 31.8% and 36.4% of non-boundary errors. Regional control reduced these fractions to 19.3% and 20.9% and raised the recall of size-filtered false-negative components from 0.722 to 0.818 and from 0.611 to 0.763. The mean warned area grew by 1.6 and 4.9 percentage points, and its 95th percentile reached 23.5% and 38.5%. The effect held across three seeds and a second backbone, and area-matched heuristics did not reproduce it. On the external PH set, frozen thresholds exceeded both targets, whereas recalibration with 100 local images brought the mean over repeated splits below them. Regional calibration improves non-boundary error coverage at a measured increase in warned area.
Figures & tables
| Method family | Produces or controls | RACE-Triage instead |
|---|---|---|
| Uncertainty estimation [ 10 , 13 , 14 ] | Pixel-wise uncertainty scores | Calibrates warning regions for a given score |
| Error and quality prediction [ 17 , 18 , 20 ] | Error maps or case-level quality scores | Converts an error map into warnings with a stated risk |
| Conformal segmentation [ 23 , 24 , 25 ] | Prediction sets or quality ranges for the segmentation | Leaves the mask unchanged |
| Conformal false-negative or false-discovery control [ 47 , 48 , 49 ] | Missed or spurious parts of the predicted object | Controls the share of all segmentation errors left unwarned |
| Group-wise risk control [ 26 , 29 ] | Risk per predefined group, e.g., per organ | Spatial groups derived from the predicted contour |
| Dataset | Modality | Images | Partition sizes | Test Dice | Error pixels (%) | Cal. | Test | share (%) |
|---|---|---|---|---|---|---|---|---|
| ISIC 2018 | Dermoscopy | 2594 | 1686 / 259 / 324 / 325 | 0.883 | 3.2 | 322 / 128 | 323 / 131 | 21.6 |
| BUSI | Breast ultrasound | 780 | 507 / 78 / 97 / 98 | 0.751 | 5.6 | 82 / 49 | 83 / 52 | 37.9 |
| PH 2 | Dermoscopy (external) | 200 | – / – / – / 200 | 0.862 | 5.4 | – | 200 / 118 | 33.4 |
| ISIC 2018 | BUSI | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Error score | Params (M) | ms | AUPRC | Brier | CR@10 | @10 | AUPRC | Brier | CR@10 | @10 | ||
| Predictive entropy | 0 | 0.1 | 0.318 | 0.192 | – | 0.512 | 0.214 | 0.276 | 0.163 | – | 0.384 | 0.152 |
| Test-time augmentation | 0 | 31.4 | 0.336 | 0.205 | – | 0.538 | 0.236 | 0.291 | 0.172 | – | 0.405 | 0.169 |
| Fold-ensemble disagreement | 38.8 | 19.7 | 0.352 | 0.221 | – | 0.561 | 0.268 | 0.309 | 0.188 | – | 0.431 | 0.204 |
| Local head (P1) | 0.01 | 0.4 | 0.394 | 0.251 | 2.48 | 0.603 | 0.331 | 0.341 | 0.207 | 4.71 | 0.466 | 0.248 |
| P1 + distance (P2) | 0.01 | 1.1 | 0.405 | 0.262 | 2.44 | 0.618 | 0.352 | 0.350 | 0.214 | 4.66 | 0.479 | 0.262 |
| Policy | Area (%) | Corr. (%) | Reg. | FP rec. | FN rec. | |||
|---|---|---|---|---|---|---|---|---|
| ISIC 2018 (in-domain, ) | ||||||||
| Pooled CRC, fixed | 0.181 | 0.137 | 0.318 | 9.8 | 7.3 | 3.4 | 0.802 | 0.722 |
| Regional CRC, fixed (ours) | 0.190 | 0.188 | 0.193 | 11.4 | 8.9 | 4.7 | 0.829 | 0.818 |
| Pooled CRC, adaptive | 0.183 | 0.142 | 0.301 | 9.3 | 6.7 | 3.1 | 0.795 | 0.731 |
| Regional CRC, adaptive | 0.192 | 0.190 | 0.197 | 10.6 | 8.1 | 4.2 | 0.824 | 0.812 |
| Top-ranked, 10% area | 0.174 | 0.121 | 0.352 | 10.0 | 7.5 | 3.6 | 0.789 | 0.716 |
| Primary split | Re-partitions | ||||||||
| Policy | Guarantee | Area (%) | Reg. | FN rec. | Mean | (%) | |||
| ISIC 2018 | |||||||||
| Regional CRC, fixed (ours) | group | 0.190 | 0.188 | 0.193 | 11.4 | 4.7 | 0.818 | 0.194 | 41 |
| Regional plug-in (no term) | – | 0.194 | 0.191 | 0.199 | 11.2 | 4.6 | 0.812 | 0.201 | 52 |
| Fixed group quotas | – | 0.219 | 0.214 | 0.236 | 11.4 | 4.1 | 0.792 | 0.231 | 89 |
| Distance-boosted ranking + pooled CRC | pooled | 0.186 | 0.171 | 0.238 | 11.0 | 4.3 | 0.781 | 0.234 | 91 |
| Pooled | Regional | Area (%) | ||||||
| Setting | Test Dice | Pooled | Regional | |||||
| ISIC 2018 | ||||||||
| U-Net, seed 1 (main results) | 0.883 | 0.260 | 0.120 | 0.318 | 0.188 | 0.193 | 9.8 | 11.4 |
| U-Net, 3 seeds (mean SD) | 0.881 0.003 | 0.258 0.008 | 0.122 0.003 | 0.309 0.012 | 0.189 0.003 | 0.197 0.007 | 9.9 0.3 | 11.6 0.4 |
| DeepLabV3+, own | 0.897 | 0.270 | 0.110 | 0.296 | 0.186 | 0.191 | 8.9 | 10.3 |
| DeepLabV3+, U-Net , recalibrated | 0.897 | 0.245 | 0.095 | 0.327 | 0.190 | 0.198 | 10.6 | 12.6 |
| Pooled CRC | Regional CRC | ||||
| Risk score | AUPRC | Area | Area | ||
| ISIC 2018 (US-Loss segmenter, test Dice 0.886) | |||||
| US-Loss entropy | 0.361 | 0.402 | 10.6 | 0.198 | 16.9 |
| Multiscale (refitted) | 0.452 | 0.309 | 9.6 | 0.194 | 11.2 |
| BUSI (US-Loss segmenter, test Dice 0.758) | |||||
| US-Loss entropy | 0.318 | 0.441 | 16.3 | 0.211 | 27.8 |
| Policy | Area | FN rec. | (%) | ||
|---|---|---|---|---|---|
| Pooled, frozen ISIC | 0.162 | 0.372 | 12.4 | 0.649 | 100 |
| IQR | 0.146–0.179 | 0.341–0.401 | 11.3–13.5 | ||
| Regional, frozen ISIC | 0.215 | 0.263 | 13.9 | 0.751 | 94 |
| IQR | 0.196–0.233 | 0.241–0.287 | 12.6–15.1 | ||
| Pooled, recalibrated | 0.144 | 0.349 | 13.4 | 0.668 | 100 |
| IQR | 0.129–0.160 | 0.318–0.379 | 12.0–14.9 |
| Variant | Area | Reg. | FN rec. | ||
|---|---|---|---|---|---|
| Default ( px) | 0.188 | 0.193 | 11.4 | 4.7 | 0.818 |
| Narrower band ( px) | 0.191 | 0.196 | 11.9 | 5.2 | 0.821 |
| No distance channel in | 0.187 | 0.195 | 12.1 | 4.9 | 0.806 |
| Local head P1 as | 0.190 | 0.198 | 17.8 | 7.6 | 0.734 |
| Overlay dilated by 2 px | 0.151 | 0.172 | 15.3 | 2.9 | 0.861 |
| Union with component CRC | 0.176 | 0.181 | 12.9 | 5.0 | 0.897 |
| Dataset | Viol. | Viol. | Either | Area (%) | |
|---|---|---|---|---|---|
| ISIC 2018 | 16 / 6 | 0.046 | 0.053 | 0.058 | 23.7 |
| (0.026–0.075) | (0.022–0.107) | (0.036–0.090) | |||
| BUSI | 4 / 2 | 0.060 | 0.077 | 0.082 | 38.6 |
| (0.020–0.135) | (0.021–0.185) | (0.036–0.155) |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| ISIC 2018 | BUSI | ||||||
|---|---|---|---|---|---|---|---|
| Pooled | Regional | Pooled | Regional | ||||
| 0.10 | 25 | 0.722 | 0.818 | +0.096 | 0.611 | 0.763 | +0.152 |
| 0.25 | 25 | 0.583 | 0.709 | +0.126 | 0.472 | 0.642 | +0.170 |
| 0.50 | 25 | 0.412 | 0.537 | +0.125 | 0.318 | 0.469 | +0.151 |
| 0.10 | 100 | 0.701 | 0.815 | +0.114 | 0.598 | 0.771 | +0.173 |
| 0.50 | 100 | 0.368 | 0.511 | +0.143 | 0.291 | 0.447 | +0.156 |
| Subset | Pooled | Regional | Area (%) | |
|---|---|---|---|---|
| Error type in | ||||
| ISIC 2018, FP pixels | 74 | 0.274 | 0.171 | – |
| ISIC 2018, FN pixels | 96 | 0.341 | 0.204 | – |
| BUSI, FP pixels | 38 | 0.318 | 0.188 | – |
| BUSI, FN pixels | 33 | 0.402 | 0.226 | – |
| BUSI diagnostic category | ||||
Medical Image Segmentation
Medical image segmentation delineates organs, lesions, and other structures in scans to support measurement, diagnosis, treatment planning, and research.
See all 822 papers