Dual teacher model for semi-supervised ABUS tumor segmentation

被引:1
作者
Pan, Pan [1 ]
Chen, Houjin [1 ]
Li, Yanfeng [1 ]
Li, Jiaxin [1 ]
Cheng, Zhanyi [1 ]
Wang, Shu [2 ]
机构
[1] Beijing Jiaotong Univ, Sch Elect & Informat Engn, Beijing 100044, Peoples R China
[2] Peking Univ Peoples Hosp, Beijing 100044, Peoples R China
基金
中国国家自然科学基金;
关键词
Deep learning; Mean teacher; Automated breast ultrasound (ABUS) image; Semi-supervised learning; Pseudo label; ULTRASOUND; NETWORKS; LESIONS;
D O I
10.1016/j.bspc.2023.105514
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Automated Breast Ultrasound (ABUS) has emerged as a crucial modality for the detection of breast tumors in clinical settings. However, obtaining accurate pixel-level annotations of medical images for ABUS tumor seg-mentation is a labor-intensive and expensive process, which hinders the performance of deep learning methods. In this work, we propose a novel semi-supervised tumor segmentation model, namely the Dual-Teacher model, which effectively leverages a small set of labeled images and a large number of unlabeled images for improved segmentation. The Dual-Teacher model extends the original mean teacher (MT) model, MT model only utilizes commonly used consistency criteria, by incorporating an auxiliary teacher branch the Dual-Teacher model captures complementary information from the teacher model and guides the training of the student model. We further introduce a pseudo-label update strategy that combines self-training and consistency constraints to improve the accuracy of pseudo-labels. To optimize the training of the Dual-Teacher model, we design an adaptive loss function that considers the pseudo-labels generated by both teacher models. Our proposed method is evaluated on an ABUS dataset comprising 107 patients with 170 volumes, totaling 3736 2D labeled slices. Experimental results demonstrate that the Dual-Teacher model achieves a Dice similarity coefficient (DSC) of 74.6 %, precision of 75.3 % and recall of 73.9 %, indicating promising performance compared to existing methods.
引用
收藏
页数:11
相关论文
共 53 条
[1]  
Berthelot D, 2019, ADV NEUR IN, V32
[2]   Dilated densely connected U-Net with uncertainty focus loss for 3D ABUS mass segmentation [J].
Cao, Xuyang ;
Chen, Houjin ;
Li, Yanfeng ;
Peng, Yahui ;
Wang, Shu ;
Cheng, Lin .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2021, 209
[3]   Uncertainty Aware Temporal-Ensembling Model for Semi-Supervised ABUS Mass Segmentation [J].
Cao, Xuyang ;
Chen, Houjin ;
Li, Yanfeng ;
Peng, Yahui ;
Wang, Shu ;
Cheng, Lin .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2021, 40 (01) :431-443
[4]  
Cascante-Bonilla P., 2020, CURRICULUM LABELING
[5]   Discriminative Consistent Domain Generation for Semi-supervised Learning [J].
Chen, Jun ;
Zhang, Heye ;
Zhang, Yanping ;
Zhao, Shu ;
Mohiaddin, Raad ;
Wong, Tom ;
Firmin, David ;
Yang, Guang ;
Keegan, Jennifer .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2019, PT II, 2019, 11765 :595-604
[6]  
Chen L.C, 2020, Scene Segmentation, DOI [10.1007/978-3-030, DOI 10.1007/978-3-030]
[7]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851
[8]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[9]  
Codella NCF, 2018, I S BIOMED IMAGING, P168, DOI 10.1109/ISBI.2018.8363547
[10]   Randaugment: Practical automated data augmentation with a reduced search space [J].
Cubuk, Ekin D. ;
Zoph, Barret ;
Shlens, Jonathon ;
Le, Quoc, V .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :3008-3017