DCT: Disentanglement-Guided Consolidation and Consistency Training for Semi-Supervised Cross-Site Continual Segmentation

被引:0
作者
Zhang, Jingyang [1 ,2 ,3 ,4 ]
Pei, Jialun [4 ]
Xu, Dunyuan [3 ,4 ]
Jin, Yueming [5 ]
Heng, Pheng-Ann [3 ,4 ]
机构
[1] Southeast Univ, Key Lab New Generat Artificial Intelligence Techno, Nanjing 211189, Peoples R China
[2] Southeast Univ, Sch Comp Sci & Engn, Nanjing 211189, Peoples R China
[3] Chinese Univ Hong Kong, Dept Comp Sci & Engn, Hong Kong, Peoples R China
[4] Chinese Univ Hong Kong, Inst Med Intelligence & XR, Hong Kong, Peoples R China
[5] Natl Univ Singapore, Dept Biomed Engn & Elect & Comp Engn, Singapore 119077, Singapore
关键词
Training; Biomedical imaging; Image segmentation; Data models; Continuing education; Robustness; Semisupervised learning; Metalearning; Video surveillance; Protocols; Continual learning; semi-supervised learning; parameter consolidation; consistency training;
D O I
10.1109/TMI.2024.3469528
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Continual Learning (CL) is recognized to be a storage-efficient and privacy-protecting approach for learning from sequentially-arriving medical sites. However, most existing CL methods assume that each site is fully labeled, which is impractical due to budget and expertise constraint. This paper studies the Semi-Supervised Continual Learning (SSCL) that adopts partially-labeled sites arriving over time, with each site delivering only limited labeled data while the majority remains unlabeled. In this regard, it is challenging to effectively utilize unlabeled data under dynamic cross-site domain gaps, leading to intractable model forgetting on such unlabeled data. To address this problem, we introduce a novel Disentanglement-guided Consolidation and Consistency Training ((DCT)-T-2) framework, which roots in an Online Semi-Supervised representation Disentanglement (OSSD) perspective to excavate content representations of partially labeled data from sites arriving over time. Moreover, these content representations are required to be consolidated for site-invariance and calibrated for style-robustness, in order to alleviate forgetting even in the absence of ground truth. Specifically, for the invariance on previous sites, we retain historical content representations when learning on a new site, via a Content-inspired Parameter Consolidation (CPC) method that prevents altering the model parameters crucial for content preservation. For the robustness against style variation, we develop a Style-induced Consistency Training (SCT) scheme that enforces segmentation consistency over style-related perturbations to recalibrate content encoding. We extensively evaluate our method on fundus and cardiac image segmentation, indicating the advantage over existing SSCL methods for alleviating forgetting on unlabeled data.
引用
收藏
页码:903 / 914
页数:12
相关论文
共 26 条
  • [1] CGT: Consistency Guided Training in Semi-Supervised Learning
    Hasan, Nesreen
    Ghorban, Farzin
    Velten, Joerg
    Kummert, Anton
    PROCEEDINGS OF THE 17TH INTERNATIONAL JOINT CONFERENCE ON COMPUTER VISION, IMAGING AND COMPUTER GRAPHICS THEORY AND APPLICATIONS (VISAPP), VOL 5, 2022, : 55 - 64
  • [2] Consistency-Guided Differential Decoding for Enhancing Semi-Supervised Medical Image Segmentation
    Zeng, Qingjie
    Xie, Yutong
    Lu, Zilin
    Lu, Mengkang
    Zhang, Jingfeng
    Xia, Yong
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2025, 44 (01) : 44 - 56
  • [3] Semi-supervised Left Atrium Segmentation with Mutual Consistency Training
    Wu, Yicheng
    Xu, Minfeng
    Ge, Zongyuan
    Cai, Jianfei
    Zhang, Lei
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT II, 2021, 12902 : 297 - 306
  • [4] FDCT: Fusion-Guided Dual-View Consistency Training for semi-supervised tissue segmentation on MRI
    Chen, Zailiang
    Hou, Yazheng
    Liu, Hui
    Ye, Ziyu
    Zhao, Rongchang
    Shen, Hailan
    COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 160
  • [5] Semi-supervised semantic segmentation with cross teacher training
    Xiao, Hui
    Li, Dong
    Xu, Hao
    Fu, Shuibo
    Yan, Diqun
    Song, Kangkang
    Peng, Chengbin
    NEUROCOMPUTING, 2022, 508 : 36 - 46
  • [6] Querying Labeled for Unlabeled: Cross-Image Semantic Consistency Guided Semi-Supervised Semantic Segmentation
    Wu, Linshan
    Fang, Leyuan
    He, Xingxin
    He, Min
    Ma, Jiayi
    Zhong, Zhun
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (07) : 8827 - 8844
  • [7] Cross Task Temporal Consistency for Semi-supervised Medical Image Segmentation
    Jeevan, Govind
    Pawan, S. J.
    Rajan, Jeny
    MACHINE LEARNING IN MEDICAL IMAGING, MLMI 2022, 2022, 13583 : 140 - 150
  • [8] Adaptive Hierarchical Dual Consistency for Semi-Supervised Left Atrium Segmentation on Cross-Domain Data
    Chen, Jun
    Zhang, Heye
    Mohiaddin, Raad
    Wong, Tom
    Firmin, David
    Keegan, Jennifer
    Yang, Guang
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2022, 41 (02) : 420 - 433
  • [9] Consistency-Guided Meta-learning for Bootstrapping Semi-supervised Medical Image Segmentation
    Wei, Qingyue
    Yu, Lequan
    Li, Xianhang
    Shao, Wei
    Xie, Cihang
    Xing, Lei
    Zhou, Yuyin
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT IV, 2023, 14223 : 183 - 193
  • [10] Semi-supervised Tubular Structure Segmentation with Cross Geometry and Hausdorff Distance Consistency
    Zhu, Ruiyun
    Oda, Masahiro
    Hayashi, Yuichiro
    Kitasaka, Takayuki
    Mori, Kensaku
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT VIII, 2024, 15008 : 612 - 622