Abstract:Brain structure segmentation in Magnetic Resonance image (MRI) plays an important role in the diagnosis of various neuropsychiatric diseases. Recently, the high cost of manual annotation has motivated research in medical image analysis to focus on unsupervised learning methods. As a promising solution, unsupervised domain adaptation (UDA) has received increasing attention in various medical image segmentation tasks. In this work, we propose a novel UDA framework for brain structure segmentation from MRI volumes. Specifically, we design a self-training-based network architecture to learn cross-domain representations, which is achieved by transferring essential knowledge from a well-trained cumbersome teacher network to a non-trained compact student network using distillation learning technology. The pseudo-labels generated by the pre-trained student model are used to supervise the domain adaptation process of the teacher model. Furthermore, we develop a mutual information maximization alignment module to learn domain-invariant features and cross-domain segmentation knowledge, in which contrastive learning techniques are utilized to maximize the correlation between samples. Finally, we build a unified distillation framework by integrating mutual information distillation and output map distillation, correcting pseudo-labels during the self-training stage, and continuously refining the final predictions of compact networks. Extensive experiments conducted on two datasets with vastly different styles and settings demonstrate that our method is effective in improving segmentation performance on unlabeled target images, and outperforms the state-of-the-art domain adaptation approaches by a large margin. The source code is available at https://github.com/huqian999/UDA-MIMA.

ASC: Appearance and Structure Consistency for Unsupervised Domain Adaptation in Fetal Brain MRI Segmentation

Unsupervised Domain Adaptation for Brain Structure Segmentation Via Mutual Information Maximization Alignment

Unsupervised Domain Adaptation for Anatomical Structure Detection in Ultrasound Images

Uncertainty-aware domain alignment for anatomical structure segmentation

$M^3$-UDA: A New Benchmark for Unsupervised Domain Adaptive Fetal Cardiac Structure Detection

Unsupervised Cross-modality Adaptation via Dual Structural-Oriented Guidance for 3D Medical Image Segmentation

A 3-D Anatomy-Guided Self-Training Segmentation Framework for Unpaired Cross-Modality Medical Image Segmentation

Diffuse-UDA: Addressing Unsupervised Domain Adaptation in Medical Image Segmentation with Appearance and Structure Aligned Diffusion Models

Tissue Segmentation of Thick-Slice Fetal Brain MR Scans with Guidance from High-Quality Isotropic Volumes

Unsupervised Domain Adaptation for Neuron Membrane Segmentation based on Structural Features

Structure-Enhanced Unsupervised Domain Adaptation for CT Whole-Brain Segmentation

Disentangle Domain Features for Cross-Modality Cardiac Image Segmentation.

SELF-SEMANTIC CONTOUR ADAPTATION FOR CROSS MODALITY BRAIN TUMOR SEGMENTATION

A Structure-aware Framework of Unsupervised Cross-Modality Domain Adaptation via Frequency and Spatial Knowledge Distillation

Structure preservation constraints for unsupervised domain adaptation intracranial vessel segmentation

Deep Symmetric Adaptation Network for Cross-Modality Medical Image Segmentation

Unsupervised Domain Adaptation for Cross-Modality Cardiac Image Segmentation Based on Contrastive Image Synthesis

Synthetic magnetic resonance images for domain adaptation: Application to fetal brain tissue segmentation

Learning Spatiotemporal Probabilistic Atlas of Fetal Brains with Anatomically Constrained Registration Network

AtlasSeg: Atlas Prior Guided Dual-U-Net for Cortical Segmentation in Fetal Brain MRI

SMC-UDA: Structure-Modal Constraint for Unsupervised Cross-Domain Renal Segmentation