中文

NePhi:用于近似微分同胚医学图像配准的神经形变场

计算机视觉与模式识别 2024-09-30 v3

摘要

本工作提出 NePhi,一种可生成近似微分同胚变换的泛化神经形变模型。与基于学习的主流配准方法中使用的体素级变换场不同,NePhi 以函数方式表示形变,从而在训练与推理时的内存消耗、推理时间、配准精度以及变换正则性设计空间内带来极大灵活性。具体而言,NePhi 1)相较体素级学习方法所需内存更少,2)通过预测潜码提升推理速度,优于当前仅依赖优化的神经形变配准方法,3)通过实例优化提升精度,4)展现出医学图像配准极为看重的优异形变正则性。我们在 2D 合成数据集以及真实 3D 医学图像数据集(如肺部与脑部)上展示了 NePhi 的性能。结果表明,在单分辨率配准设定下 NePhi 可匹配体素表示的精度。对于多分辨率配准,我们的方法在借助实例优化匹配当前 SOTA 基于学习的配准方法精度的同时,将内存需求降低了五倍。我们的代码见 https://github.com/uncbiag/NePhi。

关键词

引用

@article{arxiv.2309.07322,
  title  = {$\texttt{NePhi}$: Neural Deformation Fields for Approximately Diffeomorphic Medical Image Registration},
  author = {Lin Tian and Hastings Greer and Raúl San José Estépar and Roni Sengupta and Marc Niethammer},
  journal= {arXiv preprint arXiv:2309.07322},
  year   = {2024}
}

备注

ECCV 2024