Optimal Latent Vector Alignment for Unsupervised Domain Adaptation in Medical Image Segmentation
This paper addresses the domain shift problem for segmentation. As a solution, we propose OLVA, a novel and lightweight unsupervised domain adaptation method based on a Variational Auto-Encoder (VAE) and Optimal Transport (OT) theory. Thanks to the VAE, our model learns a shared cross-domain latent space that follows a normal distribution, which reduces the domain shift. To guarantee valid segmentations, our shared latent space is designed to model the shape rather than the intensity variations. We further rely on an OT loss to match and align the remaining discrepancy between the two domains in the latent space. We demonstrate OLVA's effectiveness for the segmentation of multiple cardiac structures on the public Multi-Modality Whole Heart Segmentation (MM-WHS) dataset, where the source domain consists of annotated 3D MR images and the unlabelled target domain of 3D CTs. Our results show remarkable improvements with an additional margin of 12.5\% dice score over concurrent generative training approaches.
Code (0)
등록된 구현이 없습니다.
Tasks
Domain AdaptationHeart SegmentationImage SegmentationMedical Image SegmentationSegmentationSemantic SegmentationUnsupervised Domain AdaptationvalidSimilar Papers 제목 키워드 기반
Label Alignment Regularization for Distribution Shift
Recent work has highlighted the label alignment property (LAP) in supervised learning, where the vector of all labels in the dataset is mostly in the span of the top few singular vectors of the data matrix. Drawing inspi…
Domain AdaptationRepresentation LearningSentiment AnalysisUnsupervised Domain AdaptationRecursively Conditional Gaussian for Ordinal Unsupervised Domain Adaptation
The unsupervised domain adaptation (UDA) has been widely adopted to alleviate the data scalability issue, while the existing works usually focus on classifying independently discrete labels. However, in many tasks (e.g.,…
Age EstimationDomain AdaptationMedical DiagnosisOrdinal Classification+1Discriminative Feature Alignment: Improving Transferability of Unsupervised Domain Adaptation by Gaussian-guided Latent Alignment
In this study, we focus on the unsupervised domain adaptation problem where an approximate inference model is to be learned from a labeled data domain and expected to generalize well to an unlabeled data domain. The succ…
Data AugmentationDecoderDomain AdaptationDomain Generalization+3One Size does not Fit All: Heterogeneous Latent Space Alignment for Unsupervised Domain Adaptation
Domain shift remains a major obstacle to the reliable deployment of machine learning models in high-stakes environments such as healthcare. While Domain adaptation aims to mitigate these effects, existing approaches suff…
Unsupervised Domain AdaptationMedical Image SegmentationDistributed Vector Representations for Unsupervised Automatic Short Answer Grading
We address the problem of automatic short answer grading, evaluating a collection of approaches inspired by recent advances in distributional text representations. In addition, we propose an unsupervised approach for det…
automatic short answer gradingReading Comprehensiontext similarityWord Embeddings