Abstract:Medical image segmentation has witnessed significant advancements with the emergence of deep learning. However, the reliance of most neural network models on a substantial amount of annotated data remains a challenge for medical image segmentation. To address this issue, few-shot segmentation methods based on meta-learning have been employed. Presently, the methods primarily focus on aligning the support set and query set to enhance performance, but this approach hinders further improvement of the model's effectiveness. In this paper, our objective is to propose a few-shot medical segmentation model that acquire comprehensive feature representation capabilities, which will boost segmentation accuracy by capturing both local and long-range features. To achieve this, we introduce a plug-and-play attention module that dynamically enhances both query and support features, thereby improving the representativeness of the extracted features. Our model comprises four key modules: a dual-path feature extractor, an attention module, an adaptive prototype prediction module, and a multi-scale prediction fusion module. Specifically, the dual-path feature extractor acquires multi-scale features by obtaining features of 32{\times}32 size and 64{\times}64 size. The attention module follows the feature extractor and captures local and long-range information. The adaptive prototype prediction module automatically adjusts the anomaly score threshold to predict prototypes, while the multi-scale fusion prediction module integrates prediction masks of various scales to produce the final segmentation result. We conducted experiments on publicly available MRI datasets, namely CHAOS and CMR, and compared our method with other advanced techniques. The results demonstrate that our method achieves state-of-the-art performance.

Efficient Multiscale Spatial Attention 3D Abdominal Multiorgan Segmentation Model

3D Multiple-Contextual ROI-Attention Network for Efficient and Accurate Volumetric Medical Image Segmentation.

Adaptive Decomposition and Shared Weight Volumetric Transformer Blocks for Efficient Patch-Free 3D Medical Image Segmentation.

Spatial Context-Aware Self-Attention Model For Multi-Organ Segmentation

A Lightweight Network for Abdominal Multi-Organ Segmentation Based on Multi-Scale Context Fusion and Dual Self-Attention

DCA: Densely Cross-scale Attention Network for Anatomically-plausible Medical Image Segmentation.

Large-Kernel Attention for 3D Medical Image Segmentation

Target area distillation and section attention segmentation network for accurate 3D medical image segmentation

Multiscale lightweight 3D segmentation algorithm with attention mechanism: Brain tumor image segmentation

[Multi-scale medical image segmentation based on pixel encoding and spatial attention mechanism]

Robust and Efficient Abdominal CT Segmentation Using Shape Constrained Multi-Scale Attention Network

Efficient Context-Aware Network for Abdominal Multi-organ Segmentation

SCANeXt: Enhancing 3D Medical Image Segmentation with Dual Attention Network and Depth-Wise Convolution

A Multiscale Attentional Unet Model for Automatic Segmentation in Medical Ultrasound Images.

W-Net: Dual Supervised Medical Image Segmentation Model with Multi-Dimensional Attention and Cascade Multi-Scale Convolution

Few-Shot Medical Image Segmentation with Large Kernel Attention

w-Net: Dual Supervised Medical Image Segmentation Model with Multi-Dimensional Attention and Cascade Multi-Scale Convolution

Hybrid Attention Mechanism of Feature Fusion for Medical Image Segmentation

Attention Based Multi-Scale Nested Network for Biomedical Image Segmentation

A Lightweight Multi-Scale Multi-Angle Dynamic Interactive Transformer-CNN Fusion Model for 3D Medical Image Segmentation

Cross-layer Connected Network with Adaptive Attention Mechanism for 3D Multi-Organ and Tumor Segmentations from CT