arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.09016cs.CV
收藏

用于医学图像配准的金字塔注意力网络

Pyramid Attention Network for Medical Image Registration

School of Biomedical Engineering, Shenzhen University(深圳大学生物医学工程学院);Medical UltraSound Image Computing (MUSIC) Lab(医学超声图像计算实验室);Smart Medical Imaging, Learning and Engineering (SMILE) Lab(智能医学影像、学习与工程实验室)

Zhuoyuan Wang, Haiqiao Wang, Yi Wang

AI总结:

针对医学图像大形变配准中CNN忽略特征增强与Transformer存在信息冗余的问题,提出金字塔注意力网络(PAN),结合双流金字塔编码器与多头局部注意力解码器,在多个MRI数据集上取得优于现有方法的配准性能。

摘要:

基于深度学习的配准网络的出现解决了传统迭代方法耗时的挑战。然而,当前配准网络在全面捕捉空间关系方面的潜力尚未被充分探索,导致在大形变图像配准中性能不足。纯卷积神经网络(CNN)忽略了特征增强,而当前基于Transformer的网络容易产生信息冗余。为缓解这些问题,我们提出了一种用于可变形医学图像配准的金字塔注意力网络(PAN)。具体而言,所提出的PAN结合了带有通道注意力的双流金字塔编码器,以提升特征表示。此外,引入了多头局部注意力Transformer作为解码器,用于分析运动模式并生成形变场。在两个公开的脑部磁共振成像(MRI)数据集和一个腹部MRI数据集上的大量实验表明,我们的方法在取得良好配准性能的同时,优于多种基于CNN和Transformer的配准网络。代码已公开于https://github.com/JuliusWang-7/PAN。

英文摘要:

The advent of deep-learning-based registration networks has addressed the time-consuming challenge in traditional iterative methods.However, the potential of current registration networks for comprehensively capturing spatial relationships has not been fully explored, leading to inadequate performance in large-deformation image registration.The pure convolutional neural networks (CNNs) neglect feature enhancement, while current Transformer-based networks are susceptible to information redundancy.To alleviate these issues, we propose a pyramid attention network (PAN) for deformable medical image registration.Specifically, the proposed PAN incorporates a dual-stream pyramid encoder with channel-wise attention to boost the feature representation.Moreover, a multi-head local attention Transformer is introduced as decoder to analyze motion patterns and generate deformation fields.Extensive experiments on two public brain magnetic resonance imaging (MRI) datasets and one abdominal MRI dataset demonstrate that our method achieves favorable registration performance, while outperforming several CNN-based and Transformer-based registration networks.Our code is publicly available at https://github.com/JuliusWang-7/PAN.

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

↑