Improving pneumonia diagnosis accuracy via systematic convolutional neural network-based image enhancement

Ziqi Wang, Justin Hall, Rami J. Haddad

Research output: Contribution to book or proceedingConference articlepeer-review

1 Scopus citations

Abstract

Chest X-rays play a significant role in diagnosing pneumonia due to the technology's cost-effectiveness and rapid development times. Detecting pneumonia in chest Xrays is a challenging process that relies heavily upon the availability of trained radiologists and high-quality imagery. Training qualified interpreters require significant resources, while medical imaging remains prone to a wide variety of deficiencies. Therefore, an automated system for pneumonia diagnosis consisting of three phases is proposed. An initial sorting phase consisting of a trained ResNet-18 convolutional neural network separates the dataset according to the interpretive quality of the images, creating a high and low-quality class. The unique image translation capabilities of the CycleGAN network are leveraged in the enhancement phase to translate low-quality images into improved versions. A final ResNet-18 network serves to classify pneumonia in the diagnosis phase. The enhancement system improved mixed quality diagnosis accuracy by 12.1% to 86.7%, with training sets composed of enhanced images achieving an accuracy 15.8% higher than their low-quality counterparts. The system's generalized method for image augmentation successfully mitigates the deficiencies of low-quality data, allowing for a higher accuracy diagnosis than otherwise possible.

Original languageEnglish
Title of host publicationSoutheastCon 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9780738111315
DOIs
StatePublished - Mar 10 2021
Event2021 SoutheastCon, SoutheastCon 2021 - Atlanta, United States
Duration: Mar 10 2021Mar 13 2021

Publication series

NameConference Proceedings - IEEE SOUTHEASTCON
Volume2021-March
ISSN (Print)1091-0050
ISSN (Electronic)1558-058X

Conference

Conference2021 SoutheastCon, SoutheastCon 2021
Country/TerritoryUnited States
CityAtlanta
Period03/10/2103/13/21

Scopus Subject Areas

  • Computer Networks and Communications
  • Software
  • Electrical and Electronic Engineering
  • Control and Systems Engineering
  • Signal Processing

Keywords

  • Cycle-GAN
  • Dataset Augmentation
  • Deep Learning
  • Image Enhancement
  • Pneumonia
  • ResNet-18
  • Systemized Networks
  • Transfer Learning
  • X-ray

Fingerprint

Dive into the research topics of 'Improving pneumonia diagnosis accuracy via systematic convolutional neural network-based image enhancement'. Together they form a unique fingerprint.

Cite this