Coronavirus, a virus that spread worldwide rapidly and was eventually declared a pandemic. The rapid spread made it essential to detect Coronavirus infected people to control the further spread. Recent studies show that radiological images such as X-Rays and CT scans provide essential information in detecting infection using deep learning models. This paper proposes a shallow architecture based on Capsule Networks with convolutional layers to detect COVID-19 infected persons. The proposed method combines the ability of the capsule network to understand spatial information with convolutional layers for efficient feature extraction. Due to the model's shallow architecture, it has 23M parameters to train and requires fewer training samples. The proposed system is fast and robust and correctly classifies the X-Ray images into three classes, i.e. COVID-19, No Findings, and Viral Pneumonia. Experimental results on the X-Ray dataset show that our model performs well despite having fewer samples for the training and achieved an average accuracy of 96.47% for multi-class and 97.69% for binary classification on 5-fold cross-validation. The proposed model would be useful to researchers and medical professionals for assistance and prognosis for COVID-19 infected patients.
Keywords: COVID-19 detection; Capsule networks; Chest X-ray classification; Medical imaging.
© The Author(s), under exclusive licence to Springer Science+Business Media, LLC, part of Springer Nature 2023, Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.