U-Net Based Chest X-ray Segmentation with Ensemble Classification for Covid-19 and Pneumonia

Hashara Kumarasinghe; Shammi Kolonne; Chamodi Fernando; Dulani Meedeniya

doi:10.3991/ijoe.v18i07.30807

Authors

K. A. S. H. Kumarasinghe University of Moratuwa https://orcid.org/0000-0001-5926-2169
S. L. Kolonne University of Moratuwa https://orcid.org/0000-0001-7957-0345
K. C. M. Fernando University of Moratuwa https://orcid.org/0000-0002-1243-1567
D. Meedeniya University of Moratuwa https://orcid.org/0000-0002-4520-3819

DOI:

https://doi.org/10.3991/ijoe.v18i07.30807

Keywords:

Segmentation, U-Net, classification, CNN, chest X-rays

Abstract

Respiratory diseases have been known to be a main cause of death worldwide. Pneumonia and Covid-19 are two of the dominant diseases. Several deep learning based studies are available in the literature that classifies infection conditions in chest X-ray images. In addition, image segmentation has been also applied to obtain promising results in deep learning approaches. This paper focuses on using a modified version of the U-Net architecture to conduct segmentation on chest X-rays and then use segmented images for classification to assess the impact on the performance. We achieved an Intersection over Union of 93.53% with the proposed modified U-Net architecture and achieved 99.83% accuracy on segmentation aided ensemble classification.

Author Biographies

K. A. S. H. Kumarasinghe, University of Moratuwa

Ms. K.A.S.H. Kumarasinghe is a B.Sc. Eng. undergraduate in the Department of Computer Science and Engineering at the University of Moratuwa, Sri Lanka. Her main research interests are computer vision and deep learning.

S. L. Kolonne, University of Moratuwa

Ms. S. L. Kolonne is a B.Sc. Eng. undergraduate in the Department of Computer Science and Engineering at the University of Moratuwa, Sri Lanka. Her main research interests are computer vision and deep learning.

K. C. M. Fernando, University of Moratuwa

Ms. K.C.M. Fernando is a B.Sc. Eng. undergraduate in the Department of Computer Science and Engineering at the University of Moratuwa, Sri Lanka. Her main research interests are computer vision and deep learning.

D. Meedeniya, University of Moratuwa

Prof. D. Meedeniya is a Professor in Computer Science and Engineering at the University of Moratuwa, Sri Lanka. She holds a PhD in Computer Science from the University of St Andrews, United Kingdom. She is the director of the Bio-Health Informatics group at her department and engages in many collaborative research. She is a co-author of 100+ publications in indexed journals, peer-reviewed conferences and international book chapters. She serves as a reviewer, program committee and editorial team member in many international conferences and journals. Her main research interests are Software modelling and design, Bio-Health Informatics, Deep Learning and Technology-enhanced learning. She is a Fellow of HEA (UK), MIET, MIEEE, Member of ACM and a Chartered Engineer registered at EC (UK).