Strategies to improve deep learning-based salivary gland segmentation

Abstract

Background Deep learning-based delineation of organs-at-risk for radiotherapy purposes has been investigated to reduce the time-intensiveness and inter-/intra-observer variability associated with manual delineation. We systematically evaluated ways to improve the performance and reliability of deep learning for organ-at-risk segmentation, with the salivary glands as the paradigm. Improving deep learning performance is clinically relevant with applications ranging from the initial contouring process, to on-line adaptive radiotherapy. Methods Various experiments were designed: increasing the amount of training data (1) with original images, (2) with traditional data augmentation and (3) with domain-specific data augmentation; (4) the influence of data quality was tested by comparing training/testing on clinical versus curated contours, (5) the effect of using several custom cost functions was explored, and (6) patient-specific Hounsfield unit windowing was applied during inference; lastly, (7) the effect of model ensembles was analyzed. Model performance was measured with geometric parameters and model reliability with those parameters’ variance. Results A positive effect was observed from increasing the (1) training set size, (2/3) data augmentation, (6) patient-specific Hounsfield unit windowing and (7) model ensembles. The effects of the strategies on performance diminished when the base model performance was already ‘high’. The effect of combining all beneficial strategies was an increase in average Sørensen–Dice coefficient of about 4% and 3% and a decrease in standard deviation of about 1% and 1% for the submandibular and parotid gland, respectively. Conclusions A subset of the strategies that were investigated provided a positive effect on model performance and reliability. The clinical impact of such strategies would be an expected reduction in post-segmentation editing, which facilitates the adoption of deep learning for autonomous automated salivary gland segmentation.

Impact of imperfection in medical imaging data on deep learning‐based segmentation performance: An experimental study using synthesized data

“…23 In order to relate this effect to real clinical data, we can look at one of our earlier studies. 19 For one of the experiments in that study, 24 parotid glands were meticulously curated to a voxel-precise level. The SDC between those segmentations and the original clinical segmentations was 0.78.…”

Apparent Diffusion Coefficient (ADC) Histogram Analysis in Parotid Gland Tumors: Evaluating a Novel Approach for Differentiation between Benign and Malignant Parotid Lesions Based on Full Histogram Distributions

“…Therefore, whole lesion measurements might capture additional information for precise tumor classification and might become a standard procedure in parotid lesion assessment, whereas single-slice measurements may leave out important information by chance. Future developments, like deep learning-based automated algorithms for whole lesion segmentation, may facilitate the implementation in clinical routine [ 37 ].…”

Deep Learning for Automated Elective Lymph Node Level Segmentation for Head and Neck Cancer Radiotherapy

“…All models were trained using 5-fold cross-validation, with a train\test split of 48\12 cases every fold. To minimize the training variation, we used ensemble learning [ 9 , 21 , 22 , 23 ], where the highest cumulated in-class segmentation probability of 5 sequentially trained networks decided the final segmentation map. The training and evaluation times were saved.…”