Deep Learning Models for Pneumonia Identification and Classification Based on X-Ray Images
Diagnosis based on chest X-rays is widely used and approved for the diagnosis of various diseases such as Pneumonia. Manually screening of theses X-ray images technician or radiologist involves expertise and time consuming. Addressing this, we propose an automated approach for the diagnosis of pneumonia by assisting doctors in spotting infected areas in the X-ray images. We propose a deep Convolutional Neural Network (CNN) model for efficiently detecting the presence of pneumonia in the X-ray images. The proposed CNN is designed with 5 convolution blocks followed by 4 fully connected layers. In order to boost the performance of the model, we incorporate batch normalization, dynamic dropout, learning rate decay, L2 regularization weight decay along with Adam optimizer and binary Cross-Entropy loss function while training the model using back propagating algorithm. The proposed model is validated on two publicly accessible benchmark datasets, and the experimental studies conducted on these datasets indicate that the proposed model is efficient. The suggested CNN architecture with specified hyper parameters allows the model to outperform several existing models by achieving accuracy of 97.73% and 91.17% respectively for binary and multi-class classification tasks of pneumonia disease.