A Hybrid Capsule Network for Pneumonia Detection Using Image Augmentation Based on Generative Adversarial Network
- 30 June 2021
- journal article
- research article
- Published by International Information and Engineering Technology Association in Traitement du Signal
- Vol. 38 (3), 619-627
- https://doi.org/10.18280/ts.380309
Abstract
Pneumonia, featured by inflammation of the air sacs in one or both lungs, is usually detected by examining chest X-ray images. This paper probes into the classification models that can distinguish between normal and pneumonia images. As is known, trained networks like AlexNet and GoogleNet are deep network architectures, which are widely adopted to solve many classification problems. They have been adapted to the target datasets, and employed to classify new data generated through transfer learning. However, the classical architectures are not accurate enough for the diagnosis of pneumonia. Therefore, this paper designs a capsule network with high discrimination capability, and trains the network on Kaggle' s online pneumonia dataset, which contains chest X-ray images of many adults and children. The original dataset consists of 1,583 normal images, and 4,273 pneumonia images. Then, two data augmentation approaches were applied to the dataset, and their effects on classification accuracy were compared in details. The model parameters were optimized through five different experiments. The results show that the highest classification accuracy (93.91% even on small images) was achieved by the capsule network, coupled with data augmentation by generative adversarial network (GAN), using optimized parameters. This network outperformed the classical strategies.Keywords
This publication has 18 references indexed in Scilit:
- A Weighted Voting Ensemble Self-Labeled Algorithm for the Detection of Lung Abnormalities from X-RaysAlgorithms, 2019
- A Computer Aided Diagnosis System for Skin Cancer DetectionPublished by Springer Science and Business Media LLC ,2018
- GAN-based synthetic medical image augmentation for increased CNN performance in liver lesion classificationNeurocomputing, 2018
- SegAN: Adversarial Network with Multi-scale L1 Loss for Medical Image SegmentationNeuroinformatics, 2018
- Identifying Medical Diagnoses and Treatable Diseases by Image-Based Deep LearningCell, 2018
- Community-Acquired Pneumonia in AdultsDeutsches Ärzteblatt international, 2017
- Focal Loss for Dense Object DetectionPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2017
- Impact of pneumonia and lung cancer on mortality of women with hypertensionScientific Reports, 2016
- Lung Ultrasound for the Diagnosis of Pneumonia in Children: A Meta-analysisPEDIATRICS, 2015
- Viral pneumoniaThe Lancet, 2011