Unsupervised Domain Adaptation for Medical Image Segmentation by Disentanglement Learning and Self-Training
Qingsong Xie,Yuexiang Li,Nanjun He,Munan Ning,Kai Ma,Guoxing Wang,Yong Lian,Yefeng Zheng
DOI: https://doi.org/10.1109/tmi.2022.3192303
IF: 10.6
2022-01-01
IEEE Transactions on Medical Imaging
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.
engineering, biomedical,imaging science & photographic technology, electrical & electronic,computer science, interdisciplinary applications,radiology, nuclear medicine & medical imaging