An ensemble-based approach by fine-tuning the deep transfer learning models to classify pneumonia from chest X-ray images

by   Sagar Kora Venu, et al.

Pneumonia is caused by viruses, bacteria, or fungi that infect the lungs, which, if not diagnosed, can be fatal and lead to respiratory failure. More than 250,000 individuals in the United States, mainly adults, are diagnosed with pneumonia each year, and 50,000 die from the disease. Chest Radiography (X-ray) is widely used by radiologists to detect pneumonia. It is not uncommon to overlook pneumonia detection for a well-trained radiologist, which triggers the need for improvement in the diagnosis's accuracy. In this work, we propose using transfer learning, which can reduce the neural network's training time and minimize the generalization error. We trained, fine-tuned the state-of-the-art deep learning models such as InceptionResNet, MobileNetV2, Xception, DenseNet201, and ResNet152V2 to classify pneumonia accurately. Later, we created a weighted average ensemble of these models and achieved a test accuracy of 98.46 98.96 their highest levels ever reported in the literature, which can be considered a benchmark for the accurate pneumonia classification.


Transfer Learning with Deep Convolutional Neural Network (CNN) for Pneumonia Detection using Chest X-ray

Pneumonia is a life-threatening disease, which occurs in the lungs cause...

Transfer-Learning-Aware Neuro-Evolution for Diseases Detection in Chest X-Ray Images

The neural network needs excessive costs of time because of the complexi...

Ensemble of Convolutional Neural Networks for Automatic Grading of Diabetic Retinopathy and Macular Edema

In this manuscript, we automate the procedure of grading of diabetic ret...

Deep Learning Applied to Chest X-Rays: Exploiting and Preventing Shortcuts

While deep learning has shown promise in improving the automated diagnos...

Study on Transfer Learning Capabilities for Pneumonia Classification in Chest-X-Rays Image

Over the last year, the severe acute respiratory syndrome coronavirus-2 ...

Pneumonia Detection in Chest X-Rays using Neural Networks

With the advancement in AI, deep learning techniques are widely used to ...