Search papers, labs, and topics across Lattice.
This study introduces VIDS-Seg, a novel approach for uncertainty quantification in pediatric cardiac ultrasound segmentation, addressing the critical need for models that can identify their own failures, especially in underrepresented populations. By leveraging the VIDS framework, VIDS-Seg applies amortized variational inference to enhance the reliability of segmentation predictions in a zero-shot setting, specifically evaluating on pediatric echocardiography data. The results show that VIDS-Seg not only matches the accuracy of competitive models but also significantly improves the spatial alignment of predicted uncertainty with segmentation errors, leading to better clinical outcomes in infant cardiac assessments.
VIDS-Seg reveals that OOD-aware uncertainty quantification can effectively flag silent failures in pediatric cardiac segmentation without the need for retraining on specialized data.
Reliable clinical deployment of machine learning requires models that know when they are likely to fail, particularly for subgroups underrepresented in training data. A common case is pediatric care, where models trained on adult cohorts can silently under-perform on children with no indication that something has gone wrong. As retraining with labeled pediatric data is often infeasible, detecting such failures at inference time is a critical clinical need. Building on the VIDS (Variational Inference under Distribution Shifts) framework, we introduce VIDS-Seg, which applies amortized variational inference over a lightweight prediction head to make this adaptive, OOD-aware prior tractable for dense image segmentation. We evaluate VIDS-Seg on left ventricular segmentation in echocardiography, a setting where pediatric anatomy differs systematically from the adult population most segmentation models are trained on, training on an adult cohort (EchoNet-Dynamic) and evaluating zero-shot on a pediatric cohort (EchoNet-Pediatric). Across all age strata, VIDS-Seg matches competitive baselines in segmentation accuracy while producing substantially higher spatial correspondence between predicted uncertainty and segmentation error, an advantage that persists even after applying temperature scaling to all baselines. Downstream, it yields more accurate and stable ejection fraction estimates and more reliable detection of cardiac malfunction in the infant subgroup. Our results indicate that OOD-aware uncertainty quantification can serve as a practical safety layer for deployed segmentation models, enabling detection of silent failures in underrepresented subgroups without retraining or additional labeled data.
