Effect of Training Data Volume on Performance of Convolutional Neural Network Pneumothorax Classifiers

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_20k = 0.86, AUC_291k = 0.95, p < 0.001; DenseNet-121 AUC_20k = 0.85, AUC_291k = 0.93, p < 0.001; EfficientNet AUC_20k = 0.92, AUC _{291 k} = 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.

Keywords: Convolutional neural network; Dataset size; Deep learning; Pneumothorax; Volume.