Boundary Contrast Domain Adaptation for Cross-modality Medical Image Segmentation

Haoran Zhang,Xi Lin,Suxian Xiang,Chenxi Huang,Lvqing Yang,Yan Wang
DOI: https://doi.org/10.1109/icme57554.2024.10688182
2024-01-01
Abstract:Unsupervised domain adaptation (UDA) methods have achieved significant success in the cross-modality medical image segmentation tasks. However, due to the inherent properties of medical images, i.e., the large distribution gap between different modalities and the low intensity contrast between different categories of organ structures, it is challenging for current UDA methods to perform well on the boundary regions. In this paper, we propose a boundary contrast domain adaptation framework for cross-modality medical image segmentation. Concretely, we consider the same category prototypes as positive samples to reduce category-level distribution differences between domains, and boundary features as negative samples to improve the discriminative ability of ambiguous boundary regions. And we construct mixed samples by bidirectional cross-domain cutmix for self-training to further reduce the domain gap. Moreover, we dynamically assign weights to different parts of the pseudo-labels to prevent model degradation. Experimental results show our proposed method outperforms the state-of-the-art methods.
What problem does this paper attempt to address?