机构:[1]Univ Elect Sci & Technol China, Sch Mech & Elect Engn, Chengdu, Peoples R China[2]Sichuan Univ, West China Hosp, Dept Radiat Oncol, Chengdu, Peoples R China四川大学华西医院[3]Tongji Univ, Coll Elect & Informat Technol, Shanghai, Peoples R China[4]SenseTime Res, Shanghai, Peoples R China[5]Sichuan Univ, West China Hosp SenseTime Joint Lab, West China Biomed Big Data Ctr, West China Hosp, Chengdu, Peoples R China四川大学华西医院[6]Univ Elect Sci & Technol China, Sichuan Canc Hosp & Inst, Dept Radiat Oncol, Chengdu, Peoples R China四川省肿瘤医院
Gross Target Volume (GTV) segmentation plays an irreplaceable role in radiotherapy planning for Nasopharyngeal Carcinoma (NPC). Despite that Convolutional Neural Networks (CNN) have achieved good performance for this task, they rely on a large set of labeled images for training, which is expensive and time-consuming to acquire. In this paper, we propose a novel framework with Uncertainty Rectified Pyramid Consistency (URPC) regularization for semi-supervised NPC GTV segmentation. Concretely, we extend a backbone segmentation network to produce pyramid predictions at different scales. The pyramid predictions network (PPNet) is supervised by the ground truth of labeled images and a multi-scale consistency loss for unlabeled images, motivated by the fact that prediction at different scales for the same input should be similar and consistent. However, due to the different resolution of these predictions, encouraging them to be consistent at each pixel directly has low robustness and may lose some fine details. To address this problem, we further design a novel uncertainty rectifying module to enable the framework to gradually learn from meaningful and reliable consensual regions at different scales. Experimental results on a dataset with 258 NPC MR images showed that with only 10% or 20% images labeled, our method largely improved the segmentation performance by leveraging the unlabeled images, and it also outperformed five state-of-the-art semi-supervised segmentation methods. Moreover, when only 50% labeled images, URPC achieved an average Dice score of 82.74% that was close to fully supervised learning. Code is available at: https://github.com/HiLab-git/SSL4MIS.
基金:
National Natural Science Foundations of ChinaNational Natural Science Foundation of China (NSFC) [81771921, 61901084]; key research and development project of Sichuan province, China [20ZDYF2817]
语种:
外文
被引次数:
WOS:
第一作者:
第一作者机构:[1]Univ Elect Sci & Technol China, Sch Mech & Elect Engn, Chengdu, Peoples R China
通讯作者:
推荐引用方式(GB/T 7714):
Luo Xiangde,Liao Wenjun,Chen Jieneng,et al.Efficient Semi-supervised Gross Target Volume of Nasopharyngeal Carcinoma Segmentation via Uncertainty Rectified Pyramid Consistency[J].MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT II.2021,12902:318-329.doi:10.1007/978-3-030-87196-3_30.