计算机科学 > 计算机视觉与模式识别
[提交于 2025年8月2日
]
标题: 基于深度学习贝叶斯建模的不确定性感知分割质量预测:皮肤癌和肝脏分割的全面评估与解释
标题: Uncertainty-Aware Segmentation Quality Prediction via Deep Learning Bayesian Modeling: Comprehensive Evaluation and Interpretation on Skin Cancer and Liver Segmentation
摘要: 图像分割是计算生物医学图像分析中的关键步骤,通常在训练和验证过程中使用Dice系数等指标进行评估。然而,在没有手动标注的临床环境中,评估分割质量变得具有挑战性,缺乏可靠性指标的模型面临采用障碍。为解决这一差距,我们提出了一种新框架,在测试时无需真实标注即可预测分割质量。我们的方法引入了两个互补框架:一个利用预测的分割和不确定性图,另一个整合原始输入图像、不确定性图和预测的分割图。我们提出了两种基准分割模型——SwinUNet和带有ResNet50的特征金字塔网络的贝叶斯适应版本,使用蒙特卡洛Dropout、集成和测试时间增强来量化不确定性。我们在二维皮肤病变和三维肝脏分割数据集上评估了四种不确定性估计:置信度图、熵、互信息和期望成对Kullback-Leibler散度,并分析了它们与分割质量指标的相关性。我们的框架在HAM10000数据集上达到了93.25的R2分数和96.58的皮尔逊相关性,优于之前的分割质量评估方法。对于三维肝脏分割,使用熵的测试时间增强达到了85.03的R2分数和65.02的皮尔逊相关性,展示了跨模态的鲁棒性。此外,我们提出了一种聚合策略,将多个不确定性估计结合成每张图像的一个分数,提供更稳健和全面的分割质量评估。最后,我们使用Grad-CAM和基于UMAP的嵌入分析来解释模型的行为和可靠性,突出了不确定性集成的影响。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.