CLASSIFICATION PERFORMANCE COMPARISONS OF DEEP LEARNING MODELS IN PNEUMONIA DIAGNOSIS USING CHEST X-RAY IMAGES

Abstract

In recent years, the analysis of medical images using deep learning techniques has become an area of increasing popularity. Advances in this area have been particularly evident after the discovery of deep artificial neural network models and achieving more successful performance results than other traditional models. In this study, the performance comparison of different deep learning models used to efficiently diagnose pneumonia on chest x-ray images was performed. The data set used in the study consists of a total of 5840 chest x-ray images of individuals. In order to classify these data, three different deep learning models are used: Convolutional Neural Network, Convolutional Neural Network with Data Augmentation and Transfer Learning. The images in the data set were classified into two categories as pneumonia and healthy people using these three deep learning models. The performances of these three deep learning models used in classification were compared in terms of loss and accuracy. In the comparison of three different deep learning models with two different performance values, 5216 chest x-ray images in the data set were used to train the deep learning model and the remaining 624 were used to test the model. At the end of the study, the most successful performance result was obtained by convolutional neural network model applied with data augmentation technique. According to the best results of this study, this model was able to accurately predict the class of 93.4% of the test data.