arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.08652cs.CV
收藏

基于神经元同构流的保拓扑形状重建与配准

Topology-Preserving Shape Reconstruction and Registration via Neural Diffeomorphic Flow

University of California, Irvine(加州大学欧文分校)

Shanlin Sun, Kun Han, Deying Kong, Hao Tang, Xiangyi Yan, Xiaohui Xie

摘要:

深度隐式函数(DIF)表示3D几何形状,具有通过深度神经网络学习的连续符号距离函数。最近已经提出了基于DIF的方法来同时处理形状重建和密集点对应,通过学习DIF建模的形状模板来捕获同一类形状之间的语义关系。这些方法在重建3D形状和推断对应关系方面提供了很大的灵活性和准确性。然而,与基于网格的模板匹配方法不同,从这些方法构建的点对应本质上不保留形状的拓扑结构。这限制了它们在3D几何形状上的应用,其中存在基础拓扑结构并且重要,例如医学图像中的解剖结构。在本文中,我们提出了一个新的模型,称为神经同构流(NDF)学习深隐式形状模板,表示形状作为模板的条件同构变形,本质上保持形状拓扑。该变形是通过一个自动解码器实现的,该解码器由神经常微分方程(NODE)块组成,逐步将形状映射到隐式模板。我们进行了广泛的实验上的几个医学图像器官分割数据集,以评估NDF重建和对齐形状的有效性。NDF在准确性和质量方面始终实现最先进的器官形状重建和配准结果。源代码可在https://github.com/Siwensun/Neural_Diffeomorphic_Flow--NDF上公开获得。

英文摘要:

Deep Implicit Functions (DIFs) represent 3D geometry with continuous signed distance functions learned through deep neural nets. Recently DIFs-based methods have been proposed to handle shape reconstruction and dense point correspondences simultaneously, capturing semantic relationships across shapes of the same class by learning a DIFs-modeled shape template. These methods provide great flexibility and accuracy in reconstructing 3D shapes and inferring correspondences. However, the point correspondences built from these methods do not intrinsically preserve the topology of the shapes, unlike mesh-based template matching methods. This limits their applications on 3D geometries where underlying topological structures exist and matter, such as anatomical structures in medical images. In this paper, we propose a new model called Neural Diffeomorphic Flow (NDF) to learn deep implicit shape templates, representing shapes as conditional diffeomorphic deformations of templates, intrinsically preserving shape topologies. The diffeomorphic deformation is realized by an auto-decoder consisting of Neural Ordinary Differential Equation (NODE) blocks that progressively map shapes to implicit templates. We conduct extensive experiments on several medical image organ segmentation datasets to evaluate the effectiveness of NDF on reconstructing and aligning shapes. NDF achieves consistently state-of-the-art organ shape reconstruction and registration results in both accuracy and quality. The source code is publicly available at https://github.com/Siwensun/Neural_Diffeomorphic_Flow--NDF.

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

↑