arXiv:2409.03470cs.CVcs.AI2024-09中稿 · publication at the…被引 2

让医学影像分割的不确定性只出现在错误区域,提升自动质检效率。

Improving Uncertainty-Error Correspondence in Deep Bayesian Medical Image Segmentation

  • 用准确性-不确定性损失训练模型,让不确定只出现在错的地方
  • 在头颈部和前列腺数据集上,准确区域的不确定性显著降低
  • 适合需要高可靠自动质检的医疗AI开发者

深度学习在医学图像分割中的应用缓解了手动勾画的瓶颈,但人工工作转向了对自动化结果的质量评估(QA),即检测和修正误差。一种半自动QA的潜在方案是利用深度贝叶斯不确定性来推荐可能出错的区域,从而减少误差检测时间。已有研究探讨了不确定性与误差之间的对应关系,但尚未有工作致力于提升贝叶斯不确定性图的“实用性”——即仅在不准确区域出现,而不在准确区域出现。本文通过在FlipOut模型上引入准确性-不确定性(AvU)损失,使不确定性仅在不准确区域显现。我们在两个放疗体部部位的数据集(头颈部CT和前列腺MR)上验证该方法。使用受试者工作特征(ROC)和精确率-召回率(PR)曲线,以预测熵作为不确定性热图,评估其与体素级误差的对应关系。数值结果表明,相比贝叶斯基线,所提方法成功抑制了准确体素上的不确定性,同时在不准确体素上保持了相似的不确定性强度。实验代码已公开于https://github.com/prerakmody/bayesuncertainty-error-correspondence。

原文摘要 · Abstract (English)

Increased usage of automated tools like deep learning in medical image segmentation has alleviated the bottleneck of manual contouring. This has shifted manual labour to quality assessment (QA) of automated contours which involves detecting errors and correcting them. A potential solution to semi-automated QA is to use deep Bayesian uncertainty to recommend potentially erroneous regions, thus reducing time spent on error detection. Previous work has investigated the correspondence between uncertainty and error, however, no work has been done on improving the "utility" of Bayesian uncertainty maps such that it is only present in inaccurate regions and not in the accurate ones. Our work trains the FlipOut model with the Accuracy-vs-Uncertainty (AvU) loss which promotes uncertainty to be present only in inaccurate regions. We apply this method on datasets of two radiotherapy body sites, c.f. head-and-neck CT and prostate MR scans. Uncertainty heatmaps (i.e. predictive entropy) are evaluated against voxel inaccuracies using Receiver Operating Characteristic (ROC) and Precision-Recall (PR) curves. Numerical results show that when compared to the Bayesian baseline the proposed method successfully suppresses uncertainty for accurate voxels, with similar presence of uncertainty for inaccurate voxels. Code to reproduce experiments is available at https://github.com/prerakmody/bayesuncertainty-error-correspondence

医学影像贝叶斯网络不确定性分割

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