# One Size does not Fit All: Heterogeneous Latent Space Alignment for Unsupervised Domain Adaptation

arXiv: https://arxiv.org/abs/2606.21415 · HTML: https://mlsanity.com/paper/2606.21415/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 0 of its 37 references that are in the map.
- First posted on arXiv 19 Jun 2026.

## Abstract (arXiv)

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 suffer from limited expressiveness of latent representations and a reliance on handcrafted, static augmentations. In this work, we address these limitations by proposing a novel deep learning architecture for Unsupervised Domain Adaptation (UDA), specifically optimized for medical image segmentation. Our framework, ADualVUOT, integrates a dual-encoder Variational Autoencoder (VAE) with Continuous Normalizing Flows (CNFs) to increase modeling flexibility and posterior expressiveness. To achieve domain alignment, we leverage Unbalanced Optimal Transport (UOT) through the Gaussian-Gromov-Wasserstein (GGW) distance, which handles structural and topological discrepancies between domains. Furthermore, we incorporate an adversarial augmentation scheme to synthesize worst-case compositions, thus enhancing model robustness. Extensive experiments on medical imaging benchmarks show significant gains over prior OT-based approaches.