Automated pneumonia detection on chest X-ray images: A deep learning approach with different optimizers and transfer learning architectures

Abstract Pneumonia is a disease that leads to the death of individuals within a short period since the flow of fluid in the lungs. Hence, initial diagnosis and drugs are very important to avoid the progress of the disease. This paper proposes a novel deep learning approach for automatic detection of pneumonia using deep transfer learning to simplify the detection process with improved accuracy. This work was aimed to preprocess the input chest X-ray images to identify the presence of pneumonia using U-Net architecture based segmentation and classifies the pneumonia as normal and abnormal (Bacteria, viral) using pre-trained on ImageNet dataset models such as ResNet50, InceptionV3, InceptionResNetV2. Besides, to extract the efficient features and improve accuracy of pre-trained models two optimizers, namely, Adam and Stochastic Gradient Descent (SGD) used and its performances are analyzed with batch sizes of 16 and 32. Based on the values obtained, the performances of undertaken pre-trained models are analyzed and compared with other Convolutional Neural Network (CNN) models such as DenseNet-169+SVM, VGG16, RetinaNet + Mask RCNN, VGG16 and Xception, Fully connected RCNN, etc using various measures. From the results observed that the proposed ResNet50 model work achieved 93.06% accuracy, 88.97 % precision rate, 96.78% Recall rate and 92.71% F1-score rate, which than is higher than the other models aforementioned.