Three-stage ensemble of image net pre-trained networks for pneumonia detection

Uddamvathanak, Rom, Chawla, Shailey, and Feng, Yang (2019) Three-stage ensemble of image net pre-trained networks for pneumonia detection. International Journal of Electrical, Electronics and Data Communication, 7 (7). pp. 24-28.

[img]
Preview
PDF (Published version) - Published Version
Available under License Creative Commons Attribution.

Download (401kB) | Preview
View at Publisher Website: http://www.iraj.in/journal/journal_file/...
 
64


Abstract

Focusing on detection of pneumenia disease in the Chest X-Ray images, this paper proposes a three-stage ensemble methodology utilizing multiple pre-trained Convolutional Neural Networks (CNNs). In the first-stage ensemble, k subsets of training data are firstly randomly generated, each of which is then used to retrain a pre-trained CNN to produce k CNN models for the ensemble in the first stage. In the second-stage ensemble, multiple ensemble CNN models based on multiple pre-trained CNNs are integrated to reduce variance and improve the performance of the prediction. The third-stage ensemble is based on image augmentation, i.e., the original set of images are augmented to generate a few sets of additional images, after which each set of images are input to the ensemble models from the first two stages, and the outputs based multiple sets of images are then integrated. In integrating outputs in each stage, four ensemble techniques are introduced including averaging, feed forward neural network-based, decision tree-based, and majority voting. Thorough experiments were conducted on Chest X-Ray images from a Kaggle challenge, and the results showed the effectiveness of the proposed three-stage ensemble method in detecting pneumonia disease in the images.

Item ID: 61456
Item Type: Article (Other)
ISSN: 2321-2950
Keywords: Ensemble, Convolutional Neural Network, Chest X-Ray, Pneumonia, Imagenet Pre-Trained Model, Pneumonia Detection.
Related URLs:
Copyright Information: This work is licensed under a Creative Commons Attribution 4.0 International License.
Date Deposited: 07 Nov 2022 01:18
FoR Codes: 08 INFORMATION AND COMPUTING SCIENCES > 0801 Artificial Intelligence and Image Processing > 080106 Image Processing @ 50%
08 INFORMATION AND COMPUTING SCIENCES > 0801 Artificial Intelligence and Image Processing > 080108 Neural, Evolutionary and Fuzzy Computation @ 50%
SEO Codes: 89 INFORMATION AND COMMUNICATION SERVICES > 8902 Computer Software and Services > 890299 Computer Software and Services not elsewhere classified @ 100%
Downloads: Total: 64
Last 12 Months: 9
More Statistics

Actions (Repository Staff Only)

Item Control Page Item Control Page