mlsanityData to 14 Oct 2026

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

arXiv 2409.05324 ↗First posted 9 Sep 2024cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 32 references that are in the map. First posted on arXiv 9 Sep 2024.

Builds on

20 of 32 references.

References · 20

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  4. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
  5. 2022A ConvNet for the 2020scited by 3,579
  6. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,558
  7. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
  8. 2019UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentationcited by 586
  9. 2021Medical Transformer: Gated Axial-Attention for Medical Image Segmentationcited by 320
  10. 2021TransBTS: Multimodal Brain Tumor Segmentation Using Transformercited by 265
  11. 2022Medical Image Segmentation Review: The success of U-Netcited by 191
  12. 2021Polyp-PVT: Polyp Segmentation with Pyramid Vision Transformerscited by 159
8 more
  1. 2021MISSFormer: An Effective Medical Image Segmentation Transformercited by 103
  2. 2018DUNet: A deformable network for retinal vessel segmentationcited by 90
  3. 2022Stepwise Feature Fusion: Local Guides Globalcited by 56
  4. 2022D-Former: A U-shaped Dilated Transformer for 3D Medical Image Segmentationcited by 35
  5. 2023Multi-scale Hierarchical Vision Transformer with Cascaded Attention Decoding for Medical Image Segmentationcited by 34
  6. 2022Class-Aware Adversarial Transformers for Medical Image Segmentationcited by 33
  7. 2022Contextual Attention Network: Transformer Meets U-Netcited by 33
  8. 2023FusionU-Net: U-Net with Enhanced Skip Connection for Pathology Image Segmentationcited by 2

Built on by

Not cited yet.