Lung Region Segmentation Using Modified U-Net Architecture

Authors: Zhakaw Hamza Hamad1 & Taban Fouad Majeed2
1Computer Science Department, College of Science, Salaheddin University, Erbil, Iraq
2Computer Science Department, College of Science, Salaheddin University, Erbil, Iraq

Abstract: In order to combat the continuing COVID-19 pandemic, early infection diagnosis is essential. Lung infections can be detected using a screening technique called chest X-ray imaging (CXR). During pandemics, integrating machine learning techniques with medicine analysis is crucial in relieving the enormous load on healthcare systems and clinicians. As the continuing COVID-19 crisis intensifies in countries with dense populations and few testing kits, radiological Imaging can serve as a crucial diagnostic tool to properly classify covid-19 patients and administer the appropriate medication on time. In light of this objective, we describe our Research on segmenting lung areas utilizing a deep learning architecture and chest X-ray images as the source material. We used 2 public datasets, Montgomery & Shenzhen and Covid-Qu dataset. Also, the Research created the KURD-covid dataset and collected 1300 x-ray images from local health care centres that contain two labels, covid-19 and normal x-ray with manual masks for all images. We proposed the segmentation model based on state-of-art U-Net architecture. The result of the Covid-Qu daset was., %97 IoU and %99.5 DSC, and in KURD-covid is %71 IoU and %82 DSC.

Keywords: Deep Learning, Convolutional Neural Network, Segmentation, U-Net, Covid-19, X-Ray Images

Download the PDF Document

Doi: 10.23918/eajse.v8i3p25

Published: December 20, 2022

References

Ai, T., Yang, Z., Hou, H., Zhan, C., Chen, C., Lv, W., . . . Xia, L. (2020). Correlation of chest CT and RT-PCR testing for coronavirus disease 2019 (COVID-19) in China: a report of 1014 cases. Radiology, 296(2), E32-E40.

Alnaser, A., Gong, B., & Moeller, K. (2016). Evaluation of open-source software for the lung segmentation. Current Directions in Biomedical Engineering, 2(1), 515-518.

An, J., Cai, Q., Qu, Z., & Gao, Z. (2021). COVID-19 Screening in Chest X-Ray Images Using Lung Region Priors. IEEE Journal of Biomedical and Health Informatics, 25(11), 4119-4127.

Apostolopoulos, I. D., & Mpesiana, T. A. (2020). Covid-19: automatic detection from x-ray images utilizing transfer learning with convolutional neural networks. Physical and Engineering Sciences in Medicine, 43(2), 635-640.

Arias-Garzón, D., Alzate-Grisales, J. A., Orozco-Arias, S., Arteaga-Arteaga, H. B., Bravo-Ortiz, M. A., Mora-Rubio, A., . . . Cardona-Morales, O. (2021). COVID-19 detection in X-ray images using convolutional neural networks. Machine Learning with Applications, 6, 100138.

Bassi, P. R., & Attux, R. (2021). COVID-19 detection using chest X-rays: Is lung segmentation important for generalization? arXiv preprint arXiv:2104.06176.

Brenner, D. J., & Hall, E. J. (2007). Computed tomography—an increasing source of radiation exposure. New England journal of medicine, 357(22), 2277-2284.

Candemir, S., Jaeger, S., Palaniappan, K., Musco, J. P., Singh, R. K., Xue, Z., . . . McDonald, C. J. (2013). Lung segmentation in chest radiographs using anatomical atlases with nonrigid registration. IEEE Transactions on Medical Imaging, 33(2), 577-590.

Chowdhury, M. E., Rahman, T., Khandakar, A., Mazhar, R., Kadir, M. A., Mahbub, Z. B., . . . Al Emadi, N. (2020). Can AI help in screening viral and COVID-19 pneumonia? IEEE Access, 8, 132665-132676.

Fang, Y., Zhang, H., Xie, J., Lin, M., Ying, L., Pang, P., & Ji, W. (2020). Sensitivity of chest CT for COVID-19: comparison to RT-PCR. Radiology, 296(2), E115-E117.

Frid-Adar, M., Ben-Cohen, A., Amer, R., & Greenspan, H. (2018). Improving the segmentation of anatomical structures in chest radiographs using u-net with an imagenet pre-trained encoder Image Analysis for Moving Organ, Breast, and Thoracic Images (pp. 159-168): Springer.

Gaál, G., Maga, B., & Lukács, A. (2020). Attention u-net based adversarial architectures for chest x-ray lung segmentation. arXiv preprint arXiv:2003.10304.

Gopatoti, A., & Vijayalakshmi, P. (2022). Optimized chest X-ray image semantic segmentation networks for COVID-19 early detection. Journal of X-Ray Science and Technology(Preprint), 1-22.

Jaeger, S., Karargyris, A., Candemir, S., Folio, L., Siegelman, J., Callaghan, F., . . . Antani, S. (2013). Automatic tuberculosis screening using chest radiographs. IEEE Transactions on Medical Imaging, 33(2), 233-245.

Li, B., Kang, G., Cheng, K., & Zhang, N. (2019). Attention-guided convolutional neural network for detecting pneumonia on chest x-rays. Paper presented at the 2019 41st annual international conference of the ieee engineering in medicine and biology society (EMBC).

Long, C., Xu, H., Shen, Q., Zhang, X., Fan, B., Wang, C., . . . Li, H. (2020). Diagnosis of the Coronavirus disease (COVID-19): rRT-PCR or CT? European journal of radiology, 126, 108961.

Long, J., Shelhamer, E., & Darrell, T. (2015). Fully convolutional networks for semantic segmentation. Paper presented at the Proceedings of the IEEE conference on computer vision and pattern recognition.

Minaee, S., Boykov, Y. Y., Porikli, F., Plaza, A. J., Kehtarnavaz, N., & Terzopoulos, D. (2021). Image segmentation using deep learning: A survey. IEEE Transactions on Pattern Analysis and Machine Intelligence.

Oh, Y., Park, S., & Ye, J. C. (2020). Deep learning COVID-19 features on CXR using limited training data sets. IEEE Transactions on Medical Imaging, 39(8), 2688-2700.

Oulefki, A., Agaian, S., Trongtirakul, T., & Laouar, A. K. (2021). Automatic COVID-19 lung infected region segmentation and measurement using CT-scans images. Pattern recognition, 114, 107747.

Ozturk, T., Talo, M., Yildirim, E. A., Baloglu, U. B., Yildirim, O., & Acharya, U. R. (2020). Automated detection of COVID-19 cases using deep neural networks with X-ray images. Computers in Biology and Medicine, 121, 103792.

Rahman, T., Khandakar, A., Qiblawey, Y., Tahir, A., Kiranyaz, S., Kashem, S. B. A., . . . Khan, M. S. (2021). Exploring the effect of image enhancement techniques on COVID-19 detection using chest X-ray images. Computers in Biology and Medicine, 132, 104319.

Rajaraman, S., Siegelman, J., Alderson, P. O., Folio, L. S., Folio, L. R., & Antani, S. K. (2020). Iteratively pruned deep learning ensembles for COVID-19 detection in chest X-rays. IEEE Access, 8, 115041-115050.

Rashid, R., Akram, M. U., & Hassan, T. (2018). Fully convolutional neural network for lungs segmentation from chest X-rays. Paper presented at the International Conference Image Analysis and Recognition.

Ronneberger, O., Fischer, P., & Brox, T. (2015). U-net: Convolutional networks for biomedical image segmentation. Paper presented at the International Conference on Medical image computing and computer-assisted intervention.

Salehi, S., Abedi, A., Balakrishnan, S., & Gholamrezanezhad, A. (2020). Coronavirus disease 2019 (COVID-19): a systematic review of imaging findings in 919 patients. Ajr Am J Roentgenol, 215(1), 87-93.

Shan, F., Gao, Y., Wang, J., Shi, W., Shi, N., Han, M., . . . Shi, Y. (2020). Lung infection quantification of COVID-19 in CT images with deep learning. arXiv preprint arXiv:2003.04655.

Shi, F., Wang, J., Shi, J., Wu, Z., Wang, Q., Tang, Z., . . . Shen, D. (2020). Review of artificial intelligence techniques in imaging data acquisition, segmentation, and diagnosis for COVID-19. IEEE reviews in biomedical engineering, 14, 4-15.

Štifanić, D., Musulin, J., Jurilj, Z., Šegota, S., Lorencin, I., Anđelić, N., . . . Filipović, N. (2021). Semantic segmentation of chest X-ray images based on the severity of COVID-19 infected patients. EAI Endorsed Transactions on Bioengineering and Bioinformatics, 1, e3.

Tahir, A. M., Chowdhury, M. E., Khandakar, A., Rahman, T., Qiblawey, Y., Khurshid, U., . . . Al-Maadeed, S. (2021). COVID-19 infection localization and severity grading from chest X-ray images. Computers in Biology and Medicine, 139, 105002.

Teixeira, L. O., Pereira, R. M., Bertolini, D., Oliveira, L. S., Nanni, L., Cavalcanti, G. D., & Costa, Y. M. (2021). Impact of lung segmentation on the diagnosis and explanation of COVID-19 in chest X-ray images. Sensors, 21(21), 7116.

Zeiler, M. D., Krishnan, D., Taylor, G. W., & Fergus, R. (2010). Deconvolutional networks. Paper presented at the 2010 IEEE Computer Society Conference on computer vision and pattern recognition.

Zhu, N., Zhang, D., Wang, W., Li, X., Yang, B., Song, J., . . . Lu, R. (2020). A novel coronavirus from patients with pneumonia in China, 2019. New England journal of medicine.