Abstract:In this work, we address the task of few-shot medical image segmentation (MIS) with a novel proposed framework based on the learning registration to learn segmentation (LRLS) paradigm. To cope with the limitations of lack of authenticity, diversity, and robustness in the existing LRLS frameworks, we propose the better registration better segmentation (BRBS) framework with three main contributions that are experimentally shown to have substantial practical merit. First, we improve the authenticity in the registration-based generation program and propose the knowledge consistency constraint strategy that constrains the registration network to learn according to the domain knowledge. It brings the semantic-aligned and topology-preserved registration, thus allowing the generation program to output new data with great space and style authenticity. Second, we deeply studied the diversity of the generation process and propose the space-style sampling program, which introduces the modeling of the transformation path of style and space change between few atlases and numerous unlabeled images into the generation program. Therefore, the sampling on the transformation paths provides much more diverse space and style features to the generated data effectively improving the diversity. Third, we first highlight the robustness in the learning of segmentation in the LRLS paradigm and propose the mix misalignment regularization, which simulates the misalignment distortion and constrains the network to reduce the fitting degree of misaligned regions. Therefore, it builds regularization for these regions improving the robustness of segmentation learning. Without any bells and whistles, our approach achieves a new state-of-the-art performance in few-shot MIS on two challenging tasks that outperform the existing LRLS-based few-shot methods. We believe that this novel and effective framework will provide a powerful few-shot benchmark for the field of medical image and efficiently reduce the costs of medical image research. All of our code will be made publicly available online.

StyleSeg V2: Towards Robust One-shot Segmentation of Brain Tissue via Optimization-free Registration Error Perception

Style Curriculum Learning for Robust Medical Image Segmentation

Learning Better Registration to Learn Better Few-Shot Medical Image Segmentation: Authenticity, Diversity, and Robustness

Neural Style Transfer Improves 3D Cardiovascular MR Image Segmentation on Inconsistent Data

RegSeg: An End-to-End Network for Multimodal RGB-Thermal Registration and Semantic Segmentation

One registration is worth two segmentations

Symmetric Diffeomorphic Image Registration with Multi-Label Segmentation Masks

One-shot neuroanatomy segmentation through online data augmentation and confidence aware pseudo label

Segmentation Style Discovery: Application to Skin Lesion Images

Online Reflective Learning for Robust Medical Image Segmentation

DeepBrainSeg: Automated Brain Region Segmentation for Micro-Optical Images with a Convolutional Neural Network

Conditional Segmentation in Lieu of Image Registration

BrainSegFounder: Towards 3D foundation models for neuroimage segmentation

MaxStyle: Adversarial Style Composition for Robust Medical Image Segmentation

An Unsupervised Registration Method for Brain Images Based on Contour Guidance

One-Shot Traumatic Brain Segmentation with Adversarial Training and Uncertainty Rectification

Semantic-Aware Registration with Weakly-Supervised Learning

Learning Semi-Supervised Medical Image Segmentation from Spatial Registration

Structure-Aware Stylized Image Synthesis for Robust Medical Image Segmentation

Generalizable Cross-modality Medical Image Segmentation via Style Augmentation and Dual Normalization

Bi-level Learning of Task-Specific Decoders for Joint Registration and One-Shot Medical Image Segmentation