Effect of Training Data Volume on Performance of Convolutional Neural Network Pneumothorax Classifiers.
Journal:
Journal of digital imaging
Published Date:
Mar 3, 2022
Abstract
Large datasets with high-quality labels required to train deep neural networks are challenging to obtain in the radiology domain. This work investigates the effect of training dataset size on the performance of deep learning classifiers, focusing on chest radiograph pneumothorax detection as a proxy visual task in the radiology domain. Two open-source datasets (ChestX-ray14 and CheXpert) comprising 291,454 images were merged and convolutional neural networks trained with stepwise increase in training dataset sizes. Model iterations at each dataset volume were evaluated on an external test set of 525 emergency department chest radiographs. Learning curve analysis was performed to fit the observed AUCs for all models generated. For all three network architectures tested, model AUCs and accuracy increased rapidly from 2 × 10 to 20 × 10 training samples, with more gradual increase until the maximum training dataset size of 291 × 10 images. AUCs for models trained with the maximum tested dataset size of 291 × 10 images were significantly higher than models trained with 20 × 10 images: ResNet-50: AUC = 0.86, AUC = 0.95, p < 0.001; DenseNet-121 AUC = 0.85, AUC = 0.93, p < 0.001; EfficientNet AUC = 0.92, AUC = 0.98, p < 0.001. Our study established learning curves describing the relationship between dataset training size and model performance of deep learning convolutional neural networks applied to a typical radiology binary classification task. These curves suggest a point of diminishing performance returns for increasing training data volumes, which algorithm developers should consider given the high costs of obtaining and labelling radiology data.