Strategies to improve deep learning-based salivary gland segmentation
Open Access
- 1 December 2020
- journal article
- research article
- Published by Springer Science and Business Media LLC in Radiation Oncology
- Vol. 15 (1), 1-9
- https://doi.org/10.1186/s13014-020-01721-1
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 Sorensen-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.Funding Information
- Varian Medical Systems
This publication has 15 references indexed in Scilit:
- Is accurate contouring of salivary and swallowing structures necessary to spare them in head and neck VMAT plans?Radiotherapy and Oncology, 2018
- Synthetic data augmentation using GAN for improved liver lesion classificationPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2018
- 3D U-Net: Learning Dense Volumetric Segmentation from Sparse AnnotationPublished by Springer Science and Business Media LLC ,2016
- CT-based delineation of organs at risk in the head and neck region: DAHANCA, EORTC, GORTEC, HKNPCSG, NCIC CTG, NCRI, NRG Oncology and TROG consensus guidelinesRadiotherapy and Oncology, 2015
- Fully convolutional networks for semantic segmentationPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2015
- 3D Variation in delineation of head and neck organs at riskRadiation Oncology, 2012
- Radiotherapy Dose–Volume Effects on Salivary Gland FunctionInternational Journal of Radiation Oncology*Biology*Physics, 2010
- Symmetry of the submandibular glands in humans—a postmortem study assessing the linear morphometric parametersOral Surgery, Oral Medicine, Oral Pathology, Oral Radiology, and Endodontology, 2006
- Best practices for convolutional neural networks applied to visual document analysisPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2005
- Neural network ensemblesIEEE Transactions on Pattern Analysis and Machine Intelligence, 1990