Abstract:Despite the numerous advancements in Convolutional Neural Networks (CNNs) and Transformers, especially in the field of medical image segmentation, two fundamental issues remain. First, the image segmentation task often struggles with effectively modelling global contexts with multi-scales to achieve accurate segmentation results. The second issue concerns the computational burden associated with processing high-resolution medical images and producing fine-grained predictions. Dealing with this level of detail, demands significant computational resources, leading to a computationally intensive process. UNet-like encoder-decoder architectures, which are still the number one widely used architecture in many state-of-the-art applications, struggle to address these complications. While UNet's naive skip connections help to recover spatial information, they fall short in capturing the hierarchical relationships at different scales and the overall context of the image as they combine features from different layers without accounting for their differences, which leads to less accurate segmentation results. We propose an enhanced UNet-like Transformer-based framework with attentive skip connections to tackle these problems: first, instead of simply integrating features extracted from the encoder with the decoder, we added a Transformer-based skip connection module, and second, we optimized the calculations within the skip connection module by employing a merging cross-covariance attention mechanism rather than the conventional self-attention operation, which not only bridges the gaps between multiple levels of semantics and captures more complex dependencies but can also process high-resolution images more efficiently due to its linear complexity in the number of tokens. While retaining the U-shaped encoder-decoder structure, we also replace UNet's CNN layers with hierarchically equivalent Swin Transformer blocks, capturing both global interactions and local dependencies.

Sfe-Transunet: A Transformer-Based U-Net With Skipped Features Enhancer For Medical Image Segmentation

Mixed Transformer U-Net for Medical Image Segmentation

TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation

FTUNet: A Feature-Enhanced Network for Medical Image Segmentation Based on the Combination of U-Shaped Network and Vision Transformer

TransAttUnet: Multi-level Attention-guided U-Net with Transformer for Medical Image Segmentation.

FCTrans UNet: A Hybrid CNN and Transformer Model for Medical Image Segmentations

EG-TransUNet: a transformer-based U-Net with enhanced and guided models for biomedical image segmentation

E-Transunet: Enhanced Transunet for Medical Image Segmentation

Isc-Transunet: Medical Image Segmentation Network Based On The Integration Of Self-Attention And Convolution

Multiscale Transunet + + : Dense Hybrid U-Net with Transformer for Medical Image Segmentation

SeUNet-Trans: A Simple yet Effective UNet-Transformer Model for Medical Image Segmentation

DA-TransUNet: Integrating Spatial and Channel Dual Attention with Transformer U-Net for Medical Image Segmentation

SAttisUNet: UNet-like Swin Transformer with Attentive Skip Connections for Enhanced Medical Image Segmentation

Enhanced Transformer Encoder and Hybrid Cascaded Upsampler for Medical Image Segmentation.

Trans-UNeter: A new Decoder of TransUNet for Medical Image Segmentation.

TSCA-Net: Transformer based spatial-channel attention segmentation network for medical images

DS-TransUNet:Dual Swin Transformer U-Net for Medical Image Segmentation

ATTransUNet: an Enhanced Hybrid Transformer Architecture for Ultrasound and Histopathology Image Segmentation

DS-TransUNet: Dual Swin Transformer U-Net for Medical Image Segmentation

3D TransUNet: Advancing Medical Image Segmentation through Vision Transformers

P-TransUNet: an improved parallel network for medical image segmentation