arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1906.02191eess.IVcs.LGstat.ML
收藏

用于器官分割细化的基于不确定性的图卷积网络

Uncertainty-based graph convolutional networks for organ segmentation refinement

Technische Universität München(慕尼黑工业大学);Johns Hopkins University(约翰斯·霍普金斯大学);ETH Zurich(苏黎世联邦理工学院)

Roger D. Soberanis-Mukul, Nassir Navab, Shadi Albarqouni

摘要:

CT容积中的器官分割是许多计算机辅助介入和诊断方法中的重要预处理步骤。近年来,卷积神经网络在这项任务中占据了主导地位。然而,由于该问题由于器官形状的高度可变性和组织之间的相似性而呈现出具有挑战性的环境,因此在输出分割中产生假阴性和假阳性区域是常见的问题。最近的工作表明,模型的不确定性分析可以为我们提供有用的信息,潜在的错误分割。在此背景下,我们提出了一种基于不确定性分析和图卷积网络的分割细化方法。我们使用特定输入量中卷积网络的不确定性水平来制定半监督图学习问题,该问题通过训练图卷积网络来解决。为了测试我们的方法,我们改进了2D U网的初始输出。我们用NIH胰腺数据集和医学分割十项全能的脾脏数据集验证了我们的框架。我们表明,我们的方法优于国家的最先进的CRF细化方法,提高骰子得分1%的胰腺和2%的脾脏,相对于原来的U-Net的预测。最后,我们讨论了该模型的结果和目前的局限性,为今后的工作在这一研究方向。出于可重复性的目的,我们将我们的代码公开。

英文摘要:

Organ segmentation in CT volumes is an important pre-processing step in many computer assisted intervention and diagnosis methods. In recent years, convolutional neural networks have dominated the state of the art in this task. However, since this problem presents a challenging environment due to high variability in the organ's shape and similarity between tissues, the generation of false negative and false positive regions in the output segmentation is a common issue. Recent works have shown that the uncertainty analysis of the model can provide us with useful information about potential errors in the segmentation. In this context, we proposed a segmentation refinement method based on uncertainty analysis and graph convolutional networks. We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem that is solved by training a graph convolutional network. To test our method we refine the initial output of a 2D U-Net. We validate our framework with the NIH pancreas dataset and the spleen dataset of the medical segmentation decathlon. We show that our method outperforms the state-of-the art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen, with respect to the original U-Net's prediction. Finally, we discuss the results and current limitations of the model for future work in this research direction. For reproducibility purposes, we make our code publicly available.

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

↑