WSR betting bound re-certifies CT segmentation AI with 25 local cases
7 of 12 organs fail recall thresholds when nnU-Net transfers to new clinical sites...
Deploying AI segmentation models across hospitals is risky: performance degrades under clinical domain shift. A new arXiv paper by Souraj Adhikary, Negar Chabi, and Andre Mastmeyer tackles this with distribution-free risk control that adds per-organ recall guarantees to frozen segmentation models. Using an AMOS-trained nnU-Net as a case study, the team audited transfer to the RAOS dataset and discovered that 7 out of 12 organs exceeded an α=0.10 recall threshold—meaning over a tenth of diseased tissue would be missed without adjustment.
The paper systematically compares Risk-Controlling Prediction Sets (RCPS), Conformal Risk Control (CRC), and a novel Waudby-Smith-Ramdas (WSR) betting bound for local re-certification. Classic fixed or global thresholds offer no per-organ guarantees, and smaller calibration sets can mask exceedances with conservative or vacuous thresholds. WSR re-certifies six Tier-1 organs with only 25 local cases versus 30-40 for Hoeffding-Bentkus (HB). CRC needs even fewer (10-15), but shows a heavier individual-case tail—a critical safety concern in medical imaging. No Tier-2 organ met the illustrative precision criterion at 25 cases, highlighting the trade-off between recall and precision in low-data settings.
- AMOS-trained nnU-Net fails recall thresholds (α=0.10) for 7/12 organs on RAOS transfer
- WSR betting bound re-certifies 6 Tier-1 organs with 25 local cases, vs 30-40 for Hoeffding-Bentkus
- CRC requires only 10-15 local cases but has heavier worst-case tail risk per organ
Why It Matters
Radiation-safe AI segmenters need per-organ guarantees; this method cuts re-certification cost and improves trust for clinical deployment.