COVID detection from Chest X-Ray Images using multi-scale attention.

Researchers

Journal

Modalities

Models

Abstract

Deep learning based methods have shown great promise in achieving accurate automatic detection of Coronavirus Disease (COVID) – 19 from Chest X-Ray (CXR) images. However, incorporating explainability in these solutions remains relatively less explored. We present a hierarchical classification approach for separating normal, non-COVID pneumonia (NCP) and COVID cases using CXR images. We demonstrate that the proposed method achieves clinically consistent explanations. We achieve this using a novel multi-scale attention architecture called Multi-scale Attention Residual Learning (MARL) and a new loss function based on conicity for training the proposed architecture. The proposed classification strategy has two stages. The first stage uses a model derived from DenseNet to separate pneumonia cases from normal cases while the second stage uses the MARL architecture to discriminate between COVID and NCP cases. With a five-fold cross validation, the proposed method achieves 93%, 96.28%, and 84.51% accuracy respectively over three public datasets for normal vs. NCP vs. COVID classification. This is competitive to the state-of-the-art methods. We also provide explanations in the form of GradCAM attributions, which are well aligned with expert annotations. The attributions are also seen to clearly indicate that MARL deems the peripheral regions of the lungs to be more important in the case of COVID cases while central regions are seen as more important in NCP cases. This observation matches the criteria described by radiologists in clinical literature, thereby attesting to the utility of the derived explanations.