COVID-19 Pneumonia Detection in Chest X-ray Images Using Transfer Learning of Convolutional Neural Networks

The COVID-19 pandemic is the defining global health crisis and the greatest challenge. Since its emergence and has spread rising daily worldwide. The early diagnosis of COVID-19 can improve a chance of survival and can support clinical treatment. An automatically COVID-19 pneumonia detection will support the medical diagnosis to examine the chest X-ray image. For this reason, this work intent to develop the CNN model by the process of transfer learning. The models will be created from the three pre-trained models include Xception, VGG16, and Inception-Resnet-V2 model. Then, these newly proposed models will be applied to detect COVID-19 pneumonia from the X-ray image dataset. The proposed models enhance to diagnose the chest X-ray images of patients who have pneumonia by COVID-19. Since the existing COVID-19 X-ray dataset has a small number of images containing 323 images, this work uses the data augmentation techniques to increase the virtual number of images. The results reveal that the proposed models have performed the classification task for detecting pneumonia. The proposed model achieved an accuracy level of Xception, VGG16, and Inception-Resnet-V2 is 97.19%, 95.42% and 93.87%, respectively. It reveals that the CNN model with the Xception has higher accuracy than VGG16 and Inception-ResNet-V2 model.

[1]  Manop Phankokkruad,et al.  A Comparison of Efficiency Improvement for Long Short-Term Memory Model Using Convolutional Operations and Convolutional Neural Network , 2019, 2019 International Conference on Information and Communications Technology (ICOIACT).

[2]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[3]  Qiang Yang,et al.  A Survey on Transfer Learning , 2010, IEEE Transactions on Knowledge and Data Engineering.

[4]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[5]  Mohammad Faizal Ahmad Fauzi,et al.  Classification of Infection and Fluid Regions in Chest X-Ray Images , 2016, 2016 International Conference on Digital Image Computing: Techniques and Applications (DICTA).

[6]  Salekul Islam,et al.  Real-time Bangla Sign Language Detection using Xception Model with Augmented Dataset , 2019, 2019 IEEE International WIE Conference on Electrical and Computer Engineering (WIECON-ECE).

[7]  The DeepRadiology Team Pneumonia Detection in Chest Radiographs , 2018, ArXiv.

[8]  Konstantin Simonov,et al.  Lung boundary detection for chest X-ray images classification based on GLCM and probabilistic neural networks , 2019, KES.

[9]  Dipayan Biswas,et al.  Transfer Learning Based Plant Diseases Detection Using ResNet50 , 2019, 2019 4th International Conference on Electrical Information and Communication Technology (EICT).

[10]  Fei Yang,et al.  Jointly Optimize Data Augmentation and Network Training: Adversarial Data Augmentation in Human Pose Estimation , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[11]  François Chollet,et al.  Xception: Deep Learning with Depthwise Separable Convolutions , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[12]  Joseph Paul Cohen,et al.  COVID-19 Image Data Collection , 2020, ArXiv.

[13]  Michael H. Goldbaum,et al.  Labeled Optical Coherence Tomography (OCT) and Chest X-Ray Images for Classification , 2018 .

[14]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[15]  D. Juliet,et al.  Transfer Learning with ResNet-50 for Malaria Cell-Image Classification , 2019, 2019 International Conference on Communication and Signal Processing (ICCSP).