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

Perspective+ Unet:通过双路径融合与高效非局部注意力增强分割以获得更优感受野

Perspective+ Unet: Enhancing Segmentation with Bi-Path Fusion and Efficient Non-Local Attention for Superior Receptive Fields

Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院);College of Electronics and Information Engineering, Shenzhen University(深圳大学电子与信息工程学院)

Jintong Hu, Siyan Chen, Zhiyi Pan, Sen Zeng, Wenming Yang

AI总结:

针对医学图像分割中CNN难捕长距依赖、全局注意力计算成本高的问题,提出Perspective+ Unet架构,通过双路径卷积、高效非局部Transformer及跨尺度整合策略提升分割效果,在ACDC和Synapse数据集上验证有效。

摘要:

医学图像的精准分割是提取关键临床信息的基础,在提升诊断准确性、制定有效治疗方案以及改善患者预后方面发挥着关键作用。尽管卷积神经网络(CNNs)与非局部注意力方法已在医学图像分割领域取得显著成效,但前者因依赖局部特征难以捕获长距离空间依赖关系,后者尝试通过全局注意力机制解决该问题时,却面临严峻的计算与特征整合挑战。为克服现有医学图像分割方法的局限性,我们提出一种新型架构——Perspective+ Unet。该框架具备三大核心创新点:(1)在编码器阶段引入双路径策略,融合传统卷积与空洞卷积的输出结果。这一设计既保留了局部感受野,又显著扩大了感受野范围,使其能够在保持细节敏感度的同时,更好地理解图像的全局结构。(2)框架集成了名为ENLTB的高效非局部Transformer模块,该模块利用核函数近似实现有效的长距离依赖捕获,且计算复杂度与空间复杂度均为线性。(3)采用空间跨尺度整合器策略,在模型各阶段融合全局依赖关系与局部上下文线索,对不同层级的特征进行精细化调整,以实现全局与局部信息的协调。在ACDC和Synapse数据集上的实验结果,验证了我们所提出的Perspective+ Unet的有效性。代码已收录于补充材料中。

英文摘要:

Precise segmentation of medical images is fundamental for extracting critical clinical information, which plays a pivotal role in enhancing the accuracy of diagnoses, formulating effective treatment plans, and improving patient outcomes. Although Convolutional Neural Networks (CNNs) and non-local attention methods have achieved notable success in medical image segmentation, they either struggle to capture long-range spatial dependencies due to their reliance on local features, or face significant computational and feature integration challenges when attempting to address this issue with global attention mechanisms. To overcome existing limitations in medical image segmentation, we propose a novel architecture, Perspective+ Unet. This framework is characterized by three major innovations: (i) It introduces a dual-pathway strategy at the encoder stage that combines the outcomes of traditional and dilated convolutions. This not only maintains the local receptive field but also significantly expands it, enabling better comprehension of the global structure of images while retaining detail sensitivity. (ii) The framework incorporates an efficient non-local transformer block, named ENLTB, which utilizes kernel function approximation for effective long-range dependency capture with linear computational and spatial complexity. (iii) A Spatial Cross-Scale Integrator strategy is employed to merge global dependencies and local contextual cues across model stages, meticulously refining features from various levels to harmonize global and local information. Experimental results on the ACDC and Synapse datasets demonstrate the effectiveness of our proposed Perspective+ Unet. The code is available in the supplementary material.

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

↑