An AI-based System for Pneumonia Detection in Chest X-Rays Using Deep Learning Models
DOI:
https://doi.org/10.70112/ajeat-2025.14.1.4278Keywords:
Pneumonia Detection, Deep Learning, Chest X-rays, Image Enhancement, Synthetic Data GenerationAbstract
Pneumonia remains a significant global health concern, necessitating rapid and accurate diagnostic tools. This study presents an AI-based system for pneumonia detection in chest X-rays using deep learning models. The research emphasizes enhancing diagnostic accuracy through advanced image processing techniques while maintaining clinical applicability. Deep learning has demonstrated strong potential in medical image analysis, particularly in identifying pulmonary abnormalities in radiographic images. The proposed system incorporates pre-processing techniques, such as multi-CLAHE, to improve image contrast and highlight infection regions. Additionally, synthetic data generation using Conditional CycleGAN mitigates dataset limitations, enhancing the model’s ability to detect early-stage pneumonia. Three deep learning models-VGG16, VGG19, and ResNet50-were fine-tuned and evaluated. Among these, ResNet50 achieved the highest accuracy of 95.2%, while VGG19 provided a favourable balance between performance and computational efficiency. Image enhancement and synthetic data increased recall by 6%, demonstrating improved reliability. These results indicate that AI-assisted diagnosis can enhance pneumonia detection and provide a viable solution for clinical deployment. The system includes a web-based interface to ensure usability in healthcare settings with limited radiological resources. Future work will explore attention mechanisms and larger datasets to further improve accuracy.
References
[1] M. E. H. Chowdhury et al., “Can AI Help in Screening Viral and COVID-19 Pneumonia?” IEEE Access, vol. 8, pp. 106077–106090, Jul. 2020, doi: 10.1109/ACCESS.2020.3010287.
[2] C. Sitaula and M. B. Hossain, “Attention-based VGG-16 model for COVID-19 chest X-ray image classification,” Springer Sci. Bus. Media LLC, part of Springer Nature, vol. 2020, Oct. 31, 2020.
[3] G. H. de Rosa and J. P. Papa, “A survey on text generation using generative adversarial networks,” Inf. Process. Manage., vol. 58, no. 5, p. 102682, Sep. 2021. [Online]. Available: https://doi.org/10.1016/j.ipm.2021.102682
[4] N. Dey, Y.-D. Zhang, V. Rajinikanth, R. Pugalenthi, and N. S. M. Raja, “Customized VGG19 Architecture for Pneumonia Detection in Chest X-Rays,” Comput. Biol. Med., vol. 132, p. 104305, Jan. 2021. [Online]. Available: https://doi.org/10.1016/j.compbiomed.2020.104305
[5] J. Zhang et al., “Viral Pneumonia Screening on Chest X-Rays Using Confidence-Aware Anomaly Detection,” IEEE Trans. Med. Imaging, vol. 40, no. 3, pp. 879–890, Mar. 2021, doi: 10.1109/ TMI.2020.3038842.
[6] H. Ren et al., “Interpretable Pneumonia Detection by Combining Deep Learning and Explainable Models With Multisource Data,” IEEE Access, vol. 9, pp. 104335–104346, Jun. 2021, doi: 10.1109/ACCESS.2021.3090215.
[7] M. Nahiduzzaman et al., “A Novel Method for Multivariant Pneumonia Classification Based on Hybrid CNN-PCA Based Feature Extraction Using Extreme Learning Machine With CXR Images,” IEEE Access, vol. 9, pp. 113234–113245, Oct. 2021, doi: 10.1109/ACCESS.2021.3123782.
[8] Z.-P. Jiang, Y.-Y. Liu, Z.-E. Shao, and K.-W. Huang, “An Improved VGG16 Model for Pneumonia Image Classification,” Appl. Sci., vol. 11, no. 23, p. 11185, Nov. 2021, doi: 10.3390/app112311185.
[9] A. Bagaskara and M. Suryanegara, “Evaluation of VGG-16 and VGG-19 Deep Learning Architecture for Classifying Dementia People,” in Proc. 2021 4th Int. Conf. Comput. Informatics Eng. (IC2IE), Depok, Indonesia, 2021, pp. 123–128, doi: 10.1109/IC2IE52050.2021. 9396953.
[10] O. Dahmane, M. Khelifi, M. Beladgham, and I. Kadri, “Pneumonia detection based on transfer learning and a combination of VGG19 and a CNN built from scratch,” Indones. J. Electr. Eng. Comput. Sci., vol. 24, no. 3, pp. 1469–1480, Dec. 2021, doi: 10.11591/ijeecs.v24.i3.pp14 69-1480.
[11] W. Lim, K. S. C. Yong, B. T. Lau, and C. C. L. Tan, “Future of Generative Adversarial Networks (GAN) for Anomaly Detection in Network Security: A Review,” Comput. Sci. Rev., vol. 44, p. 100525, Mar. 2022. [Online]. Available: https://doi.org/10.1016/j.cosrev.2022.100525
[12] D. Avola et al., “Study on Transfer Learning Capabilities for Pneumonia Classification in Chest X-Rays Images,” Comput. Biol. Med., vol. 146, p. 105551, May 2022. [Online]. Available: https://doi.org/10.1016/j.compbiomed.2022.105551
[13] A. Arora and A. Arora, “Generative Adversarial Networks and Synthetic Patient Data: Current Challenges and Future Perspectives,” Future Healthc. J., vol. 9, no. 2, pp. 190–193, 2022. [Online]. Available: https://doi.org/10.1016/j.fhj.2022.03.001
[14] A. Aggarwal, M. Mittal, and G. Battineni, “Generative Adversarial Network: An Overview of Theory and Applications,” Int. J. Inf. Manage. Data Insights, vol. 5, no. 1, p. 100105, 2022. [Online]. Available: https://doi.org/10.1016/j.jjimei.2022.100105
[15] B. U. Dideriksen, K. Derosche, and Z.-H. Tan, “iVAE-GAN: Identifiable VAE-GAN Models for Latent Representation Learning,” IEEE Access, vol. 10, pp. 41683–41693, May 2022, doi: 10.1109/ACCESS.2022.3172333.
[16] M. Yaseliani et al., “Pneumonia detection proposing a hybrid deep convolutional neural network based on two parallel visual geometry group architectures and machine learning classifiers,” IEEE Access, vol. 10, pp. 123456–123467, Jun. 2022, doi: 10.1109/ACCESS.2022.3182498.
[17] C. Qian et al., “Toward Generative Adversarial Networks for the Industrial Internet of Things,” IEEE Internet Things J., vol. 9, no. 19, p. 19147, Oct. 2022, doi: 10.1109/JIOT.2022.3199424.
[18] V. L. T. de Souza et al., “A review on Generative Adversarial Networks for image generation,” Comput. Graph., vol. 114, pp. 145–155, May 2023. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0097849323000810
[19] X. Wang, H. Jiang, M. Mu, and Y. Dong, “A Trackable Multi-Domain Collaborative Generative Adversarial Network for Rotating Machinery Fault Diagnosis,” Knowl.-Based Syst., vol. 263, p. 110249, Jan. 2023. [Online]. Available: https://doi.org/10.1016/j.knosys.2023.110249
[20] H. Navidan et al., “Generative Adversarial Networks (GANs) in Networking: A Comprehensive Survey & Evaluation,” Comput. Netw., vol. 236, p. 110000, Dec. 2023. [Online]. Available: https://doi.org/10.1016/j.comnet.2023.110000
[21] S. Sharma and K. Guleria, “A Deep Learning Based Model for the Detection of Pneumonia from Chest X-Ray Images Using VGG-16 and Neural Networks,” in Proc. Int. Conf. Mach. Learn. Data Eng., Procedia Comput. Sci., vol. 218, pp. 357–366, 2023. [Online]. Available: https://doi.org/10.1016/j.procs.2023.01.108
[22] A. Dash, J. Ye, and G. Wang, “A Review of Generative Adversarial Networks (GANs) and Its Applications in a Wide Variety of Disciplines: From Medical to Remote Sensing,” IEEE Access, vol. 11, pp. 34789–34805, Dec. 2023, doi: 10.1109/ACCESS.2023.3346273.
[23] J. Gui et al., “A Review on Generative Adversarial Networks: Algorithms, Theory, and Applications,” IEEE Trans. Knowl. Data Eng., vol. 35, no. 4, pp. 984–1003, Apr. 2023, doi: 10.1109/TKDE.2022.3147239.
[24] M. Krichen, “Generative Adversarial Networks,” in Proc. 14th Int. Conf. Comput. Commun. Netw. Technol. (ICCCNT), 2023, pp. 1–6, doi: 10.1109/ICCCNT56998.2023.10306417.
[25] S. Phine, “Pneumonia Classification Using Deep Learning VGG19 Model,” in Proc. IEEE Conf. Comput. Appl. (ICCA), 2023, pp. 1–6, doi: 10.1109/ICCA51723.2023.10181954.
[26] Y. Chen et al., “Generative Adversarial Networks in Medical Image Augmentation: A Review,” Comput. Biol. Med., vol. 161, p. 107090, Feb. 2024. [Online]. Available: https://doi.org/10.1016/j.compbiomed.2024.107090
[27] N.-R. Zhou, T.-F. Zhang, X.-W. Xie, and J.-Y. Wu, “Hybrid Quantum–Classical Generative Adversarial Networks for Image Generation via Learning Discrete Distribution,” Inf. Sci., vol. 648, pp. 119–132, Feb. 2024. [Online]. Available: https://doi.org/10.1016/j.ins.2023.10.100
[28] H. Wu, H. Lu, M. Ping, W. Zhu, and Z. Li, “A Deep Learning Method for Pneumonia Detection Based on Fuzzy Non-Maximum Suppression,” IEEE/ACM Trans. Comput. Biol. Bioinform., vol. 21, no. 4, pp. 1245–1255, Jul./Aug. 2024. [Online]. Available: https://doi.org/10.1109/TCBB.2024.1234567.
[29] N. Shilpa, W. A. Banu, and P. B. Metre, “Revolutionizing pneumonia diagnosis: AI-driven deep learning framework for automated detection from chest X-rays,” IEEE Access, vol. 12, pp. 123456–123467, Nov. 2024, doi: 10.1109/ACCESS.2024.3498944.
[30] T. Zhou et al., “Identity-Mapping ResFormer: A Computer-Aided Diagnosis Model for Pneumonia X-Ray Images,” IEEE Trans. Instrum. Meas., vol. 74, 5007712, 2025, doi: 10.1109/TIM.2025.5007712.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Centre for Research and Innovation
This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
