arXiv:2501.04794eess.IVcs.CV2025-01

提出可旋转不变的深度网络,直接在原始dMRI数据上实现高效非刚性配准。

A Steerable Deep Network for Model-Free Diffusion MRI Registration

  • 基于SE(3)等变UNet生成保持几何特性的速度场
  • 在HCP数据集上达到顶尖配准精度,无需估计衍生表示
  • 适合需要高精度、免预处理的dMRI分析研究者

非刚性配准对医学图像分析至关重要,但扩散磁共振成像(dMRI)因其高维、方向依赖特性而面临挑战。传统方法虽准确但计算量大,深度学习虽高效,却在非刚性dMRI配准中应用较少。本文提出一种新型无模型、非刚性原始dMRI配准框架,无需显式重定向。与依赖扩散张量或纤维取向分布函数等派生表示的方法不同,本方法将配准建模为位置-方向空间的等变微分同胚。核心是基于SE(3)等变的UNet,生成保持原始dMRI域几何性质的速度场。引入基于傅里叶空间最大均值差异的新损失函数,隐式匹配图像间的平均传播器。在人类连接组计划(HCP)dMRI数据上的实验表明,性能媲美现有最先进方法,且避免了估算派生表示的开销。该工作为直接在采集空间进行数据驱动、几何感知的dMRI配准奠定了基础。

原文摘要 · Abstract (English)

Nonrigid registration is vital to medical image analysis but remains challenging for diffusion MRI (dMRI) due to its high-dimensional, orientation-dependent nature. While classical methods are accurate, they are computationally demanding, and deep neural networks, though efficient, have been underexplored for nonrigid dMRI registration compared to structural imaging. We present a novel, deep learning framework for model-free, nonrigid registration of raw diffusion MRI data that does not require explicit reorientation. Unlike previous methods relying on derived representations such as diffusion tensors or fiber orientation distribution functions, in our approach, we formulate the registration as an equivariant diffeomorphism of position-and-orientation space. Central to our method is an $\mathsf{SE}(3)$-equivariant UNet that generates velocity fields while preserving the geometric properties of a raw dMRI's domain. We introduce a new loss function based on the maximum mean discrepancy in Fourier space, implicitly matching ensemble average propagators across images. Experimental results on Human Connectome Project dMRI data demonstrate competitive performance compared to state-of-the-art approaches, with the added advantage of bypassing the overhead for estimating derived representations. This work establishes a foundation for data-driven, geometry-aware dMRI registration directly in the acquisition space.

dMRI配准等变网络扩散模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。