arXiv:2409.05324cs.CV2024-09被引 2

改进医学图像器官分割的U型网络,提升特征融合与注意力效果。

An Effective UNet Using Feature Interaction and Fusion for Organ Segmentation in Medical Image

  • 引入通道空间交互模块增强跳接特征质量
  • 多尺度融合与注意力机制使平均Dice达86.05%和92.58%
  • 轻量设计适合临床部署,参数仅8691万

当前预训练编码器在医学图像分割中广泛应用,因其具备提取丰富通用特征的能力。然而,现有方法未能充分挖掘这些特征,限制了分割性能。本文提出一种新型U型模型,包含三个即插即用模块:通道空间交互模块通过建模编码器与解码器间的跨阶段交互,提升跳接特征质量;基于通道注意力的模块结合挤压-激励机制与卷积层,强化关键特征并抑制无关信息;多级融合模块聚合多尺度解码特征,提升最终预测的空间细节与一致性。在Synapse多器官分割数据集和ACDC心脏诊断挑战数据集上的实验表明,该模型优于现有最先进方法,平均Dice分数分别达到86.05%和92.58%,提升1.15%和0.26%。此外,模型在精度与计算复杂度间取得平衡,仅需8691万参数和23.26吉弗洛普运算。

原文摘要 · Abstract (English)

Nowadays, pre-trained encoders are widely used in medical image segmentation due to their strong capability in extracting rich and generalized feature representations. However, existing methods often fail to fully leverage these features, limiting segmentation performance. In this work, a novel U-shaped model is proposed to address the above issue, including three plug-and-play modules. A channel spatial interaction module is introduced to improve the quality of skip connection features by modeling inter-stage interactions between the encoder and decoder. A channel attention-based module integrating squeeze-and-excitation mechanisms with convolutional layers is employed in the decoder blocks to strengthen the representation of critical features while suppressing irrelevant ones. A multi-level fusion module is designed to aggregate multi-scale decoder features, improving spatial detail and consistency in the final prediction. Comprehensive experiments on the synapse multi-organ segmentation dataset and automated cardiac diagnosis challenge dataset demonstrate that the proposed model outperforms existing state-of-the-art methods, achieving the highest average Dice score of 86.05% and 92.58%, yielding improvements of 1.15% and 0.26%, respectively. In addition, the proposed model provides a balance between accuracy and computational complexity, with only 86.91 million parameters and 23.26 giga floating-point operations.

医学图像U型网络特征融合分割

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