arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.18545eess.IVcs.AIcs.CV

利用多轴交叉协方差注意力推进可变形医学图像配准

Advancing Deformable Medical Image Registration with Multi-axis Cross-covariance Attention

The University of Sydney(悉尼大学);Shanghai Jiao Tong University(上海交通大学);Royal Prince Alfred Hospital(皇家阿尔弗雷德王子医院)

Mingyuan Meng, Michael Fulham, Lei Bi, Jinman Kim

AI总结:

提出多轴交叉协方差注意力(MAXCA)块,通过并行应用区域和扩张XCA,高效捕获高分辨率图像特征间的全局与局部长程依赖,提升可变形医学图像配准性能。

摘要:

可变形图像配准是医学图像分析的基本需求。最近,transformers 因其通过自注意力(SA)捕获长程依赖的能力,被广泛用于基于深度学习的配准方法。然而,SA 的高计算和内存负载(随空间分辨率呈二次增长)阻碍了 transformers 处理高分辨率图像特征中的细微纹理信息,例如在完整和半图像分辨率下。这限制了可变形配准,因为高分辨率纹理信息对于在细微解剖结构间找到精确的逐像素对应至关重要。交叉协方差注意力(XCA)作为 SA 的“转置”版本,跨特征通道操作,其复杂度随空间分辨率线性增长,为捕获高分辨率图像特征间的长程依赖提供了可行性。然而,现有的基于 XCA 的 transformers 仅捕获粗略的全局长程依赖,不适用于主要依赖细粒度局部对应的可变形图像配准。在本研究中,我们提出通过嵌入一种新的 XCA 机制来改进现有的基于深度学习的配准方法。为此,我们设计了一个针对可变形医学图像配准优化的基于 XCA 的 transformer 块,名为多轴 XCA(MAXCA)。我们的 MAXCA 作为通用网络块,可嵌入各种配准网络架构。它通过多轴设计并行应用区域和扩张 XCA,能同时捕获高分辨率图像特征间的全局和局部长程依赖。在两个基准完善的跨/患者内配准任务和七个公开医学数据集上的大量实验表明,我们的 MAXCA 块实现了最先进的配准性能。

英文摘要:

Deformable image registration is a fundamental requirement for medical image analysis. Recently, transformers have been widely used in deep learning-based registration methods for their ability to capture long-range dependency via self-attention (SA). However, the high computation and memory loads of SA (growing quadratically with the spatial resolution) hinder transformers from processing subtle textural information in high-resolution image features, e.g., at the full and half image resolutions. This limits deformable registration as the high-resolution textural information is crucial for finding precise pixel-wise correspondence between subtle anatomical structures. Cross-covariance Attention (XCA), as a "transposed" version of SA that operates across feature channels, has complexity growing linearly with the spatial resolution, providing the feasibility of capturing long-range dependency among high-resolution image features. However, existing XCA-based transformers merely capture coarse global long-range dependency, which are unsuitable for deformable image registration relying primarily on fine-grained local correspondence. In this study, we propose to improve existing deep learning-based registration methods by embedding a new XCA mechanism. To this end, we design an XCA-based transformer block optimized for deformable medical image registration, named Multi-Axis XCA (MAXCA). Our MAXCA serves as a general network block that can be embedded into various registration network architectures. It can capture both global and local long-range dependency among high-resolution image features by applying regional and dilated XCA in parallel via a multi-axis design. Extensive experiments on two well-benchmarked inter-/intra-patient registration tasks with seven public medical datasets demonstrate that our MAXCA block enables state-of-the-art registration performance.

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

↑