Enhancing Chest X-Ray Image Classification for Lung Diseases through Machine Learning


  • Sungju Park Seoul Scholars International
  • Hyunseo Cho Seoul Scholars International
  • Ahyoung Chung Seoul Scholars International
  • Timothy Han Seoul Scholars International
  • Taeoh Yi Seoul Scholars International
  • Angela Paik Seoul Scholars International
  • Taeheon Lee Seoul Scholars International




Chest x-ray, lung diseases, machine learning, VGGnet, classification, tuberculosis, COVID-19, lung opacity, pneumonia


Chest X-ray imaging is a widely used diagnostic tool for the detection and classification of various lung diseases. In this study, we propose a methodology to enhance the classification accuracy of chest X-ray images by leveraging machine learning techniques. Specifically, we employ the VGG-19 architecture to classify chest X-ray images into five distinct lung conditions: normal, tuberculosis, COVID-19, lung opacity, and pneumonia. A comprehensive dataset consisting of 31,787 chest X-ray images sourced from multiple medical institutions and hospitals worldwide is collected. Each image is labeled by expert radiologists with one of the five lung conditions. The dataset is then preprocessed and trained with 30 epochs. The trained VGGnet model achieved overall test accuracy of 95.11%, demonstrating its capability to accurately classify chest X-ray images into the five targeted lung conditions. The proposed methodology holds significant potential for improving the efficiency and accuracy of lung disease diagnosis based on chest X-ray images. By employing machine learning techniques, we can automate the classification process, providing clinicians with valuable decision support and expediting treatment plans. Moreover, the developed model can assist in the identification of lung diseases, including critical conditions such as COVID-19, enabling prompt and effective patient management.


Download data is not yet available.

References or Bibliography

Chowdhury, M. E., Rahman, T., Khandakar, A., Mazhar, R., Kadir, M. A., Mahbub, Z. B., Islam, K. R., Khan, M. S., Iqbal, A., Emadi, N. A., Reaz, M. B., & Islam, M. T. (2020). Can ai help in screening viral and covid-19 pneumonia? IEEE Access, 8, 132665–132676.


Chung, A. Actualmed COVID-19 chest x-ray data initiative. https://github.com/agchung/Actualmed-COVID-chestxray-dataset (2020).

Cohen, J. P., Morrison, P., Dao, L., Roth, K., Duong, T., & Ghassem, M. (2020a). Covid-19 image data collection: Prospective predictions are the future. Machine Learning for Biomedical Imaging, 1(December 2020), 1–38. https://doi.org/10.59275/j.melba.2020-48g7

Kermany, Daniel; Zhang, Kang; Goldbaum, Michael (2018), “Labeled Optical Coherence Tomography (OCT) and Chest X-Ray Images for Classification”, Mendeley Data, V2, doi: 10.17632/rscbjbr9sj.2

Lopes, C., Martins, J., Cavadas, V., Lourenço, A., & Campilho, A. (2021). COVID-19 classification from chest X-ray images using deep learning and transfer learning. Diagnostics, 11(4), 700.

Rahman, T., Khandakar, A., Kadir, M. A., Islam, K. R., Islam, K. F., Mazhar, R., Hamid, T., Islam, M. T., Kashem, S., Mahbub, Z. B., Ayari, M. A., & Chowdhury, M. E. (2020). Reliable tuberculosis detection using chest X-ray with deep learning, segmentation and visualization. IEEE Access, 8, 191586–191601. https://doi.org/10.1109/access.2020.3031384

Rahman, T., Khandakar, A., Qiblawey, Y., Tahir, A., Kiranyaz, S., Abul Kashem, S. B., Islam, M. T., Al Maadeed, S., Zughaier, S. M., Khan, M. S., & Chowdhury, M. E. H. (2021). Exploring the effect of image enhancement techniques on COVID-19 detection using chest X-ray images. Computers in Biology and Medicine, 132, 104319. https://doi.org/10.1016/j.compbiomed.2021.104319

Rajpurkar, P., Irvin, J., Zhu, K., Yang, B., Mehta, H., Duan, T., ... & Lungren, M. P. (2017). CheXNet: Radiologist-level pneumonia detection on chest X-rays with deep learning. arXiv preprint arXiv:1711.05225.

Simonyan, K., & Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556.

Speets, A. M. (2006). Upper abdominal ultrasound in general practice: Indications, diagnostic yield and consequences for patient management. Family Practice, 23(5), 507–511. https://doi.org/10.1093/fampra/cml027



How to Cite

Park, S., Cho, H., Chung, A. ., Han, T., Yi, T., Paik, A., & Lee, T. (2023). Enhancing Chest X-Ray Image Classification for Lung Diseases through Machine Learning. Journal of Student Research, 12(3). https://doi.org/10.47611/jsrhs.v12i3.5126



HS Research Projects