Pneumonia detection in chest X-ray images using compound scaled deep learning model

Abstract

Pneumonia is the leading cause of death worldwide for children under 5 years of age. For pneumonia diagnosis, chest X-rays are examined by trained radiologists. However, this process is tedious and time-consuming. Biomedical image diagnosis techniques show great potential in medical image examination. A model for the identification of pneumonia, trained on chest X-ray images, has been proposed in this paper. The compound scaled ResNet50, which is the upscaled version of ResNet50, has been used in this paper. ResNet50 is a multilayer layer convolution neural network having residual blocks. As it was very difficult to obtain a sufficiently large dataset for detection tasks, data augmentation techniques were used to increase the training dataset. Transfer learning is also used while training the models. The proposed model could help in detecting the disease and can assist the radiologists in their clinical decision-making process. The model was evaluated and statistically validated to overfitting and generalization errors. Different scores, such as testing accuracy, F1, recall, precision and AUC score, were computed to check the efficacy of the proposed model. The proposed model attained a test accuracy of 98.14% and an AUC score of 99.71 on the test data from the Guangzhou Women and Children’s Medical Center pneumonia dataset.

Keywords

• pneumonia
• chest x-ray images
• convolution neural network (cnn)
• resnet50
• transfer learning