Enhanced Target Domain Representation Based Unsupervised Cross-Domain Medical Image Segmentation

被引:0
作者
Liu, Kai [1 ]
Lu, Runuo [1 ]
Zheng, Xiaorou [1 ]
Dong, Shoubin [1 ]
机构
[1] Guangdong Provincial Key Laboratory of Multimodal Big Data Intelligent Analysis, School of Computer Science and Engineering, South China University of Technology, Guangzhou
来源
Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence | 2024年 / 37卷 / 09期
基金
中国国家自然科学基金;
关键词
Adversarial Learning; Consistency Regularization; Medical Images; Semantic Segmentation; Unsupervised Domain Adaptation(UDA);
D O I
10.16451/j.cnki.issn1003-6059.202409001
中图分类号
学科分类号
摘要
Medical images produced by different imaging modality devices exhibit varying degrees of distribution differences. Unsupervised domain adaptation methods typically aim to generalize models trained in the source domain to the unlabeled target domain by minimizing these distribution differences and using shared features between the source and target domains for result prediction. However, they often neglect the private features of the target domain. To address this issue, a method for enhanced target domain representation based unsupervised cross-domain medical image segmentation(TreUCMIS) is proposed in this paper. First, TreUCMIS acquires common features through shared feature learning, and a target domain feature encoder is trained through image reconstruction to capture the complete features of the target domain. Second, unsupervised self-training of the target domain strengthens the shared characteristics of deep and shallow features. Finally, the predicted results obtained from the shared and complete features are aligned, enabling the model to utilize the complete features of the target domain for segmentation and thus improving the generalization in the target domain. Experiments on two medical image segmentation datasets involving bidirectional domain adaptation tasks with CT and MRI(abdominal and cardiac datasets) demonstrate the effectiveness and superiority of TreUCMIS. © 2024 Science Press. All rights reserved.
引用
收藏
页码:755 / 769
页数:14
相关论文
共 37 条
  • [1] YANG C, GUO X Q, CHEN Z, Et al., Source Free Domain Adaptation for Medical Image Segmentation with Fourier Style Mining, Medical Image Analysis, (2022)
  • [2] GUAN H, LIU M X., Domain Adaptation for Medical Image Analysis: A Survey, IEEE Transactions on Biomedical Engineering, 69, 3, pp. 1173-1185, (2022)
  • [3] GHAFOORIAN M, MEHRTASH A, KAPUR T, Et al., Transfer Learning for Domain Adaptation in MRI: Application in Brain Lesion Segmentation, Proc of the 20th International Conference on Medical Image Computing and Computer Assisted Intervention, pp. 516-524, (2017)
  • [4] HOFFMAN J, TZENG E, PARK T, Et al., CyCADA: Cycle-Consistent Adversarial Domain Adaptation, Journal of Machine Learning Research, 80, pp. 1989-1998, (2018)
  • [5] ZENG G D, LERCH T D, SCHMARANZER F, Et al., Semantic Consistent Unsupervised Domain Adaptation for Cross-Modality Medical Image Segmentation, Proc of the 24th International Conference on Medical Image Computing and Computer Assisted Intervention, pp. 201-210, (2021)
  • [6] ZHU J Y, PARK T, ISOLA P, Et al., Unpaired Image-to-Image Translation Using Cycle-Consistent Adversarial Networks, Proc of the IEEE International Conference on Computer Vision, pp. 2242-2251, (2017)
  • [7] DU X Q, LIU Y G., Constraint-Based Unsupervised Domain Adaptation Network for Multi-modality Cardiac Image Segmentation, IEEE Journal of Biomedical and Health Informatics, 26, 1, pp. 67-78, (2022)
  • [8] LIU S L, YIN S Q, QU L H, Et al., Reducing Domain Gap in Frequency and Spatial Domain for Cross-Modality Domain Adaptation on Medical Image Segmentation, Proceedings of the AAAI Conference on Artificial Intelligence, 37, 2, pp. 1719-1727, (2023)
  • [9] MA X, YUAN J K, CHEN Y W, Et al., Attention-Based Cross-Layer Domain Alignment for Unsupervised Domain Adaptation, Neurocomputing, (2022)
  • [10] DOU Q, OUYANG C, CHEN C, Et al., PnP-AdaNet: Plug-and-Play Adversarial Domain Adaptation Network at Unpaired Cross-Modality Cardiac Segmentation, IEEE Access, 7, pp. 99065-99076, (2019)