2024
doi: 10.12720/jait.15.11.1205-1214
Apply a CNN-Based Ensemble Model to Chest-X Ray Image-Based Pneumonia Classification
Ngoc Ha Pham1,2,*
,
Giang Son Tran1
- 1Information, Communication and Technology Laboratory, University of Science and Technology of Hanoi, Hanoi, Vietnam
- 2Information and Communication Technology Department, FPT University, Hanoi, Vietnam
- Manuscript receivedJanuary 15, 2024
- revisedApril 24, 2024
- acceptedJuly 30, 2024
- publishedNovember 8, 2024
Abstract
Pneumonia commonly results from a lung ailment that leads to irritation and harm to the lungs. A chest X-ray is one of the most effective imaging techniques for detecting pneumonia, but diagnosing and treating it can be difficult due to its similarity to other lung conditions. To improve the accuracy of classifying X-ray images, we suggest using an ensemble model in our research that combines deep Convolutional Neural Network (CNN) architectures. The suggested approach classifies the input image as having pneumonia or not by extracting data features using an ensemble of three CNN models. The comparison involves using a single CNN model and a combination of CNN models to evaluate the ensemble architecture. This work evaluates the InceptionResNetV2, DenseNet201, and VGG16 ensemble. The suggested ensemble algorithm provides comparatively positive classification results with an accuracy of almost 95%, outperforming previous ensemble models and improving the average F1-Score by 3% compared to the single model approach.
Keywords
- pneumonia
- chest X-ray
- ensemble learning
- deep learning
- convolutional neural network
How to Cite
Ngoc Ha Pham and Giang Son Tran, "Apply a CNN-Based Ensemble Model to Chest-X Ray Image-Based Pneumonia Classification," Journal of Advances in Information Technology, vol. 15, no. 11, pp. 1205-1214, 2024. https://doi.org/10.12720/jait.15.11.1205-1214
Copyright & License
Copyright © 2024 by the authors. This is an open access article distributed under the Creative Commons Attribution License which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited (CC BY 4.0).
