A Chebyshev Confidence Guided Source-Free Domain Adaptation Framework for Medical Image Segmentation

被引:0
|
作者
Hu, Jiesi [1 ,2 ]
Yang, Yanwu [1 ,2 ]
Guo, Xutao [1 ,2 ]
Ma, Ting [1 ,3 ,4 ]
Wang, Jinghua [5 ]
机构
[1] Harbin Inst Technol Shenzhen, Sch Elect & Informat Engn, Shenzhen 150001, Peoples R China
[2] Peng Cheng Lab, Shenzhen 150001, Peoples R China
[3] Harbin Inst Technol, Peng Cheng Lab, Guangdong Prov Key Lab Aerosp Commun & Networking, Shenzhen 150001, Peoples R China
[4] Harbin Inst Technol, Int Res Inst Artifcial Intelligence, Shenzhen 150001, Peoples R China
[5] Harbin Inst Technol Shenzhen, Sch Comp Sci & Technol, Shenzhen 150001, Peoples R China
基金
中国国家自然科学基金;
关键词
Source-free domain adaptation; image segmentation; self-training; pseudo-label denoising;
D O I
10.1109/JBHI.2024.3406906
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Source-free domain adaptation (SFDA) aims to adapt models trained on a labeled source domain to an unlabeled target domain without access to source data. In medical imaging scenarios, the practical significance of SFDA methods has been emphasized due to data heterogeneity and privacy concerns. Recent state-of-the-art SFDA methods primarily rely on self-training based on pseudo-labels (PLs). Unfortunately, the accuracy of PLs may deteriorate due to domain shift, thus limiting the effectiveness of the adaptation process. To address this issue, we propose a Chebyshev confidence guided SFDA framework to accurately assess the reliability of PLs and generate self-improving PLs for self-training. The Chebyshev confidence is estimated by calculating the probability lower bound of PL confidence, given the prediction and the corresponding uncertainty. Leveraging the Chebyshev confidence, we introduce two confidence-guided denoising methods: direct denoising and prototypical denoising. Additionally, we propose a novel teacher-student joint training scheme (TJTS) that incorporates a confidence weighting module to iteratively improve PLs' accuracy. The TJTS, in collaboration with the denoising methods, effectively prevents the propagation of noise and enhances the accuracy of PLs. Extensive experiments in diverse domain scenarios validate the effectiveness of our proposed framework and establish its superiority over state-of-the-art SFDA methods. Our paper contributes to the field of SFDA by providing a novel approach for precisely estimating the reliability of PLs and a framework for obtaining high-quality PLs, resulting in improved adaptation performance.
引用
收藏
页码:5473 / 5486
页数:14
相关论文
共 50 条
  • [41] Reliable Source Approximation: Source-Free Unsupervised Domain Adaptation for Vestibular Schwannoma MRI Segmentation
    Zeng, Hongye
    Zou, Ke
    Chen, Zhihao
    Zheng, Rui
    Fu, Huazhu
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT X, 2024, 15010 : 622 - 632
  • [42] SSDA: Secure Source-Free Domain Adaptation
    Ahmed, Sabbir
    Al Arafat, Abdullah
    Rizve, Mamshad Nayeem
    Hossain, Rahim
    Guo, Zhishan
    Rakin, Adnan Siraj
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, : 19123 - 19133
  • [43] A Comprehensive Survey on Source-Free Domain Adaptation
    Li, Jingjing
    Yu, Zhiqi
    Du, Zhekai
    Zhu, Lei
    Shen, Heng Tao
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2024, 46 (08) : 5743 - 5762
  • [44] A Comparison of Strategies for Source-Free Domain Adaptation
    Su, Xin
    Zhao, Yiyun
    Bethard, Steven
    PROCEEDINGS OF THE 60TH ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS (ACL 2022), VOL 1: (LONG PAPERS), 2022, : 8352 - 8367
  • [45] SOURCE-FREE DOMAIN ADAPTATION FOR CROSS-SCENE HYPERSPECTRAL IMAGE CLASSIFICATION
    Xu, Zun
    Wei, Wei
    Zhang, Lei
    Nie, Jiangtao
    2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 3576 - 3579
  • [46] Entropy-driven Adversarial Training For Source-free Medical Image Segmentation
    Liqiang, Yuan
    Erdt, Marius
    Wang, Lipo
    Siyal, Mohammed Yakoob
    Cui, Jian
    2023 45TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE & BIOLOGY SOCIETY, EMBC, 2023,
  • [47] Source-Free Domain Adaptation for RGB-D Semantic Segmentation with Vision Transformers
    Rizzoli, Giulia
    Shenaj, Donald
    Zanuttigh, Pietro
    2024 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION WORKSHOPS, WACVW 2024, 2024, : 607 - 616
  • [48] Multi-Source Domain Adaptation for Medical Image Segmentation
    Pei, Chenhao
    Wu, Fuping
    Yang, Mingjing
    Pan, Lin
    Ding, Wangbin
    Dong, Jinwei
    Huang, Liqin
    Zhuang, Xiahai
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2024, 43 (04) : 1640 - 1651
  • [49] OptTTA: Learnable Test-Time Augmentation for Source-Free Medical Image Segmentation Under Domain Shift
    Tomar, Devavrat
    Vray, Guillaume
    Thiran, Jean-Philippe
    Bozorgtabar, Behzad
    INTERNATIONAL CONFERENCE ON MEDICAL IMAGING WITH DEEP LEARNING, VOL 172, 2022, 172 : 1192 - 1217
  • [50] Local-global pseudo-label correction for source-free domain adaptive medical image segmentation
    Ye, Yanyu
    Zhang, Zhenxi
    Tian, Chunna
    Wei, Wei
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 93