Diagnosing Pneumonia from Chest X-rays Using Deep Learning Algorithms Through Convolutional Neural Network, Transfer Learning and Fine Tuning

Authors

  • Kinjal Goswami Assistant Professor, Department of Computer Science and Engineering, Gayatri Vidya Parishad College of Engineering for Women Visakhapatnam, Andhra Pradesh, India
  • M. Bhanu Sridhar Associate Professor, Department of Information Technology, Gayatri Vidya Parishad College of Engineering for Women Visakhapatnam, Andhra Pradesh, India
  • Syed Suhana Student, Department of Computer Science and Engineering, Gayatri Vidya Parishad College of Engineering for Women Visakhapatnam, Andhra Pradesh, India

Keywords:

Chest X-rays, Convolutional Neural Network, Classify, Fine Tuning, Inflammatory, Pneumonia, Transfer Learning, VGG16, Xception

Abstract

Pneumonia is an inflammatory condition of the lungs that induces air sacs which leads to a contagious infection of lungs. Patients who are afflicted with the virus can be saved from death and the virus can be eradicated from spreading further through effective diagnosis. X-rays of the chest are frequently used to diagnose pneumonia. Detecting pneumonia from a Chest X-ray is typically slow and inaccurate. It is essential to identify pneumonia quickly so that patients can receive prompt care, especially in rural areas. This work proposes a system that evaluates chest X-rays and categorizes the images using Deep Convolutional Neural Network Architecture, Transfer Learning and Fine Tuning on different CNN architectures. As part of this project, the implemented algorithms include CNN, VGG16, and Xception. Prior to building a model, Data Augmentation and Data Balancing is performed in order to improve the model’s generalization performance and accuracy. Among the above-mentioned algorithms, it has been deduced that VGG16 model has returned best accuracy.

References

Yao S, Chen Y, Tian X, Jiang R, Ma S. An Improved Algorithm for Detecting Pneumonia Based on YOLOv3. Appl Sci. 2020 Mar; 10(5): 1818.

Jhuo S-L, Hsieh M-T, Weng T-C, Chen M-J, Yang C-M, Yeh C-H. Trend Prediction of Influenza and the Associated Pneumonia in Taiwan Using Machine Learning. 2019 International Symposium on Intelligent Signal Processing and Communication Systems (ISPACS). 2019; 1–2.

Gao P, Chen Z, Huang M, Ono N, Kanaya S, Altaf-UI-Amin M. An Approach to Construct and Validate TCM Dataset Effective against Bacterial Pneumonia. 2021 IEEE 3rd Global Conference on Life Sciences and Technologies (LifeTech). 2021; 102–103.

Yanqiu Ge, Qinghua Wang, et al. Predicting post-stroke pneumonia using deep neural network approaches. Int J Med Inform. 2019; 132: 103986.

Shangjie Yao, Yaowu Chen, Xiang Tian, Rongxin Jiang. Pneumonia Detection Using an Improved Algorithm Based on Faster R-CNN. Comput Math Methods Med. 2021; 2021: Article ID 8854892(13p).

Kuang Ming Kuo, Talley Paul C, Chi Hsien Huang, Liang Chih Cheng. Predicting hospital-acquired pneumonia among schizophrenic patients; a machine learning approach. BMC Med Inform Decis Mak. 2019; 19(1): 42. https://doi.org/10.1186/s12911-019-0792-1

Wang Q, Yang D, Li Z, Zhang X, Liu C. Deep Regression via Multi-Channel Multi-Modal Learning for Pneumonia Screening. IEEE Access. 2020; 8: 78530–78541.

Liu YC, Cheng HY, Chang TH, Ho TW, Liu TC, Yen TY, Chou CC, Chang LY, Lai F. Evaluation of the need for intensive care in children with Pneumonia: machine learning approach. JMIR Med Inform. 2022 Jan 27; 10(1): e28934.

Dawei Wang, Willis Deanna R, Yuehwern Yih. The pneumonia severity index: assessment and comparison to popular machine learning classifiers. Int J Med Inform. 2022; 163: 104778.

Afifi WNWM, Warsito IF, Sayahkarajy M, Supriyanto E. The development of an online pneumonia risk prediction system. 2017 International Conference on Robotics, Automation and Sciences (ICORAS). 2017; 1–5.

Sharma A, Raju D, Ranjan S. Detection of pneumonia clouds in chest X-ray using image processing approach. 2017 Nirma University International Conference on Engineering (NUiCONE). 2017;

–4.

Mehedi Hasan M, Omar Faruk M, Biswas Biki B, Riajuliislam M, Alam K, Farjana Shetu S. Prediction of Pneumonia Disease of Newborn Baby Based on Statistical Analysis of Maternal Condition Using Machine Learning Approach. 2021 11th International Conference on Cloud Computing, Data Science & Engineering (Confluence). 2021; 919–924.

Mao Xiaojian, Wang Heyong. Analysis of Pathogens of Pneumonia in children Based on Apriori Algorithm. 2011 IEEE International Symposium on IT in Medicine and Education. 2011; 460–463.

Fiszman M, Chapman WW, Aronsky D, Evans RS, Haug PJ. Automatic Detection of Acute Bacterial Pneumonia from Chest X-ray Reports. J Am Med Inform Assoc. 2000 Nov–Dec; 7(6): 593–604.

Gu H, Wang Y, Hong S, Gui G. Blind Channel Identification Aided Generalized Automatic Modulation Recognition Based on Deep Learning. IEEE Access. 2019; 7: 110722–110729.

Li Z, et al. Signal Analysis of Electrocardiogram and Statistical Evaluation of Myocardial Enzyme in the Diagnosis and Treatment of Patients with Pneumonia. IEEE Access. 2019; 7: 113751–113759.

Stephanakis IM, Anastassopoulos GK, Tsalkidis AD. Multiresolution autoregressive filtering for pneumonia detection in medical images. 2002 14th International Conference on Digital Signal Processing Proceedings, DSP 2002 (Cat. No.02TH8628), Santorini, Greece. 2002; 2: 1157–1159.

Kosasih K, Abeyratne UR, Swarnkar V, Triasih R. Wavelet Augmented Cough Analysis for Rapid Childhood Pneumonia Diagnosis. IEEE Trans Biomed Eng. 2015 Apr; 62(4): 1185–1194.

Published

2023-11-03

Issue

Section

Review Article