Disentanglement enables cross-domain Hippocampus Segmentation
- URL: http://arxiv.org/abs/2201.05650v1
- Date: Fri, 14 Jan 2022 19:49:53 GMT
- Title: Disentanglement enables cross-domain Hippocampus Segmentation
- Authors: John Kalkhof, Camila Gonz\'alez, Anirban Mukhopadhyay
- Abstract summary: Limited amount of labelled training data are a common problem in medical imaging.
This makes it difficult to train a well-generalised model and often leads to failure in unknown domains.
We address this issue by disentangling a T1-weighted MRI image into its content and domain.
- Score: 1.2020488155038649
- License: http://creativecommons.org/licenses/by-nc-sa/4.0/
- Abstract: Limited amount of labelled training data are a common problem in medical
imaging. This makes it difficult to train a well-generalised model and
therefore often leads to failure in unknown domains. Hippocampus segmentation
from magnetic resonance imaging (MRI) scans is critical for the diagnosis and
treatment of neuropsychatric disorders. Domain differences in contrast or shape
can significantly affect segmentation. We address this issue by disentangling a
T1-weighted MRI image into its content and domain. This separation enables us
to perform a domain transfer and thus convert data from new sources into the
training domain. This step thus simplifies the segmentation problem, resulting
in higher quality segmentations. We achieve the disentanglement with the
proposed novel methodology 'Content Domain Disentanglement GAN', and we propose
to retrain the UNet on the transformed outputs to deal with GAN-specific
artefacts. With these changes, we are able to improve performance on unseen
domains by 6-13% and outperform state-of-the-art domain transfer methods.
Related papers
- SMC-UDA: Structure-Modal Constraint for Unsupervised Cross-Domain Renal
Segmentation [100.86339246424541]
We propose a novel Structure-Modal Constrained (SMC) UDA framework based on a discriminative paradigm and introduce edge structure as a bridge between domains.
With the structure-constrained self-learning and progressive ROI, our methods segment the kidney by locating the 3D spatial structure of the edge.
experiments show that our proposed SMC-UDA has a strong generalization and outperforms generative UDA methods.
arXiv Detail & Related papers (2023-06-14T02:57:23Z) - Domain Re-Modulation for Few-Shot Generative Domain Adaptation [71.47730150327818]
Generative Domain Adaptation (GDA) involves transferring a pre-trained generator from one domain to a new domain using only a few reference images.
Inspired by the way human brains acquire knowledge in new domains, we present an innovative generator structure called Domain Re-Modulation (DoRM)
DoRM not only meets the criteria of high quality, large synthesis diversity, and cross-domain consistency, but also incorporates memory and domain association.
arXiv Detail & Related papers (2023-02-06T03:55:35Z) - MSCDA: Multi-level Semantic-guided Contrast Improves Unsupervised Domain
Adaptation for Breast MRI Segmentation in Small Datasets [5.272836235045653]
We propose a novel Multi-level Semantic-guided Contrastive Domain Adaptation framework.
Our approach incorporates self-training with contrastive learning to align feature representations between domains.
In particular, we extend the contrastive loss by incorporating pixel-to-pixel, pixel-to-centroid, and centroid-to-centroid contrasts.
arXiv Detail & Related papers (2023-01-04T19:16:55Z) - Reliable Joint Segmentation of Retinal Edema Lesions in OCT Images [55.83984261827332]
In this paper, we propose a novel reliable multi-scale wavelet-enhanced transformer network.
We develop a novel segmentation backbone that integrates a wavelet-enhanced feature extractor network and a multi-scale transformer module.
Our proposed method achieves better segmentation accuracy with a high degree of reliability as compared to other state-of-the-art segmentation approaches.
arXiv Detail & Related papers (2022-12-01T07:32:56Z) - Reducing Domain Gap in Frequency and Spatial domain for Cross-modality
Domain Adaptation on Medical Image Segmentation [5.371816551086118]
Unsupervised domain adaptation (UDA) aims to learn a model trained on source domain and performs well on unlabeled target domain.
We propose a simple yet effective UDA method based on frequency and spatial domain transfer uner multi-teacher distillation framework.
Our proposed method achieves superior performance compared to state-of-the-art methods.
arXiv Detail & Related papers (2022-11-28T11:35:39Z) - Causality-inspired Single-source Domain Generalization for Medical Image
Segmentation [12.697945585457441]
We propose a simple data augmentation approach to expose a segmentation model to synthesized domain-shifted training examples.
Specifically, 1) to make the deep model robust to discrepancies in image intensities and textures, we employ a family of randomly-weighted shallow networks.
We remove spurious correlations among objects in an image that might be taken by the network as domain-specific clues for making predictions, and they may break on unseen domains.
arXiv Detail & Related papers (2021-11-24T14:45:17Z) - Unsupervised Domain Adaptation with Semantic Consistency across
Heterogeneous Modalities for MRI Prostate Lesion Segmentation [19.126306953075275]
We introduce two new loss functions that promote semantic consistency.
In particular, we address the challenge of enhancing performance on VERDICT-MRI, an advanced diffusion-weighted imaging technique.
arXiv Detail & Related papers (2021-09-19T17:33:26Z) - Cross-Modality Brain Tumor Segmentation via Bidirectional
Global-to-Local Unsupervised Domain Adaptation [61.01704175938995]
In this paper, we propose a novel Bidirectional Global-to-Local (BiGL) adaptation framework under a UDA scheme.
Specifically, a bidirectional image synthesis and segmentation module is proposed to segment the brain tumor.
The proposed method outperforms several state-of-the-art unsupervised domain adaptation methods by a large margin.
arXiv Detail & Related papers (2021-05-17T10:11:45Z) - Harnessing Uncertainty in Domain Adaptation for MRI Prostate Lesion
Segmentation [15.919637739630353]
We consider translating from mp-MRI to VERDICT, a richer MRI modality involving an acquisition optimized protocol for cancer characterization.
Our results show that this allows us to extract systematically better image representations for the target domain, when used in tandem with both simple, CycleGAN-based baselines.
arXiv Detail & Related papers (2020-10-14T21:30:27Z) - Shape-aware Meta-learning for Generalizing Prostate MRI Segmentation to
Unseen Domains [68.73614619875814]
We present a novel shape-aware meta-learning scheme to improve the model generalization in prostate MRI segmentation.
Experimental results show that our approach outperforms many state-of-the-art generalization methods consistently across all six settings of unseen domains.
arXiv Detail & Related papers (2020-07-04T07:56:02Z) - Unsupervised Bidirectional Cross-Modality Adaptation via Deeply
Synergistic Image and Feature Alignment for Medical Image Segmentation [73.84166499988443]
We present a novel unsupervised domain adaptation framework, named as Synergistic Image and Feature Alignment (SIFA)
Our proposed SIFA conducts synergistic alignment of domains from both image and feature perspectives.
Experimental results on two different tasks demonstrate that our SIFA method is effective in improving segmentation performance on unlabeled target images.
arXiv Detail & Related papers (2020-02-06T13:49:47Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.