Convolutional Neural Networks with Transfer Learning for Pneumonia Detection

Orlando Iparraguirre-Villanueva, Victor Guevara-Ponce, Ofelia Roque Paredes, Fernando Sierra-Liñan, Joselyn Zapata-Paulini, Michael Cabanillas-Carbonell

Research output: Contribution to journalArticlepeer-review

17 Scopus citations

Abstract

Pneumonia is a type of acute respiratory infection caused by microbes, and viruses that affect the lungs. Pneumonia is the leading cause of infant mortality in the world, accounting for 81% of deaths in children under five years of age. There are approximately 1.2 million cases of pneumonia in children under five years of age and 180 000 died in 2016. Early detection of pneumonia can help reduce mortality rates. Therefore, this paper presents four convolutional neural network (CNN) models to detect pneumonia from chest X-ray images. CNNs were trained to classify X-ray images into two types: normal and pneumonia, using several convolutional layers. The four models used in this work are pre-trained: VGG16, VGG19, ResNet50, and InceptionV3. The measures that were used for the evaluation of the results are Accuracy, recall, and F1-Score. The models were trained and validated with the dataset. The results showed that the Inceptionv3 model achieved the best performance with 72.9% accuracy, recall 93.7%, and F1-Score 82%. This indicates that CNN models are suitable for detecting pneumonia with high accuracy.

Original languageEnglish
Pages (from-to)544-551
Number of pages8
JournalInternational Journal of Advanced Computer Science and Applications
Volume13
Issue number9
DOIs
StatePublished - 2022

Keywords

  • Convolutional
  • Detection
  • Neural networks
  • Pneumonia
  • Transfer learning

Fingerprint

Dive into the research topics of 'Convolutional Neural Networks with Transfer Learning for Pneumonia Detection'. Together they form a unique fingerprint.

Cite this