用极小数据集训练模型,自动修复医学图像分割的异常结果并给出置信度。
Learning a Sampling-Free Variational DNN Plugin from Tiny Training Sets to Refine OOD Segmentation With Uncertainty Estimation

- 基于小样本学习解剖几何先验,无需目标数据即可优化分割
- 在4类临床任务中显著提升分割合理性与准确性,误差大幅下降
- 无需采样、计算高效,适合医疗场景中的快速部署与不确定性评估
深度神经网络在面对不同扫描仪和成像协议导致的分布外(OOD)医学图像时常常失效。由于获取和标注新医学数据成本高昂,重新训练模型通常不切实际。为此,我们提出VarDeepPCA,一种轻量级变分DNN框架,通过利用内在几何先验来恢复/优化退化的分割图。不同于需目标域数据或大量预训练的方法,VarDeepPCA仅用少量分布内(ID)数据显式学习有效解剖结构的分布。其新颖的变分学习框架通过重新诠释softmax映射,实现精确分布建模,从而支持计算高效的无采样学习与推理,并可提供分割结果的不确定性估计。我们在14个公开数据集上验证了该框架在心肌、视网膜神经纤维层、前列腺和胎儿头部分割四个临床应用中的有效性。对比15种现有方法表明,VarDeepPCA在无额外训练数据的情况下,持续显著提升对OOD数据生成的分割图的解剖合理性与临床可用性,同时大幅降低错误率。
原文摘要 · Abstract (English)
Deep neural networks (DNNs) frequently fail to generalize to out-of-distribution (OOD) medical images because of variations in scanners and acquisition protocols. Retraining DNN models to address these distribution shifts is often impractical due to the high cost of acquiring and annotating new medical datasets. To address this, we introduce VarDeepPCA, a novel lightweight variational DNN framework designed to restore/refine degraded segmentation maps by leveraging intrinsic geometric priors. Unlike existing approaches that require target-domain data or extensive pre-training, our VarDeepPCA explicitly learns a distribution of valid anatomical geometries using only small in-distribution (ID) datasets. Theoretically, our novel variational learning framework leverages a reinterpretation of the softmax mapping to implicitly perform exact distribution modeling, thereby enabling computationally efficient, sampling-free learning and inference. This also enables VarDeepPCA to provide uncertainty estimates associated with its restored segmentation maps. We empirically validate our framework across 4 distinct clinical applications, using 14 publicly available datasets, involving segmentation of the myocardium, neuroretinal rim, prostate, and fetal head. Comparisons against 15 existing methods demonstrate that VarDeepPCA consistently restores segmentation maps produced by the existing methods on OOD data to (i) significantly improve anatomical plausibility of geometries and clinical utility of the segmentations, and (ii) significantly reduce errors, without needing any more training data than that used by existing methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。