The impact of training dataset size and ensemble inference strategies on head and neck auto-segmentation

03/30/2023

∙

Convolutional neural networks (CNNs) are increasingly being used to automate segmentation of organs-at-risk in radiotherapy. Since large sets of highly curated data are scarce, we investigated how much data is required to train accurate and robust head and neck auto-segmentation models. For this, an established 3D CNN was trained from scratch with different sized datasets (25-1000 scans) to segment the brainstem, parotid glands and spinal cord in CTs. Additionally, we evaluated multiple ensemble techniques to improve the performance of these models. The segmentations improved with training set size up to 250 scans and the ensemble methods significantly improved performance for all organs. The impact of the ensemble methods was most notable in the smallest datasets, demonstrating their potential for use in cases where large training datasets are difficult to obtain.

READ FULL TEXT

The impact of training dataset size and ensemble inference strategies on head and neck auto-segmentation

Ensemble uncertainty as a criterion for dataset expansion in distinct bone segmentation from upper-body CT images

IEA: Inner Ensemble Average within a convolutional neural network

Multiple Instance Ensembling For Paranasal Anomaly Classification In The Maxillary Sinus

Semi-Supervised Segmentation of Salt Bodies in Seismic Images using an Ensemble of Convolutional Neural Networks

Towards Tumour Graph Learning for Survival Prediction in Head Neck Cancer Patients

Polyp Detection and Segmentation using Mask R-CNN: Does a Deeper Feature Extractor CNN Always Perform Better?

Which CNNs and Training Settings to Choose for Action Unit Detection? A Study Based on a Large-Scale Dataset

The impact of training dataset size and ensemble inference strategies on head and neck auto-segmentation

Related Research

Ensemble uncertainty as a criterion for dataset expansion in distinct bone segmentation from upper-body CT images

IEA: Inner Ensemble Average within a convolutional neural network

Multiple Instance Ensembling For Paranasal Anomaly Classification In The Maxillary Sinus

Semi-Supervised Segmentation of Salt Bodies in Seismic Images using an Ensemble of Convolutional Neural Networks

Towards Tumour Graph Learning for Survival Prediction in Head Neck Cancer Patients

Polyp Detection and Segmentation using Mask R-CNN: Does a Deeper Feature Extractor CNN Always Perform Better?

Which CNNs and Training Settings to Choose for Action Unit Detection? A Study Based on a Large-Scale Dataset