Continual Improvement of Nasopharyngeal Carcinoma Segmentation with Less Labeling Effort

Kuo Men,Xinyuan Chen,Ji Zhu,Bining Yang,Ye Zhang,Junlin Yi,Jianrong Dai
DOI: https://doi.org/10.1016/j.ejmp.2020.11.005
2020-01-01
Abstract:PurposeConvolutional neural networks (CNNs) offer a promising approach to automated segmentation. However, labeling contours on a large scale is laborious. Here we propose a method to improve segmentation continually with less labeling effort.MethodsThe cohort included 600 patients with nasopharyngeal carcinoma. The proposed method was comprised of four steps. First, an initial CNN model was trained from scratch to perform segmentation of the clinical target volume. Second, a binary classifier was trained using a secondary CNN to identify samples for which the initial model gave a dice similarity coefficient (DSC) < 0.85. Third, the classifier was used to select such samples from the new coming data. Forth, the final model was fine-tuned from the initial model, using only selected samples.ResultsThe classifier can detect poor segmentation of the model with an accuracy of 92%. The proposed segmentation method improved the DSC from 0.82 to 0.86 while reducing the labeling effort by 45%.ConclusionsThe proposed method reduces the amount of labeled training data and improves segmentation by continually acquiring, fine-tuning, and transferring knowledge over long time spans.
What problem does this paper attempt to address?