Capsule Networks with Chest X-Ray Enhancement for Detection of COVID-19

Abstract

Corona virus was declared a global pandemic that has affected people worldwide. It is critical to diagnose corona virus-infected individuals to restrict the virus’s transmission. Recent research indicates that radiological methods provide valuable information in identifying infection using deep learning algorithms. Deep learning has contributed to large-scale medical data research, providing new ways and chances for diagnostic tools. This research attempted to investigate how the Capsule Networks leverage chest X-ray scans to identify the infected person. We suggest Capsule Networks identify the illness using chest X-ray data. The proposed approach is rapid and robust, classifying scans into COVID-19, No Findings, or any other issue in the lungs. The study can be used as a preliminary diagnosis by medical practitioners, and the study focuses on the COVID-19 class, a minority class in all public data sets accessible, and ensures that no COVID-19 infected individual is identified as Normal. Even with a small dataset, the model provides 96.37% accuracy for COVID-19 and for the non-COVID-19, and on multi-class classification, it provides an accuracy of 95.12%.