arXiv:2412.11458cs.CV2024-12中稿 · TNNLS被引 37

提出混合残差注意力机制,提升3D医学图像分割精度。

HResFormer: Hybrid Residual Transformer for Volumetric Medical Image Segmentation

  • 结合2D与3D Transformer,分步学习切片内与跨切片信息。
  • 在多个数据集上显著优于现有方法,最高提升4.2%平均交并比。
  • 适合需要高精度3D解剖结构理解的临床医学图像分析场景。

视觉变换器因能捕捉长程依赖,在医学图像分割中表现优异。针对3D医学图像(如CT)分割,现有方法分为2D和3D两类:2D方法忽略切片内信息,3D方法计算开销大、内存消耗高,限制了对切片内特征的表达。临床实践中,放射科医生先通过轴向视图判断,再结合冠状面形成三维解剖认知。受此启发,本文提出一种新型混合残差变换器(HResFormer),先在2D空间学习细粒度切片内特征,再融合3D信息实现完整三维理解。核心设计包括:(1) 混合局部-全局融合模块(HLGM),自适应融合2D变换器的切片内特征与3D变换器的跨切片信息,兼顾局部细节与全局上下文;(2) 残差学习机制,有效整合内外切片信息以增强三维表征。实验表明,该模型在多个主流医学图像分割基准上优于现有方法,验证了混合架构在3D医学图像分割中的潜力。

原文摘要 · Abstract (English)

Vision Transformer shows great superiority in medical image segmentation due to the ability in learning long-range dependency. For medical image segmentation from 3D data, such as computed tomography (CT), existing methods can be broadly classified into 2D-based and 3D-based methods. One key limitation in 2D-based methods is that the intra-slice information is ignored, while the limitation in 3D-based methods is the high computation cost and memory consumption, resulting in a limited feature representation for inner-slice information. During the clinical examination, radiologists primarily use the axial plane and then routinely review both axial and coronal planes to form a 3D understanding of anatomy. Motivated by this fact, our key insight is to design a hybrid model which can first learn fine-grained inner-slice information and then generate a 3D understanding of anatomy by incorporating 3D information. We present a novel \textbf{H}ybrid \textbf{Res}idual trans\textbf{Former} \textbf{(HResFormer)} for 3D medical image segmentation. Building upon standard 2D and 3D Transformer backbones, HResFormer involves two novel key designs: \textbf{(1)} a \textbf{H}ybrid \textbf{L}ocal-\textbf{G}lobal fusion \textbf{M}odule \textbf{(HLGM)} to effectively and adaptively fuse inner-slice information from 2D Transformer and intra-slice information from 3D volumes for 3D Transformer with local fine-grained and global long-range representation. \textbf{(2)} a residual learning of the hybrid model, which can effectively leverage the inner-slice and intra-slice information for better 3D understanding of anatomy. Experiments show that our HResFormer outperforms prior art on widely-used medical image segmentation benchmarks. This paper sheds light on an important but neglected way to design Transformers for 3D medical image segmentation.

3D分割医学图像视觉变换器残差学习

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