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.

Select, Attend, and Transfer: Light, Learnable Skip Connections

The Importance of Skip Connections in Biomedical Image Segmentation

Rethinking Skip Connections in Encoder-decoder Networks for Monocular Depth Estimation

Hybrid Skip: A Biologically Inspired Skip Connection for the UNet Architecture

Translated Skip Connections -- Expanding the Receptive Fields of Fully Convolutional Neural Networks

A Quantitative Insight Into the Role of Skip Connections in Deep Neural Networks of Low Complexity: A Case Study Directed at Fluid Flow Modeling

Competition vs. Concatenation in Skip Connections of Fully Convolutional Networks

Narrowing the semantic gaps in U-Net with learnable skip connections: The case of medical image segmentation

Log-DenseNet: How to Sparsify a DenseNet

Towards bi-directional skip connections in encoder-decoder architectures and beyond

Development of Skip Connection in Deep Neural Networks for Computer Vision and Medical Image Analysis: A Survey

Rethinking U-net Skip Connections for Biomedical Image Segmentation

Rethinking Skip Connection with Layer Normalization in Transformers and ResNets

UCTransNet: Rethinking the Skip Connections in U-Net from a Channel-Wise Perspective with Transformer

A Long Skip Connection for Enhanced Color Selectivity in CNN Architectures

N2NSkip: Learning Highly Sparse Networks using Neuron-to-Neuron Skip Connections

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

Faster SCDNet: Real-Time Semantic Segmentation Network with Split Connection and Flexible Dilated Convolution

FusionU-Net: U-Net with Enhanced Skip Connection for Pathology Image Segmentation

USCT-UNet: Rethinking the Semantic Gap in U-Net Network From U-Shaped Skip Connections With Multichannel Fusion Transformer