Pneumonia Diagnosis Using Chest X-ray Images and Machine Learning

In this paper, a pneumonia diagnosis system was developed using convolutional neural network (CNN) based feature extraction. InceptionV3 CNN was used to perform feature extraction from chest X-ray images. The extracted feature was used to train three classification algorithm models to predict the cases of pneumonia from a Kaggle dataset. The three models are K-Nearest Neighbor, Neural Network, and Support Vector Machines. Performance evaluation and confusion matrix were presented to represent the sensitivity, accuracy, precision, and specificity of each of the models. Results show that the Neural Network model achieved the highest sensitivity of 84.1%, followed by Support vector machines (83.5%) and K-Nearest Neighbor Algorithm (83.3%). Among all the classification models, Support vector machines model achieved the highest AUC of 93.1%.

[1]  N Ravia Shabnam Parveen,et al.  Detection of Pneumonia in chest X-ray images. , 2011, Journal of X-ray science and technology.

[2]  Aditya Tiwari,et al.  Automatic detection of major lung diseases using Chest Radiographs and classification by feed-forward artificial neural network , 2016, 2016 IEEE 1st International Conference on Power Electronics, Intelligent Control and Energy Systems (ICPEICES).

[3]  S. Balamurali,et al.  Performance Analysis of Classifier Models to Predict Diabetes Mellitus , 2015 .

[4]  E Svedström,et al.  Differentiation of bacterial and viral pneumonia in children , 2002, Thorax.

[5]  Kiwon Seo,et al.  Pneumonia Detection with Weighted Voting Ensemble of CNN Models , 2019, 2019 2nd International Conference on Artificial Intelligence and Big Data (ICAIBD).

[6]  Ankush Mittal,et al.  Pneumonia Detection Using CNN based Feature Extraction , 2019, 2019 IEEE International Conference on Electrical, Computer and Communication Technologies (ICECCT).

[7]  Mehrbakhsh Nilashi,et al.  Accuracy Improvement for Diabetes Disease Classification: A Case on a Public Medical Dataset , 2017 .

[8]  Abhishek Sharma,et al.  Detection of pneumonia clouds in chest X-ray using image processing approach , 2017, 2017 Nirma University International Conference on Engineering (NUiCONE).

[9]  Ahmad Habbie Thias,et al.  Pneumonia Detection with Deep Convolutional Architecture , 2019, 2019 International Conference of Artificial Intelligence and Information Technology (ICAIIT).

[10]  Daniel S. Kermany,et al.  Identifying Medical Diagnoses and Treatable Diseases by Image-Based Deep Learning , 2018, Cell.

[11]  Peter J. Haug,et al.  A Comparison of Classification Algorithms to Automatically Identify Chest X-Ray Reports That Support Pneumonia , 2001, J. Biomed. Informatics.

[12]  Alexandros Karargyris,et al.  Combination of texture and shape features to detect pulmonary abnormalities in digital chest X-rays , 2015, International Journal of Computer Assisted Radiology and Surgery.

[13]  V. B. Surya Prasath,et al.  Distance and Similarity Measures Effect on the Performance of K-Nearest Neighbor Classifier - A Review , 2017, Big Data.

[14]  Gilberto de Melo,et al.  Classification of images and enhancement of performance using parallel algorithm to detection of pneumonia , 2018, 2018 IEEE International Conference on Automation/XXIII Congress of the Chilean Association of Automatic Control (ICA-ACCA).

[15]  Hwanjo Yu,et al.  SVM Tutorial - Classification, Regression and Ranking , 2012, Handbook of Natural Computing.