mlsanityData to 7 Oct 2026

Rethinking Information Loss in Medical Image Segmentation with Various-sized Targets

arXiv 2403.19177 ↗First posted 28 Mar 2024cs.CV cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Medical image segmentation presents the challenge of segmenting various-size targets, demanding the model to effectively capture both local and global information. Despite recent efforts using CNNs and ViTs to predict annotations of different scales, these approaches often struggle to effectively balance the detection of targets across varying sizes. Simply utilizing local information from CNNs and global relationships from ViTs without considering potential significant divergence in latent feature distributions may result in substantial information loss. To address this issue, in this paper, we will introduce a novel Stagger Network (SNet) and argues that a well-designed fusion structure can mitigate the divergence in latent feature distributions between CNNs and ViTs, thereby reducing information loss. Specifically, to emphasize both global dependencies and local focus, we design a Parallel Module to bridge the semantic gap. Meanwhile, we propose the Stagger Module, trying to fuse the selected features that are more semantically similar. An Information Recovery Module is further adopted to recover complementary information back to the network. As a key contribution, we theoretically analyze that the proposed parallel and stagger strategies would lead to less information loss, thus certifying the SNet's rationale. Experimental results clearly proved that the proposed SNet excels comparisons with recent SOTAs in segmenting on the Synapse dataset where targets are in various sizes. Besides, it also demonstrates superiority on the ACDC and the MoNuSeg datasets where targets are with more consistent dimensions.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 26 of its 40 references that are in the map. First posted on arXiv 28 Mar 2024.

Builds on

26 of 40 references.

References · 26

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  4. 2020End-to-End Object Detection with Transformerscited by 7,384
  5. 2021SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformerscited by 2,923
  6. 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,284
  7. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
  8. 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,518
  9. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
  10. 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
  11. 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 954
  12. 2021Do Vision Transformers See Like Convolutional Neural Networks?cited by 671
14 more
  1. 2021Medical Transformer: Gated Axial-Attention for Medical Image Segmentationcited by 319
  2. 2019ResUNet-a: a deep learning framework for semantic segmentation of remotely sensed datacited by 260
  3. 2021TransFuse: Fusing Transformers and CNNs for Medical Image Segmentationcited by 258
  4. 2021UCTransNet: Rethinking the Skip Connections in U-Net from a Channel-wise Perspective with Transformercited by 197
  5. 2021MISSFormer: An Effective Medical Image Segmentation Transformercited by 103
  6. 2022HiFormer: Hierarchical Multi-scale Representations Using Transformers for Medical Image Segmentationcited by 94
  7. 2021TransAttUnet: Multi-level Attention-guided U-Net with Transformer for Medical Image Segmentationcited by 60
  8. 2020Domain Adaptive Relational Reasoning for 3D Multi-Organ Segmentationcited by 48
  9. 2021TransClaw U-Net: Claw U-Net with Transformers for Medical Image Segmentationcited by 41
  10. 2022Class-Aware Adversarial Transformers for Medical Image Segmentationcited by 33
  11. 2019Automated Design of Deep Learning Methods for Biomedical Image Segmentationcited by 27
  12. 2021More than Encoder: Introducing Transformer Decoder to Upsamplecited by 26
  13. 2022Attention Swin U-Net: Cross-Contextual Attention Mechanism for Skin Lesion Segmentationcited by 21
  14. 2022Mind The Gap: Alleviating Local Imbalance for Unsupervised Cross-Modality Medical Image Segmentationcited by 5

Built on by

Not cited yet.