DRL-STNet: Unsupervised Domain Adaptation for Cross-modality Medical Image Segmentation via Disentangled Representation Learning
<|reference_start|>DRL-STNet: Unsupervised Domain Adaptation for Cross-modality Medical Image Segmentation via Disentangled Representation Learning: Unsupervised domain adaptation (UDA) is essential for medical image segmentation, especially in cross-modality data scenarios. UDA aims to transfer knowledge from a labeled source domain to an unlabeled target domain, thereby reducing the dependency on extensive manual annotations. This paper presents DRL-STNet, a novel framework for cross-modality medical image segmentation that leverages generative adversarial networks (GANs), disentangled representation learning (DRL), and self-training (ST). Our method leverages DRL within a GAN to translate images from the source to the target modality. Then, the segmentation model is initially trained with these translated images and corresponding source labels and then fine-tuned iteratively using a combination of synthetic and real images with pseudo-labels and real labels. The proposed framework exhibits superior performance in abdominal organ segmentation on the FLARE challenge dataset, surpassing state-of-the-art methods by 11.4% in the Dice similarity coefficient and by 13.1% in the Normalized Surface Dice metric, achieving scores of 74.21% and 80.69%, respectively. The average running time is 41 seconds, and the area under the GPU memory-time curve is 11,292 MB. These results indicate the potential of DRL-STNet for enhancing cross-modality medical image segmentation tasks.<|reference_end|>
@article{lin2024drl-stnet:,
    title={DRL-STNet: Unsupervised Domain Adaptation for Cross-modality Medical
  Image Segmentation via Disentangled Representation Learning},
    author={Hui Lin, Florian Schiffers, Santiago L'opez-Tapia, Neda Tavakoli,
  Daniel Kim, Aggelos K. Katsaggelos},
    journal={arXiv preprint arXiv:2409.18340},
    year={2024},
    archivePrefix={arXiv},
    eprint={2409.18340},
    primaryClass={eess.IV cs.AI cs.CV}
}