Abstract

Pneumonia is one of the most prominent causes of mortality in children who are below the age of five years in most parts of the globe. Hence, adequate pneumonia diagnosis is of paramount importance and is what drove this research effort which has led to the development of two transfer learning-based ensemble models. One of the proposed models classifies the chest radiographs into normal and pneumonia cases with outputs being generated from VGG-16, Inception-v3, and two custom-made convolutional neural networks, PneumoNet-v1 and PneumoNet-v2. The second model distinguishes bacterial from viral pneumonia with the help of Xception, MobileNet-v2, and PneumoNet-v1. To accomplish the aim of the study, the Guangzhou Women and Children’s Medical Center dataset (Kermany Dataset) was used to benchmark model performance. PneumoNet-v1 and PneumoNet-v2 were designed with an emphasis for high classification accuracy and have individual accuracies of 96.2% and 96.8%, respectively for pneumonia detection. The first ensemble model used for classifying between healthy and infected images attained a classification accuracy of 98.03%. The second model used for differentiating between bacterial and viral demonstrated an accuracy of 91.93%. The effectiveness of transfer learning-based ensemble models as well as of the proposed custom CNN designs in enhancing the analysis of paediatric pneumonia and facilitating better diagnosis has been explored in this research.

Keywords

Transfer Learning, Weighted Ensemble, Paediatric Pneumonia, Customized CNN, Chest X-Rays (CXR),

Downloads

Download data is not yet available.

References

  1. C. Qin, D. Yao, Y. Shi, Z. Song, Computer-aided detection in chest radiography based on artificial intelligence: a survey. Biomedical engineering online, 17, (2018) 1-23. https://doi.org/10.1186/s12938-018-0544-y
  2. A. Sharma, D. Raju, S. Ranjan, (2017) Detection of pneumonia clouds in chest X-ray using image processing approach. In 2017 Nirma University International Conference on Engineering (NUiCONE), IEEE, India. https://doi.org/10.1109/NUICONE.2017.8325607
  3. B. Van Ginneken, B.T.H. Romeny, M.A. Viergever, Computer-aided diagnosis in chest radiography: a survey. IEEE Transactions on medical imaging, 20(12), (2001) 1228-1241. https://doi.org/10.1109/42.974918
  4. K. Kosasih, U. R. Abeyratne, V. Swarnkar, R. Triasih, Wavelet augmented cough analysis for rapid childhood pneumonia diagnosis. IEEE Transactions on biomedical engineering, 62(4), (2014) 1185-1194. https://doi.org/10.1109/TBME.2014.2381214
  5. H. Merzenich, L. Krille, G. Hammer, M. Kaiser, S. Yamashita, H. Zeeb, Paediatric CT scan usage and referrals of children to computed tomography in Germany-a cross-sectional survey of medical practice and awareness of radiation related health risks among physicians. BMC Health Services Research, 12, (2012) 1-7. https://doi.org/10.1186/1472-6963-12-47
  6. Z. Ge, D. Mahapatra, X. Chang, Z. Chen, L. Chi, H. Lu, Improving multi-label chest X-ray disease diagnosis by exploiting disease and health labels dependencies. Multimedia Tools and Applications, 79, (2020) 14889-14902. https://doi.org/10.1007/s11042-019-08260-2
  7. S. Goyal, R. Singh, Detection and classification of lung diseases for pneumonia and Covid-19 using machine and deep learning techniques. Journal of Ambient Intelligence and Humanized Computing, 14(4), (2023) 3239-3259. https://doi.org/10.1007/s12652-021-03464-7
  8. M.E. Chowdhury, T. Rahman, A. Khandakar, R. Mazhar, M.A. Kadir, Z.B. Mahbub, K.R. Islam, M.S Khan, A. Iqbal, N.A. Emadi, M.B.I. Reaz, M.T. Islam, Can AI help in screening viral and COVID-19 pneumonia?. Ieee Access, 8, (2020) 132665-132676. https://doi.org/10.1109/ACCESS.2020.3010287
  9. K. El Asnaoui, Y. Chawki, Using X-ray images and deep learning for automated detection of coronavirus disease. Journal of Biomolecular Structure and Dynamics, 39(10), (2021) 3615-3626. https://doi.org/10.1080/07391102.2020.1767212
  10. T. Wang, Z. Nie, R. Wang, Q. Xu, H. Huang, H. Xu, F. Xie, X.J. Liu, PneuNet: deep learning for COVID-19 pneumonia diagnosis on chest X-ray image analysis using Vision Transformer. Medical & Biological Engineering & Computing, 61(6), (2023) 1395-1408. https://doi.org/10.1007/s11517-022-02746-2
  11. M. Ali, M. Shahroz, U. Akram, M.F. Mushtaq, S.C. Altamiranda, S.A. Obregon, I.D.L.T. Díez, I. Ashraf, Pneumonia Detection Using Chest Radiographs with Novel EfficientNetV2L Model. IEEE Access, 12, (2024) 34691 – 34707. https://doi.org/10.1109/ACCESS.2024.3372588
  12. T. S. Arulananth, S.W. Prakash, R.K. Ayyasamy, V.P. Kavitha, P.G. Kuppusamy, P. Chinnasamy, Classification of Paediatric Pneumonia Using Modified DenseNet-121 Deep-Learning Model. IEEE Access, 12, (2024) 35716 – 35727. https://doi.org/10.1109/ACCESS.2024.3371151
  13. F. Sadik, A.G. Dastider, M.R. Subah, T. Mahmud, S.A. Fattah, A dual-stage deep convolutional neural network for automatic diagnosis of COVID-19 and pneumonia from chest CT images. Computers in biology and medicine, 149, (2022) 105806. https://doi.org/10.1016/j.compbiomed.2022.105806
  14. A. Hussain, S.U. Amin, H. Lee, A. Khan, N.F. Khan, S. Seo, An automated chest X-ray image analysis for covid-19 and pneumonia diagnosis using deep ensemble strategy. IEEE Access, 11, (2023) 97207 – 97220. https://doi.org/10.1109/ACCESS.2023.3312533
  15. M. Loey, S. El-Sappagh, S. Mirjalili, Bayesian-based optimized deep learning model to detect COVID-19 patients using chest X-ray image data. Computers in Biology and Medicine, 142, (2022) 105213. https://doi.org/10.1016/j.compbiomed.2022.105213
  16. A. Rattan, Ritanshu, S. Singh, J. Pruthi, (2024) Comparative Analysis of Two Approaches Utilizing Different Classification Algorithms Combined with CNN for Pneumonia Prediction in Chest X-Rays. In 2024 14th International Conference on Cloud Computing, Data Science & Engineering (Confluence), IEEE, India. https://doi.org/10.1109/Confluence60223.2024.10463235
  17. H. Malik, T. Anees, M.U. Chaudhry, R. Gono, M. Jasiński, Z. Leonowicz, P. Bernat, A novel fusion model of hand-crafted features with deep convolutional neural networks for classification of several chest diseases using X-ray images. IEEE Access, 11, (2023) 39243-39268. https://doi.org/10.1109/ACCESS.2023.3267492
  18. V. Chouhan, S.K. Singh, A. Khamparia, D. Gupta, P. Tiwari, C. Moreira, R. Damaševičius, V.H.C. De Albuquerque, A novel transfer learning based approach for pneumonia detection in chest X-ray images. Applied Sciences, 10(2), (2020) 559. https://doi.org/10.3390/app10020559
  19. P. Rajpurkar, J. Irvin, K. Zhu, B. Yang, H. Mehta, T. Duan, D. Ding, A. Bagul, C. Langlotz, K. Shpanskaya, M.P. Lungren, A.Y. Ng (2017) Chexnet: Radiologist-level pneumonia detection on chest x-rays with deep learning. arXiv. https://doi.org/10.48550/arXiv.1711.05225
  20. S. Albahli, H.T. Rauf, A. Algosaibi, V.E. Balas, AI-driven deep CNN approach for multi-label pathology classification using chest X-Rays. PeerJ Computer Science, 7, (2021) e495. https://doi.org/10.7717/peerj-cs.495
  21. T. Rahman, M.E. Chowdhury, A. Khandakar, K.R. Islam, K.F. Islam, Z.B. Mahbub, M.A. Kadir, S. Kashem, Transfer learning with deep convolutional neural network (CNN) for pneumonia detection using chest X-ray. Applied Sciences, 10(9), (2020) 3233. https://doi.org/10.3390/app10093233
  22. G. Liang, L. Zheng, A transfer learning method with deep residual network for pediatric pneumonia diagnosis. Computer methods and programs in biomedicine, 187, (2020) 104964. https://doi.org/10.1016/j.cmpb.2019.06.023
  23. U. Shah, A. Abd-Alrazeq, T. Alam, M. Househ, Z. Shah, (2020) An efficient method to predict pneumonia from chest X-rays using deep learning approach. In The importance of health informatics in public health during a pandemic, IOS Press, 457 – 460.
  24. T.B. Chandra, K. Verma, (2020) Pneumonia Detection on Chest X-Ray Using Machine Learning Paradigm. Proceedings of 3rd International Conference on Computer Vision and Image Processing. Advances in Intelligent Systems and Computing, Springer, Singapore.
  25. M.F. Hashmi, S. Katiyar, A.G. Keskar, N.D. Bokde, Z.W. Geem, Efficient pneumonia detection in chest xray images using deep transfer learning. Diagnostics, 10(6), (2020) 417. https://doi.org/10.3390/diagnostics10060417
  26. O. Stephen, M. Sain, U.J. Maduh, D.U. Jeong, An efficient deep learning approach to pneumonia classification in healthcare. Journal of healthcare engineering, 2019(1), (2019) 4180949. https://doi.org/10.3390/diagnostics10060417
  27. J. Zhang, Y. Xie, G. Pang, Z. Liao, J. Verjans, W. Li, W. Sun, J. He, Y. Li, Y. Xia, Viral pneumonia screening on chest X-rays using confidence-aware anomaly detection. IEEE transactions on medical imaging, 40(3), (2020) 879-890. https://doi.org/10.1109/TMI.2020.3040950
  28. T. Tuncer, F. Ozyurt, S. Dogan, A. Subasi, A novel Covid-19 and pneumonia classification method based on F-transform. Chemometrics and intelligent laboratory systems, 210, (2021)104256. https://doi.org/10.1016/j.chemolab.2021.104256
  29. A.K. Jaiswal, P. Tiwari, S. Kumar, D. Gupta, A. Khanna, J.J. Rodrigues, Identifying pneumonia in chest X-rays: A deep learning approach. Measurement, 145, (2019) 511-518. https://doi.org/10.1016/j.measurement.2019.05.076
  30. I. Pan, A. Cadrin-Chênevert, P.M. Cheng, Tackling the radiological society of North America pneumonia detection challenge. American Journal of Roentgenology, 213(3), (2019) 568-574. https://doi.org/10.2214/AJR.19.21512
  31. H.T. Nguyen, T. B. Tran, H.H. Luong, T.P. Le, N.C. Tran, Viral and bacterial pneumonia diagnosis via deep learning techniques and model explainability. International Journal of Advanced Computer Science and Applications, 11(7), (2020) 667-675. https://doi.org/10.14569/IJACSA.2020.0110780
  32. P.K. Nillmani, Jain, N. Sharma, M.K. Kalra, K. Viskovic, L. Saba, J.S. Suri, Four types of multiclass frameworks for pneumonia classification and its validation in X-ray scans using seven types of deep learning artificial intelligence models. Diagnostics, 12(3), (2022) 652. https://doi.org/10.3390/diagnostics12030652
  33. K. Simonyan, A. Zisserman, (2015) Very deep convolutional networks for large-scale image recognition. 3rd International Conference on Learning Representations (ICLR 2015), 1–14.
  34. C. Szegedy, V. Vanhoucke, S. Ioffe, J. Shlens, Z. Wojna, Rethinking the Inception Architecture for Computer Vision, IEEE Conference on Computer Vision and Pattern Recognition (CVPR), IEEE, USA. https://doi.org/10.1109/CVPR.2016.308
  35. G. Huang, Z. Liu, L. Van Der Maaten, K.Q. Weinberger, (2017) Densely Connected Convolutional Networks. IEEE Conference on Computer Vision and Pattern Recognition (CVPR), IEEE, USA. https://doi.org/10.1109/CVPR.2017.243
  36. F. Chollet, Xception: Deep learning with depthwise separable convolutions. In Proceedings of the IEEE conference on computer vision and pattern recognition, (2017) 1251-1258. https://doi.org/10.1109/CVPR.2017.195
  37. M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, L. Chen, MobileNetV2: Inverted Residuals and Linear Bottlenecks. IEEE/CVF Conference on Computer Vision and Pattern Recognition, USA. https://doi.org/10.1109/CVPR.2018.00474