COVID-19 Classification from Chest X-Ray Images: A Framework of Deep Explainable Artificial Intelligence.

Journal: Computational intelligence and neuroscience
Published Date:

Abstract

COVID-19 detection and classification using chest X-ray images is a current hot research topic based on the important application known as medical image analysis. To halt the spread of COVID-19, it is critical to identify the infection as soon as possible. Due to time constraints and the expertise of radiologists, manually diagnosing this infection from chest X-ray images is a difficult and time-consuming process. Artificial intelligence techniques have had a significant impact on medical image analysis and have also introduced several techniques for COVID-19 diagnosis. Deep learning and explainable AI have shown significant popularity among AL techniques for COVID-19 detection and classification. In this work, we propose a deep learning and explainable AI technique for the diagnosis and classification of COVID-19 using chest X-ray images. Initially, a hybrid contrast enhancement technique is proposed and applied to the original images that are later utilized for the training of two modified deep learning models. The deep transfer learning concept is selected for the training of pretrained modified models that are later employed for feature extraction. Features of both deep models are fused using improved canonical correlation analysis that is further optimized using a hybrid algorithm named Whale-Elephant Herding. Through this algorithm, the best features are selected and classified using an extreme learning machine (ELM). Moreover, the modified deep models are utilized for Grad-CAM visualization. The experimental process was conducted on three publicly available datasets and achieved accuracies of 99.1, 98.2, and 96.7%, respectively. Moreover, the ablation study was performed and showed that the proposed accuracy is better than the other methods.

Authors

  • Muhammad Attique Khan
    Department of Computer Science, HITEC University, Taxila, Pakistan.
  • Marium Azhar
    Department of Computer Science, University of Wah, Wah Cantt, Pakistan.
  • Kainat Ibrar
    Department of Computer Science, University of Wah, Wah Cantt, Pakistan.
  • Abdullah Alqahtani
    Department of Software Engineering, College of Computer Engineering and Sciences, Prince Sattam Bin Abdulaziz University, Al-Kharj 16278, Saudi Arabia.
  • Shtwai Alsubai
    Department of Computer Science, College of Computer Engineering and Sciences, Prince Sattam Bin Abdulaziz University, Al-Kharj, Saudi Arabia.
  • Adel Binbusayyis
    College of Computer Engineering and Sciences, Prince Sattam Bin Abdulaziz University, Al-Kharj, Saudi Arabia.
  • Ye Jin Kim
    Department of Computer Science, Hanyang University, Seoul, South Korea.
  • Byoungchol Chang
    Center for Computational Social Science, Hanyang University, Seoul 04763, Republic of Korea.