Deep learning for head and neck semi-supervised semantic segmentation

被引:3
作者
Luan, Shunyao [1 ,2 ]
Ding, Yi [2 ]
Shao, Jiakang [1 ]
Zou, Bing [3 ]
Yu, Xiao [4 ]
Qin, Nannan [5 ]
Zhu, Benpeng [1 ]
Wei, Wei [2 ]
Xue, Xudong [2 ]
机构
[1] Huazhong Univ Sci & Technol, Sch Integrated Circuits, Lab Optoelect, Wuhan, Peoples R China
[2] Huazhong Univ Sci & Technol, Hubei Canc Hosp, TongJi Med Coll, Dept Radiat Oncol, Wuhan, Hubei, Peoples R China
[3] Nanchang Univ, Affiliated Hosp 2, Dept Oncol, Nanchang, Peoples R China
[4] Univ Sci & Technol China, Affiliated Hosp USTC 1, Dept Radiat Oncol, Div Life Sci & Med, Hefei, Peoples R China
[5] Bengbu Med Coll, Affiliated Hosp 1, Bengbu, Peoples R China
关键词
radiation therapy; semi-supervised semantic segmentation; domain shift; confirmation bias; deep learning; ORGANS; RISK; IMAGES;
D O I
10.1088/1361-6560/ad25c2
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective. Radiation therapy (RT) represents a prevalent therapeutic modality for head and neck (H&N) cancer. A crucial phase in RT planning involves the precise delineation of organs-at-risks (OARs), employing computed tomography (CT) scans. Nevertheless, the manual delineation of OARs is a labor-intensive process, necessitating individual scrutiny of each CT image slice, not to mention that a standard CT scan comprises hundreds of such slices. Furthermore, there is a significant domain shift between different institutions' H&N data, which makes traditional semi-supervised learning strategies susceptible to confirmation bias. Therefore, effectively using unlabeled datasets to support annotated datasets for model training has become a critical issue for preventing domain shift and confirmation bias. Approach. In this work, we proposed an innovative cross-domain orthogon-based-perspective consistency (CD-OPC) strategy within a two-branch collaborative training framework, which compels the two sub-networks to acquire valuable features from unrelated perspectives. More specifically, a novel generative pretext task cross-domain prediction (CDP) was designed for learning inherent properties of CT images. Then this prior knowledge was utilized to promote the independent learning of distinct features by the two sub-networks from identical inputs, thereby enhancing the perceptual capabilities of the sub-networks through orthogon-based pseudo-labeling knowledge transfer. Main results. Our CD-OPC model was trained on H&N datasets from nine different institutions, and validated on the four local intuitions' H&N datasets. Among all datasets CD-OPC achieved more advanced performance than other semi-supervised semantic segmentation algorithms. Significance. The CD-OPC method successfully mitigates domain shift and prevents network collapse. In addition, it enhances the network's perceptual abilities, and generates more reliable predictions, thereby further addressing the confirmation bias issue.
引用
收藏
页数:20
相关论文
共 50 条
  • [21] Class Probability Space Regularization for semi-supervised semantic segmentation
    Yin, Jianjian
    Yan, Shuai
    Chen, Tao
    Chen, Yi
    Yao, Yazhou
    COMPUTER VISION AND IMAGE UNDERSTANDING, 2024, 249
  • [22] Linear semantic transformation for semi-supervised medical image segmentation
    Chen C.
    Chen Y.
    Li X.
    Ning H.
    Xiao R.
    Computers in Biology and Medicine, 2024, 173
  • [23] Semi-supervised Semantic Segmentation Based on Class Balanced Framework
    Yang, Jiacheng
    Dong, Jing
    Yi, Pengfei
    Liu, Rui
    Wang, Ling
    2024 9TH INTERNATIONAL CONFERENCE ON SIGNAL AND IMAGE PROCESSING, ICSIP, 2024, : 795 - 800
  • [24] Subdivided Mask Dispersion Framework for semi-supervised semantic segmentation
    Wang, Yooseung
    Jang, Jaehyuk
    Kim, Changick
    PATTERN RECOGNITION LETTERS, 2024, 179 : 58 - 64
  • [25] Semi-supervised semantic segmentation network for surface crack detection
    Wang, Wenjun
    Su, Chao
    AUTOMATION IN CONSTRUCTION, 2021, 128
  • [26] Weakly Supervised and Semi-Supervised Semantic Segmentation for Optic Disc of Fundus Image
    Lu, Zheng
    Chen, Dali
    SYMMETRY-BASEL, 2020, 12 (01):
  • [27] Twin Pseudo-training for semi-supervised semantic segmentation
    Huang, Huiwen
    Luo, Xiaonan
    Xu, Songhua
    Li, Youxing
    COMPUTERS & GRAPHICS-UK, 2023, 115 : 348 - 358
  • [28] Semantic augmentation by mixing contents for semi-supervised learning
    Sun, Remy
    Masson, Clement
    Henaff, Gilles
    Thome, Nicolas
    Cord, Matthieu
    PATTERN RECOGNITION, 2024, 145
  • [29] Deep Bayesian Active Semi-Supervised Learning
    Rottmann, Matthias
    Kahl, Karsten
    Gottschalk, Hanno
    2018 17TH IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS (ICMLA), 2018, : 158 - 164
  • [30] DEEP SEMI-SUPERVISED LEARNING FOR DOMAIN ADAPTATION
    Chen, Hung-Yu
    Chien, Jen-Tzung
    2015 IEEE INTERNATIONAL WORKSHOP ON MACHINE LEARNING FOR SIGNAL PROCESSING, 2015,