Abstract:Medical images are information carriers that visually reflect and record the anatomical structure of the human body, and play an important role in clinical diagnosis, teaching and research, etc. Modern medicine has become increasingly inseparable from the intelligent processing of medical images. In recent years, there have been more and more attempts to apply deep learning theory to medical image segmentation tasks, and it is imperative to explore a simple and efficient deep learning algorithm for medical image segmentation. In this paper, we investigate the segmentation of lung nodule images. We address the above-mentioned problems of medical image segmentation algorithms and conduct research on medical image fusion algorithms based on a hybrid channel-space attention mechanism and medical image segmentation algorithms with a hybrid architecture of Convolutional Neural Networks (CNN) and Visual Transformer. To the problem that medical image segmentation algorithms are difficult to capture long-range feature dependencies, this paper proposes a medical image segmentation model SW-UNet based on a hybrid CNN and Vision Transformer (ViT) framework. Self-attention mechanism and sliding window design of Visual Transformer are used to capture global feature associations and break the perceptual field limitation of convolutional operations due to inductive bias. At the same time, a widened self-attentive vector is used to streamline the number of modules and compress the model size so as to fit the characteristics of a small amount of medical data, which makes the model easy to be overfitted. Experiments on the LUNA16 lung nodule image dataset validate the algorithm and show that the proposed network can achieve efficient medical image segmentation on a lightweight scale. In addition, to validate the migratability of the model, we performed additional validation on other tumor datasets with desirable results. Our research addresses the crucial need for improved medical image segmentation algorithms. By introducing the SW-UNet model, which combines CNN and ViT, we successfully capture long-range feature dependencies and break the perceptual field limitations of traditional convolutional operations. This approach not only enhances the efficiency of medical image segmentation but also maintains model scalability and adaptability to small medical datasets. The positive outcomes on various tumor datasets emphasize the potential migratability and broad applicability of our proposed model in the field of medical image analysis.

AiA-UNet: Attention in Attention for Medical Image Segmentation

Mixed Transformer U-Net for Medical Image Segmentation

UNet based on dynamic convolution decomposition and triplet attention

STA-Unet: Rethink the semantic redundant for Medical Imaging Segmentation

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

Multiresolution Aggregation Transformer UNet Based on Multiscale Input and Coordinate Attention for Medical Image Segmentation

Medical Image Segmentation Based on TransUnet and A2B

TransUNet: Rethinking the U-Net architecture design for medical image segmentation through the lens of transformers

3D TransUNet: Advancing Medical Image Segmentation through Vision Transformers

CSWin-UNet: Transformer UNet with Cross-Shaped Windows for Medical Image Segmentation

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

DMSA-UNet: Dual Multi-Scale Attention makes UNet more strong for medical image segmentation

ConvWin-UNet: UNet-like hierarchical vision Transformer combined with convolution for medical image segmentation.

CAT-Unet: An Enhanced U-Net Architecture with Coordinate Attention and Skip-Neighborhood Attention Transformer for Medical Image Segmentation

ViT-UperNet: a hybrid vision transformer with unified-perceptual-parsing network for medical image segmentation

MaxViT-UNet: Multi-Axis Attention for Medical Image Segmentation

TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation

SW-UNet: a U-Net fusing sliding window transformer block with CNN for segmentation of lung nodules

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

A More Design-Flexible Medical Transformer for Volumetric Image Segmentation.

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