Current browse context:
eess.IV
Change to browse by:
References & Citations
Electrical Engineering and Systems Science > Image and Video Processing
Title: Unsupervised Domain Adaptation for Cardiac Segmentation: Towards Structure Mutual Information Maximization
(Submitted on 20 Apr 2022 (v1), last revised 27 May 2023 (this version, v3))
Abstract: Unsupervised domain adaptation approaches have recently succeeded in various medical image segmentation tasks. The reported works often tackle the domain shift problem by aligning the domain-invariant features and minimizing the domain-specific discrepancies. That strategy works well when the difference between a specific domain and between different domains is slight. However, the generalization ability of these models on diverse imaging modalities remains a significant challenge. This paper introduces UDA-VAE++, an unsupervised domain adaptation framework for cardiac segmentation with a compact loss function lower bound. To estimate this new lower bound, we develop a novel Structure Mutual Information Estimation (SMIE) block with a global estimator, a local estimator, and a prior information matching estimator to maximize the mutual information between the reconstruction and segmentation tasks. Specifically, we design a novel sequential reparameterization scheme that enables information flow and variance correction from the low-resolution latent space to the high-resolution latent space. Comprehensive experiments on benchmark cardiac segmentation datasets demonstrate that our model outperforms previous state-of-the-art qualitatively and quantitatively. The code is available at https://github.com/LOUEY233/Toward-Mutual-Information}{this https URL
Submission history
From: Changjie Lu [view email][v1] Wed, 20 Apr 2022 09:10:18 GMT (5755kb,D)
[v2] Fri, 24 Jun 2022 15:07:09 GMT (5757kb,D)
[v3] Sat, 27 May 2023 15:42:45 GMT (5741kb,D)
Link back to: arXiv, form interface, contact.