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.

Partition-A-Medical-Image: Extracting Multiple Representative Subregions for Few-Shot Medical Image Segmentation

Partition-A-Medical-Image: Extracting Multiple Representative Sub-regions for Few-shot Medical Image Segmentation

Few-Shot Medical Image Segmentation via Generating Multiple Representative Descriptors

Patch-Free 3D Medical Image Segmentation Driven by Super-Resolution Technique and Self-Supervised Guidance

Super-Resolution Based Patch-Free 3D Medical Image Segmentation with Self-Supervised Guidance

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

Beyond singular prototype: A prototype splitting strategy for few-shot medical image segmentation

Learning De-biased prototypes for few-shot medical image segmentation

Few-Shot 3D Volumetric Segmentation with Multi-Surrogate Fusion

RobustEMD: Domain Robust Matching for Cross-domain Few-shot Medical Image Segmentation

Few-Shot Medical Image Segmentation with Large Kernel Attention

Few-Shot Medical Image Segmentation with High-Fidelity Prototypes

Retrieval-augmented Few-shot Medical Image Segmentation with Foundation Models

SAM-MPA: Applying SAM to Few-shot Medical Image Segmentation using Mask Propagation and Auto-prompting

Learning what and where to segment: A new perspective on medical image few-shot segmentation

Recurrent Mask Refinement for Few-Shot Medical Image Segmentation

Self-Supervision with Superpixels: Training Few-shot Medical Image Segmentation without Annotation

Self-supervised Learning for Few-shot Medical Image Segmentation

PIS-Net: Efficient Medical Image Segmentation Network with Multivariate Downsampling for Point-of-Care

A Mutual Inclusion Mechanism for Precise Boundary Segmentation in Medical Images

Improving 3D Few-Shot Segmentation with Inference-Time Pseudo-Labeling