Implementation of Intelligent Pneumonia Detection Model, Using Convolutional Neural Network (CNN) and InceptionV4 Transfer Learning Fine Tuning
PDF

Keywords

pneumonia
inceptionv4
transfer learning
cnn
fine tuning

How to Cite

Anggit Wirasto, Purwono, P., & Muhammad Baballe Ahmad. (2024). Implementation of Intelligent Pneumonia Detection Model, Using Convolutional Neural Network (CNN) and InceptionV4 Transfer Learning Fine Tuning. Journal of Advanced Health Informatics Research, 2(1), 1–11. https://doi.org/10.59247/jahir.v2i1.180

Abstract

In Pneumonia is a type of contagious lung infection that has caused many human deaths in the form of inflammation of the alveoli. Based on WHO data, pneumonia is a type of acute infection that has caused more than 450 million cases and 4 million deaths each year. Covid-19 is one of the global pandemics that triggered many pneumonia incidents. Chest X-rays (CXR) are an important part of patient care. Radiologists can use CXR features to determine the type of pneumonia and the underlying pathogenesis. Machine learning and deep learning technologies are used to automatically detect various human diseases, thus ensuring smart healthcare. CXR features are more suitable to be analyzed by convolutional neural network (CNN). This algorithm is one of the typical deep learning architectures that has strong characteristics that are widely applied in the healthcare field. This study aims to develop a deep learning-based paradigm to distinguish Covid-19 patients from healthy and normal individuals by analyzing the presence of pneumonia disease symptoms on the CXR. This research provides an approach to the use of InceptionV4 transfer learning type in performing classification on CXR images. There are three main approaches carried out, namely making a standard CNN model, optimizing transfer learning xceptiion and fine tuning. The performance metrics results show a recall value close to 100% with a model accuracy value of 88%. Achieving a high enough recall value with a relatively small dataset makes the model built is considered to have good capabilities. The ability is also confirmed by the high ROC-AUC value with a value of 0.965

https://doi.org/10.59247/jahir.v2i1.180
PDF

References

S. Kiliçarslan, C. Közkurt, S. Baş, and A. Elen, “Detection and classification of pneumonia using novel Superior Exponential (SupEx) activation function in convolutional neural networks,” Expert Syst Appl, vol. 217, p. 119503, 2023, doi: https://doi.org/10.1016/j.eswa.2023.119503.

H. Liz, M. Sánchez-Montañés, A. Tagarro, S. Domínguez-Rodríguez, R. Dagan, and D. Camacho, “Ensembles of Convolutional Neural Network models for pediatric pneumonia diagnosis,” Future Generation Computer Systems, vol. 122, pp. 220–233, 2021, doi: https://doi.org/10.1016/j.future.2021.04.007.

M. W. Kusk and S. Lysdahlgaard, “The effect of Gaussian noise on pneumonia detection on chest radiographs, using convolutional neural networks,” Radiography, vol. 29, no. 1, pp. 38–43, 2023, doi: https://doi.org/10.1016/j.radi.2022.09.011.

S. J. Wolf et al., “Clinical Policy: Critical Issues in the Management of Adult Patients Presenting to the Emergency Department With Community-Acquired Pneumonia,” Ann Emerg Med, vol. 77, no. 1, pp. e1–e57, 2021, doi: https://doi.org/10.1016/j.annemergmed.2020.10.024.

J. El Halabi et al., “Differential Impact of Systolic and Diastolic Heart Failure on In-Hospital Treatment, Outcomes, and Cost of Patients Admitted for Pneumonia,” American Journal of Medicine Open, p. 100025, 2022, doi: https://doi.org/10.1016/j.ajmo.2022.100025.

R. D. Goldman et al., “Willingness to vaccinate children against COVID-19 declined during the pandemic,” Vaccine, 2023, doi: https://doi.org/10.1016/j.vaccine.2023.02.069.

N. Shaker, J. P. Rosenheck, B. A. Whitson, and K. Shilo, “Pulmonary hematoidin deposition in a case of severe COVID19 pneumonia,” Human Pathology Reports, vol. 27, p. 300601, 2022, doi: https://doi.org/10.1016/j.hpr.2022.300601.

N. Sri Kavya, T. shilpa, N. Veeranjaneyulu, and D. Divya Priya, “Detecting Covid19 and pneumonia from chest X-ray images using deep convolutional neural networks,” Mater Today Proc, vol. 64, pp. 737–743, 2022, doi: https://doi.org/10.1016/j.matpr.2022.05.199.

S. Piconi, S. Pontiggia, M. Franzetti, F. Branda, and D. Tosi, “Statistical models to predict clinical outcomes with anakinra vs. tocilizumab treatments for severe pneumonia in COVID19 patients,” Eur J Intern Med, 2023, doi: https://doi.org/10.1016/j.ejim.2023.01.024.

X. Ying, H. Liu, and R. Huang, “COVID-19 chest X-ray image classification in the presence of noisy labels,” Displays, vol. 77, p. 102370, 2023, doi: https://doi.org/10.1016/j.displa.2023.102370.

L. Kong and J. Cheng, “Classification and detection of COVID-19 X-Ray images based on DenseNet and VGG16 feature fusion,” Biomed Signal Process Control, vol. 77, p. 103772, 2022, doi: https://doi.org/10.1016/j.bspc.2022.103772.

H. Naeem and A. A. Bin-Salem, “A CNN-LSTM network with multi-level feature extraction-based approach for automated detection of coronavirus from CT scan and X-ray images,” Appl Soft Comput, vol. 113, p. 107918, 2021, doi: https://doi.org/10.1016/j.asoc.2021.107918.

H. Liz, J. Huertas-Tato, M. Sánchez-Montañés, J. Del Ser, and D. Camacho, “Deep learning for understanding multilabel imbalanced Chest X-ray datasets,” Future Generation Computer Systems, 2023, doi: https://doi.org/10.1016/j.future.2023.03.005.

L. F. de J. Silva, O. A. C. Cortes, and J. O. B. Diniz, “A novel ensemble CNN model for COVID-19 classification in computerized tomography scans,” Results in Control and Optimization, vol. 11, p. 100215, 2023, doi: https://doi.org/10.1016/j.rico.2023.100215.

Md. B. Hossain, S. M. H. S. Iqbal, Md. M. Islam, Md. N. Akhtar, and I. H. Sarker, “Transfer learning with fine-tuned deep CNN ResNet50 model for classifying COVID-19 from chest X-ray images,” Inform Med Unlocked, vol. 30, p. 100916, 2022, doi: https://doi.org/10.1016/j.imu.2022.100916.

M. Bhandari, T. B. Shahi, B. Siku, and A. Neupane, “Explanatory classification of CXR images into COVID-19, Pneumonia and Tuberculosis using deep learning and XAI,” Comput Biol Med, vol. 150, p. 106156, 2022, doi: https://doi.org/10.1016/j.compbiomed.2022.106156.

Y. Xu, H.-K. Lam, G. Jia, J. Jiang, J. Liao, and X. Bao, “Improving COVID-19 CT classification of CNNs by learning parameter-efficient representation,” Comput Biol Med, vol. 152, p. 106417, 2023, doi: https://doi.org/10.1016/j.compbiomed.2022.106417.

G. Vrbančič and V. Podgorelec, “Efficient ensemble for image-based identification of Pneumonia utilizing deep CNN and SGD with warm restarts,” Expert Syst Appl, vol. 187, p. 115834, 2022, doi: https://doi.org/10.1016/j.eswa.2021.115834.

P. Ghose, Md. A. Uddin, U. K. Acharjee, and S. Sharmin, “Deep viewing for the identification of Covid-19 infection status from chest X-Ray image using CNN based architecture,” Intelligent Systems with Applications, vol. 16, p. 200130, 2022, doi: https://doi.org/10.1016/j.iswa.2022.200130.

J. Wei, R. Zhu, H. Zhang, P. Li, A. Okasha, and A. K. H. Muttar, “Application of PET/CT image under convolutional neural network model in postoperative pneumonia virus infection monitoring of patients with non-small cell lung cancer,” Results Phys, vol. 26, p. 104385, 2021, doi: https://doi.org/10.1016/j.rinp.2021.104385.

Z. Wang, L. Xia, H. Yuan, R. S. Srinivasan, and X. Song, “Principles, research status, and prospects of feature engineering for data-driven building energy prediction: A comprehensive review,” Journal of Building Engineering, vol. 58, p. 105028, 2022, doi: https://doi.org/10.1016/j.jobe.2022.105028.

R. Nyirandayisabye, H. Li, Q. Dong, T. Hakuzweyezu, and F. Nkinahamira, “Automatic pavement damage predictions using various machine learning algorithms: Evaluation and comparison,” Results in Engineering, vol. 16, p. 100657, 2022, doi: https://doi.org/10.1016/j.rineng.2022.100657.

I. S. Mangkunegara and P. Purwono, “Analysis of DNA Sequence Classification Using SVM Model with Hyperparameter Tuning Grid Search CV,” in 2022 IEEE International Conference on Cybernetics and Computational Intelligence (CyberneticsCom), 2022, pp. 427–432. doi: 10.1109/CyberneticsCom55287.2022.9865624.

E. Tartaglione, C. A. Barbano, C. Berzovini, M. Calandri, and M. Grangetto, “Unveiling COVID-19 from CHEST X-Ray with Deep Learning: A Hurdles Race with Small Data.,” Int J Environ Res Public Health, vol. 17, no. 18, Sep. 2020, doi: 10.3390/ijerph17186933.

M. M. Islam, F. Karray, R. Alhajj, and J. Zeng, “A Review on Deep Learning Techniques for the Diagnosis of Novel Coronavirus (COVID-19),” IEEE Access, vol. 9, pp. 30551–30572, 2021, doi: 10.1109/ACCESS.2021.3058537.

E. Chamseddine, N. Mansouri, M. Soui, and M. Abed, “Handling class imbalance in COVID-19 chest X-ray images classification: Using SMOTE and weighted loss,” Appl Soft Comput, vol. 129, p. 109588, 2022, doi: https://doi.org/10.1016/j.asoc.2022.109588.

S. S. Dambal, M. K. Doddananjedevaru, and S. B. Gopalakrishna, “Premature Ventricular Contraction Classification Based on Spiral Search - Manta Ray Foraging and Bi-LSTM,” International Journal of Intelligent Engineering and Systems, vol. 15, no. 6, pp. 1–10, 2022, doi: 10.22266/ijies2022.1231.01.

S. Bringas, S. Salomón, R. Duque, C. Lage, and J. L. Montaña, “Alzheimer’s Disease stage identification using deep learning models,” J Biomed Inform, vol. 109, p. 103514, 2020, doi: https://doi.org/10.1016/j.jbi.2020.103514.

L. Falaschetti, L. Manoni, D. Di Leo, D. Pau, V. Tomaselli, and C. Turchetti, “A CNN-based image detector for plant leaf diseases classification,” HardwareX, vol. 12, p. e00363, 2022, doi: https://doi.org/10.1016/j.ohx.2022.e00363.

H. Chen, Y. Yang, and S. Zhang, “Learning Robust Scene Classification Model with Data Augmentation Based on Xception,” J Phys Conf Ser, vol. 1575, no. 1, p. 12009, 2020, doi: 10.1088/1742-6596/1575/1/012009.

Y. Zhu, H. JiaYI, Y. Li, and W. Li, “Image identification of cashmere and wool fibers based on the improved Xception network,” Journal of King Saud University - Computer and Information Sciences, 2022, doi: https://doi.org/10.1016/j.jksuci.2022.09.009.

C. Upasana, A. S. Tewari, and J. P. Singh, “An Attention-based Pneumothorax Classification using Modified Xception Model,” Procedia Comput Sci, vol. 218, pp. 74–82, 2023, doi: https://doi.org/10.1016/j.procs.2022.12.403.

B. Gülmez, “A novel deep neural network model based Xception and genetic algorithm for detection of COVID-19 from X-ray images,” Ann Oper Res, 2022, doi: 10.1007/s10479-022-05151-y.

C. Amisse, M. E. Jijón-Palma, and J. A. Silva Centeno, “Fine-tuning deep learning models for pedestrian detection,” Boletim de Ciencias Geodesicas, vol. 27, no. 2, pp. 1–16, 2021, doi: 10.1590/S1982-21702021000200013.

Creative Commons License

This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

Copyright (c) 2024 Anggit Wirasto, Purwono Purwono, Muhammad Baballe Ahmad