arXiv:2608.10903cs.CVcs.LG2026-08

提出可检测儿童超声心功能异常的可靠不确定性评估方法

VIDS-Seg: Towards Reliable Uncertainty Quantification in Pediatric Cardiac Ultrasound Segmentation

论文配图:VIDS-Seg: Towards Reliable Uncertainty Quantification in Pediatric Cardiac Ultrasound Segmentation
图 1 · 摘自论文原文
  • 基于变分推断构建轻量级预测头,实现推理时自适应不确定性估计
  • 在儿科数据上比基线高出38%的空间不确定性与误差对应率
  • 无需重新训练或标注数据,适合临床部署中发现模型沉默失效

机器学习在临床应用中需具备识别自身可能失效的能力,尤其对训练数据中代表性不足的亚群体。以儿科心脏超声为例,多数模型基于成人数据训练,难以有效泛化至儿童群体,且缺乏预警机制。本文提出VIDS-Seg,基于变分推断框架,在轻量级预测头中实现可扩展的分布偏移感知不确定性建模,用于密集图像分割。在左心室分割任务中,模型于成人数据集EchoNet-Dynamic上训练,零样本评估于儿科数据集EchoNet-Pediatric。结果表明,VIDS-Seg在所有年龄组均保持与基线相当的分割精度,同时显著提升预测不确定性与分割误差的空间一致性(提升38%),该优势在温度缩放后仍持续存在。下游任务中,其心输出分数估计更准确稳定,并在婴儿亚组中更可靠地检测心脏功能异常。结果表明,面向分布外的不确定性量化可作为部署模型的安全层,无需再训练或额外标注即可发现未覆盖群体中的沉默失败。

原文摘要 · Abstract (English)

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.

医学图像分割不确定性量化儿科影像零样本泛化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。