arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.02847cs.CVcs.AI
收藏

用于三维医学图像描述的深隐式统计形状模型

Deep Implicit Statistical Shape Models for 3D Medical Image Delineation

PAII Inc(PAII 公司);University of Texas at Arlington(德克萨斯大学阿灵顿分校)

Ashwin Raju, Shun Miao, Dakai Jin, Le Lu, Junzhou Huang, Adam P. Harrison

摘要:

解剖结构的3D描绘是医学成像分析中的主要目标。在深度学习之前,施加解剖学约束并产生高质量表面的统计形状模型是核心技术。在深度学习之前,施加解剖学约束并产生高质量表面的统计形状模型是核心技术。如今,全卷积网络(FCN)虽然占主导地位,但并不提供这些功能。我们提出了深度隐式统计形状模型(DISSM),这是一种新的描绘方法,它将卷积神经网络(CNN)的表示能力与SSM的鲁棒性结合在一起。DISSM使用深层隐式表面表示来产生紧凑且描述性的形状潜在空间,其允许解剖学方差的统计模型。为了可靠地适合解剖学上合理的形状的图像,我们引入了一种新的刚性和非刚性的姿态估计管道,建模为马尔可夫决策过程(MDP)。我们概述了一个训练制度,包括倒置情景训练和边缘空间学习(MSL)的深度实现。在病理肝脏分割任务上的数据集内实验表明,DISSM可以比三种领先的FCN模型(包括nnU-Net)更稳健地执行:将平均Hausdorff距离(HD)减少7.7-14.3mm,并将最差情况下的Dice-Sorensen系数(DSC)提高1.2- 2.3%。更重要的是,在直接反映临床部署场景的数据集上进行的跨数据集实验表明,DISSM将平均DSC和HD分别提高了3.5-5.9%和12.3-24.5mm,最差情况DSC提高了5.4- 7.3%。这些改进超过了用高质量表面表示轮廓的任何好处。

英文摘要:

3D delineation of anatomical structures is a cardinal goal in medical imaging analysis. Prior to deep learning, statistical shape models that imposed anatomical constraints and produced high quality surfaces were a core technology. Prior to deep learning, statistical shape models that imposed anatomical constraints and produced high quality surfaces were a core technology. Today fully-convolutional networks (FCNs), while dominant, do not offer these capabilities. We present deep implicit statistical shape models (DISSMs), a new approach to delineation that marries the representation power of convolutional neural networks (CNNs) with the robustness of SSMs. DISSMs use a deep implicit surface representation to produce a compact and descriptive shape latent space that permits statistical models of anatomical variance. To reliably fit anatomically plausible shapes to an image, we introduce a novel rigid and non-rigid pose estimation pipeline that is modelled as a Markov decision process(MDP). We outline a training regime that includes inverted episodic training and a deep realization of marginal space learning (MSL). Intra-dataset experiments on the task of pathological liver segmentation demonstrate that DISSMs can perform more robustly than three leading FCN models, including nnU-Net: reducing the mean Hausdorff distance (HD) by 7.7-14.3mm and improving the worst case Dice-Sorensen coefficient (DSC) by 1.2-2.3%. More critically, cross-dataset experiments on a dataset directly reflecting clinical deployment scenarios demonstrate that DISSMs improve the mean DSC and HD by 3.5-5.9% and 12.3-24.5mm, respectively, and the worst-case DSC by 5.4-7.3%. These improvements are over and above any benefits from representing delineations with high-quality surface.

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

↑