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.

Unsupervised Anomaly Segmentation for Brain Lesions Using Dual Semantic-Manifold Reconstruction.

Unsupervised Anomaly Segmentation using Image-Semantic Cycle Translation

Unsupervised Brain Tumor Segmentation Using a Symmetric-Driven Adversarial Network

Leveraging the Mahalanobis Distance to enhance Unsupervised Brain MRI Anomaly Detection

Autoencoders for Unsupervised Anomaly Segmentation in Brain MR Images: A Comparative Study

Simultaneous reconstruction and segmentation of MRI image by manifold learning

Scale-Space Autoencoders for Unsupervised Anomaly Segmentation in Brain MRI

Three-dimensional deep learning with spatial erasing for unsupervised anomaly segmentation in brain MRI

3D Intracranial Aneurysm Classification and Segmentation via Unsupervised Dual-branch Learning

Constrained unsupervised anomaly segmentation

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

Unsupervised Anomaly Segmentation Via Multilevel Image Reconstruction and Adaptive Attention-Level Transition

Unsupervised Anomaly Detection in 3D Brain MRI using Deep Learning with Multi-Task Brain Age Prediction

ASC-Net: Unsupervised Medical Anomaly Segmentation Using an Adversarial-based Selective Cutting Network

Dual-Path Reconstruction Guided Segmentation Network for Unsupervised Anomaly Detection and Localization.

IterMask2: Iterative Unsupervised Anomaly Segmentation via Spatial and Frequency Masking for Brain Lesions in MRI

Challenging Current Semi-Supervised Anomaly Segmentation Methods for Brain MRI

Unsupervised Region-based Anomaly Detection in Brain MRI with Adversarial Image Inpainting

Weakly Supervised Brain Lesion Segmentation via Attentional Representation Learning

Diffusion Models with Ensembled Structure-Based Anomaly Scoring for Unsupervised Anomaly Detection

Motion-artifact-augmented pseudo-label network for semi-supervised brain tumor segmentation