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.

Unsupervised augmentation optimization for few-shot medical image segmentation

Unsupervised Augmentation Optimization for Few-Shot Medical Image Segmentation

Boosting Unsupervised Contrastive Learning Using Diffusion-Based Data Augmentation from Scratch

Automatic Data Augmentation for 3D Medical Image Segmentation

Data Augmentation Using Learned Transformations for One-Shot Medical Image Segmentation

Data augmentation based on multiple oversampling fusion for medical image segmentation

Boosting Few-Shot Segmentation via Instance-Aware Data Augmentation and Local Consensus Guided Cross Attention

Joint Optimization of Class-Specific Training- and Test-Time Data Augmentation in Segmentation

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

Few-Shot Medical Image Segmentation with Large Kernel Attention

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

Modeling the Probabilistic Distribution of Unlabeled Data forOne-shot Medical Image Segmentation

Anomaly detection-inspired few-shot medical image segmentation through self-supervision with supervoxels

Rethinking Data Augmentation for Single-source Domain Generalization in Medical Image Segmentation

A data augmentation approach that ensures the reliability of foregrounds in medical image segmentation

Data Augmentation-Based Unsupervised Domain Adaptation In Medical Imaging

An Augmentation Strategy for Medical Image Processing Based on Statistical Shape Model and 3D Thin Plate Spline for Deep Learning

Enhancing Weakly Supervised 3D Medical Image Segmentation through Probabilistic-aware Learning

Optimizing Few-Shot Remote Sensing Scene Classification Based on an Improved Data Augmentation Approach

Self-supervised Learning based Transformer and Convolution Hybrid Network for one-shot Organ Segmentation

Self-supervised Learning for Few-shot Medical Image Segmentation