Classification Model Using Transfer Learning for the Detection of Pneumonia in Chest X-Ray Images

Abstract

In the current global context, there has been a significant increase in respiratory system diseases, particularly pneumonia. This disease has a higher incidence of mortality in children under five years old and adults over 60 years old because it leads to complications if not treated in time. This research leverages convolutional neural networks (CNNs) to classify images, specifically to detect the presence of pneumonia. The data processing methodology utilized in this study is CRISP-DM. The dataset consists of 5,856 images of anteroposterior chest X-rays downloaded from the open repository “Kaggle,” divided into 5,216 images for training, 16 for validation, and 624 for testing. Preprocessing involved image augmentation through modifications to the original images, scaling, and batch division in tensor format. A comparative analysis was conducted among the transfer models: DenseNet, VGG19, and ResNet50 version 2. Each transfer model was the header of a CNN with four subsequent layers. The models underwent training, validation, and testing phases. The test’s results showed that DenseNet achieved an accuracy of 0.87, VGG19 achieved 0.86, and ResNet50 achieved 0.91. These results affirm the effectiveness of ResNet50 in image classification, considering that the model’s output is binary, where 0 represents that the patient does not have pneumonia and 1 indicates that the patient has pneumonia.