arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.03470cs.CVcs.AIcs.HCcs.LG

改善深度贝叶斯医学图像分割中的不确定性-误差对应关系

Improving Uncertainty-Error Correspondence in Deep Bayesian Medical Image Segmentation

Leiden University Medical Center(莱顿大学医学中心);Erasmus Medical Center(伊拉斯姆斯医学中心);Holland Proton Therapy Centre(荷兰质子治疗中心);TU Delft(代尔夫特理工大学)

Prerak Mody, Nicolas F. Chaves-de-Plaza, Chinmay Rao, Eleftheria Astrenidou, Mischa de Ridder, Nienke Hoekstra, Klaus Hildebrandt, Marius Staring

AI总结:

针对医学分割自动轮廓QA,本文用AvU损失训练FlipOut贝叶斯模型,使不确定性集中于错误体素,并在头颈CT和前列腺MR上验证其ROC/PR表现。

摘要:

深度学习等自动化工具在医学图像分割中的日益普及,缓解了人工勾画的瓶颈。这将人工劳动转移到了自动轮廓的质量评估(QA)上,其中包括检测误差并加以纠正。半自动QA的一种潜在解决方案是使用深度贝叶斯不确定性来推荐可能存在误差的区域,从而减少用于误差检测的时间。以往工作已经研究了不确定性与误差之间的对应关系,然而,尚无工作致力于改善贝叶斯不确定性图的“效用”,使其只出现在不准确区域而不出现在准确区域。我们的工作使用Accuracy-vs-Uncertainty(AvU,准确性对不确定性)损失训练FlipOut模型,该损失促使不确定性仅出现在不准确区域。我们将该方法应用于两个放疗身体部位的数据集,即头颈部CT和前列腺MR扫描。通过受试者工作特征(ROC)曲线和精确率-召回率(PR)曲线,针对体素不准确性评估不确定性热图(即预测熵)。数值结果表明,与贝叶斯基线相比,所提出方法成功抑制了准确体素的不确定性,同时在不准确体素上保持相近的不确定性存在水平。复现实验的代码可在https://github.com/prerakmody/bayesuncertainty-error-correspondence获取。

英文摘要:

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

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑