arXiv:2512.10715cs.CV2025-12中稿 · publication at MID…被引 1

为胸部X光解剖定位分割提供不确定性量化方法,提升模型可靠性。

CheXmask-U: Quantifying uncertainty in landmark-based anatomical segmentation for X-ray images

  • 结合卷积与图生成解码器,从潜在空间直接计算不确定性。
  • 在65万张图像上验证,扰动越严重,不确定性越高,可识别错误预测。
  • 公开大规模带不确定性的标注数据集,适合医疗影像研究者使用。

本文研究胸部X光图像中基于解剖标志的分割任务中的不确定性估计。受混合神经网络架构启发——该架构结合标准卷积编码器与基于图的生成解码器,并利用其变分潜在空间,我们推导出两种互补的度量:(i) 潜在不确定性,直接来自学习到的分布参数;(ii) 预测不确定性,通过从潜在样本生成多个随机输出预测获得。通过受控扰动实验,我们发现两种不确定性度量均随扰动强度增加,反映全局与局部退化。实验表明,这些不确定性信号能有效识别不可靠预测,并在CheXmask数据集上支持分布外检测。更重要的是,我们发布了CheXmask-U(huggingface.co/datasets/mcosarinsky/CheXmask-U),一个包含657,566张胸部X光图像的大型数据集,每节点附带不确定性估计,使研究者可在使用解剖掩码时考虑分割质量的空间差异。研究结果确立了不确定性估计在提升胸部X光解剖分割方法鲁棒性与安全部署方面的潜力。完整交互式演示可通过huggingface.co/spaces/matiasky/CheXmask-U访问,源代码见github.com/mcosarinsky/CheXmask-U。

原文摘要 · Abstract (English)

In this work, we study uncertainty estimation for anatomical landmark-based segmentation on chest X-rays. Inspired by hybrid neural network architectures that combine standard image convolutional encoders with graph-based generative decoders, and leveraging their variational latent space, we derive two complementary measures: (i) latent uncertainty, captured directly from the learned distribution parameters, and (ii) predictive uncertainty, obtained by generating multiple stochastic output predictions from latent samples. Through controlled corruption experiments we show that both uncertainty measures increase with perturbation severity, reflecting both global and local degradation. We demonstrate that these uncertainty signals can identify unreliable predictions by comparing with manual ground-truth, and support out-of-distribution detection on the CheXmask dataset. More importantly, we release CheXmask-U (huggingface.co/datasets/mcosarinsky/CheXmask-U), a large scale dataset of 657,566 chest X-ray landmark segmentations with per-node uncertainty estimates, enabling researchers to account for spatial variations in segmentation quality when using these anatomical masks. Our findings establish uncertainty estimation as a promising direction to enhance robustness and safe deployment of landmark-based anatomical segmentation methods in chest X-ray. A fully working interactive demo of the method is available at huggingface.co/spaces/matiasky/CheXmask-U and the source code at github.com/mcosarinsky/CheXmask-U.

医学影像不确定性估计胸部X光分割

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