Ensemble of CheXNet and VGG-19 Feature Extractor with Random Forest Classifier for Pediatric Pneumonia Detection

Pneumonia, an acute respiratory infection, causes serious breathing hindrance by damaging lung/s. Recovery of pneumonia patients depends on the early diagnosis of the disease and proper treatment. This paper proposes an ensemble method-based pneumonia diagnosis from Chest X-ray images. The deep Convolutional Neural Networks (CNNs)—CheXNet and VGG-19 are trained and used to extract features from given X-ray images. These features are then ensembled for classification. To overcome data irregularity problem, Random Under Sampler (RUS), Random Over Sampler (ROS) and Synthetic Minority Oversampling Technique (SMOTE) are applied on the ensembled feature vector. The ensembled feature vector is then classified using several Machine Learning (ML) classification techniques (Random Forest, Adaptive Boosting, K-Nearest Neighbors). Among these methods, Random Forest got better performance metrics than others on the available standard dataset. Comparison with existing methods shows that the proposed method attains improved classification accuracy, AUC values and outperforms all other models providing 98.93% accurate prediction. The model also exhibits potential generalization capacity when tested on different dataset. Outcomes of this study can be great to use for pneumonia diagnosis from chest X-ray images.

[1]  Nuno M. Fonseca Ferreira,et al.  Classification of Images of Childhood Pneumonia using Convolutional Neural Networks , 2019, BIOIMAGING.

[2]  Vinayak Swarnkar,et al.  Wavelet Augmented Cough Analysis for Rapid Childhood Pneumonia Diagnosis , 2015, IEEE Transactions on Biomedical Engineering.

[3]  Nanditha Krishna,et al.  Feature Extraction and Classification of X-Ray Lung Images Using Haralick Texture Features , 2017 .

[4]  Lixin Zheng,et al.  A transfer learning method with deep residual network for pediatric pneumonia diagnosis , 2020, Comput. Methods Programs Biomed..

[5]  Ronald M. Summers,et al.  ChestX-ray: Hospital-Scale Chest X-ray Database and Benchmarks on Weakly Supervised Classification and Localization of Common Thorax Diseases , 2019, Deep Learning and Convolutional Neural Networks for Medical Imaging and Clinical Informatics.

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

[7]  Omar Mohd Rijal,et al.  Pair-wise discrimination of some lung diseases using chest radiography , 2014, 2014 IEEE REGION 10 SYMPOSIUM.

[8]  Okeke Stephen,et al.  An Efficient Deep Learning Approach to Pneumonia Classification in Healthcare , 2019, Journal of healthcare engineering.

[9]  Igor Rudan,et al.  Epidemiology and etiology of childhood pneumonia. , 2008, Bulletin of the World Health Organization.

[10]  C. Rout,et al.  Role of Gist and PHOG Features in Computer-Aided Diagnosis of Tuberculosis without Segmentation , 2014, PloS one.

[11]  Bidyut Baran Chaudhuri,et al.  Handling data irregularities in classification: Foundations, trends, and future challenges , 2018, Pattern Recognit..

[12]  R. Adegbola,et al.  Childhood pneumonia as a global health priority and the strategic interest of the Bill & Melinda Gates Foundation. , 2012, Clinical infectious diseases : an official publication of the Infectious Diseases Society of America.

[13]  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).

[14]  George R. Thoma,et al.  Visualizing and explaining deep learning predictions for pneumonia detection in pediatric chest radiographs , 2019, Medical Imaging.

[15]  Clarimar José Coelho,et al.  Computer-aided diagnosis in chest radiography for detection of childhood pneumonia , 2008, Int. J. Medical Informatics.

[16]  Andrew Y. Ng,et al.  CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning , 2017, ArXiv.

[17]  Zhijian Song,et al.  Computer-aided detection in chest radiography based on artificial intelligence: a survey , 2018, BioMedical Engineering OnLine.

[18]  Igor Rudan,et al.  Global estimate of the incidence of clinical pneumonia among children under five years of age. , 2004, Bulletin of the World Health Organization.

[19]  S. Bhattacharya,et al.  Pneumonia in South-East Asia Region: Public health perspective , 2012, The Indian journal of medical research.

[20]  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).

[21]  Prateek Chhikara,et al.  Deep Convolutional Neural Network with Transfer Learning for Detecting Pneumonia on Chest X-Rays , 2019, Advances in Intelligent Systems and Computing.