Unsupervised augmentation optimization for few-shot medical image segmentation
The augmentation parameters matter to few-shot semantic segmentation since they directly affect the training outcome by feeding the networks with varying perturbated samples. However, searching optimal augmentation parameters for few-shot segmentation models without annotations is a challenge that current methods fail to address. In this paper, we first propose a framework to determine the ``optimal'' parameters without human annotations by solving a distribution-matching problem between the intra-instance and intra-class similarity distribution, with the intra-instance similarity describing the similarity between the original sample of a particular anatomy and its augmented ones and the intra-class similarity representing the similarity between the selected sample and the others in the same class. Extensive experiments demonstrate the superiority of our optimized augmentation in boosting few-shot segmentation models. We greatly improve the top competing method by 1.27\% and 1.11\% on Abd-MRI and Abd-CT datasets, respectively, and even achieve a significant improvement for SSL-ALP on the left kidney by 3.39\% on the Abd-CT dataset.
Code (0)
등록된 구현이 없습니다.
Tasks
AnatomyFew-Shot Semantic SegmentationImage SegmentationMedical Image SegmentationSegmentationSemantic SegmentationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
DenseMP: Unsupervised Dense Pre-training for Few-shot Medical Image Segmentation
Few-shot medical image semantic segmentation is of paramount importance in the domain of medical image analysis. However, existing methodologies grapple with the challenge of data scarcity during the training phase, lead…
Image SegmentationMedical Image AnalysisMedical Image SegmentationSegmentation+1AdLER: Adversarial Training with Label Error Rectification for One-Shot Medical Image Segmentation
Accurate automatic segmentation of medical images typically requires large datasets with high-quality annotations, making it less applicable in clinical settings due to limited training data. One-shot segmentation based …
AnatomyData AugmentationDiversityImage Segmentation+4Adaptively Weighted Data Augmentation Consistency Regularization for Robust Optimization under Concept Shift
Concept shift is a prevailing problem in natural tasks like medical image segmentation where samples usually come from different subpopulations with variant correlations between features and labels. One common type of co…
Data AugmentationImage SegmentationMedical Image SegmentationSegmentation+2Augmentation-Interpolative AutoEncoders for Unsupervised Few-Shot Image Generation
We aim to build image generation models that generalize to new domains from few examples. To this end, we first investigate the generalization properties of classic image generators, and discover that autoencoders genera…
Image GenerationLesionPaste: One-Shot Anomaly Detection for Medical Images
Due to the high cost of manually annotating medical images, especially for large-scale datasets, anomaly detection has been explored through training models with only normal data. Lacking prior knowledge of true anomalie…
Anomaly DetectionMedical Image AnalysisSemi-supervised Anomaly DetectionSupervised Anomaly Detection