Unsupervised Domain Adaptation for Medical Image Segmentation by Disentanglement Learning and Self-Training

IEEE Trans Med Imaging. 2024 Jan;43(1):4-14. doi: 10.1109/TMI.2022.3192303. Epub 2024 Jan 2.

Abstract

Unsupervised domain adaption (UDA), which aims to enhance the segmentation performance of deep models on unlabeled data, has recently drawn much attention. In this paper, we propose a novel UDA method (namely DLaST) for medical image segmentation via disentanglement learning and self-training. Disentanglement learning factorizes an image into domain-invariant anatomy and domain-specific modality components. To make the best of disentanglement learning, we propose a novel shape constraint to boost the adaptation performance. The self-training strategy further adaptively improves the segmentation performance of the model for the target domain through adversarial learning and pseudo label, which implicitly facilitates feature alignment in the anatomy space. Experimental results demonstrate that the proposed method outperforms the state-of-the-art UDA methods for medical image segmentation on three public datasets, i.e., a cardiac dataset, an abdominal dataset and a brain dataset. The code will be released soon.

MeSH terms

  • Brain* / diagnostic imaging
  • Heart*
  • Image Processing, Computer-Assisted