IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v11y2023i6p1489-d1100898.html
   My bibliography  Save this article

A Comparative Evaluation between Convolutional Neural Networks and Vision Transformers for COVID-19 Detection

Author

Listed:
  • Saad I. Nafisah

    (Department of Computer Engineering, College of Computer and Information Sciences, King Saud University, Riyadh 11543, Saudi Arabia)

  • Ghulam Muhammad

    (Department of Computer Engineering, College of Computer and Information Sciences, King Saud University, Riyadh 11543, Saudi Arabia)

  • M. Shamim Hossain

    (Department of Software Engineering, College of Computer and Information Sciences, King Saud University, Riyadh 11543, Saudi Arabia)

  • Salman A. AlQahtani

    (Department of Computer Engineering, College of Computer and Information Sciences, King Saud University, Riyadh 11543, Saudi Arabia)

Abstract

Early illness detection enables medical professionals to deliver the best care and increases the likelihood of a full recovery. In this work, we show that computer-aided design (CAD) systems are capable of using chest X-ray (CXR) medical imaging modalities for the identification of respiratory system disorders. At present, the COVID-19 pandemic is the most well-known illness. We propose a system based on explainable artificial intelligence to detect COVID-19 from CXR images by using several cutting-edge convolutional neural network (CNN) models, as well as the Vision of Transformer (ViT) models. The proposed system also visualizes the infected areas of the CXR images. This gives doctors and other medical professionals a second option for supporting their decision. The proposed system uses some preprocessing of the images, which includes the segmentation of the region of interest using a UNet model and rotation augmentation. CNN employs pixel arrays, while ViT divides the image into visual tokens; therefore, one of the objectives is to compare their performance in COVID-19 detection. In the experiments, a publicly available dataset (COVID-QU-Ex) is used. The experimental results show that the performances of the CNN-based models and the ViT-based models are comparable. The best accuracy was 99.82%, obtained by the EfficientNetB7 (CNN-based) model, followed by the SegFormer (ViT-based). In addition, the segmentation and augmentation enhanced the performance.

Suggested Citation

  • Saad I. Nafisah & Ghulam Muhammad & M. Shamim Hossain & Salman A. AlQahtani, 2023. "A Comparative Evaluation between Convolutional Neural Networks and Vision Transformers for COVID-19 Detection," Mathematics, MDPI, vol. 11(6), pages 1-20, March.
  • Handle: RePEc:gam:jmathe:v:11:y:2023:i:6:p:1489-:d:1100898
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/11/6/1489/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/11/6/1489/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Simon van der Pol & Paula Rojas Garcia & Maarten J. Postma & Fernando AntoƱanzas Villar & Antoinette D. I. Asselt, 2021. "Economic Analyses of Respiratory Tract Infection Diagnostics: A Systematic Review," PharmacoEconomics, Springer, vol. 39(12), pages 1411-1427, December.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Wu-Chun Chung & Yan-Hui Lin & Sih-Han Fang, 2023. "FedISM: Enhancing Data Imbalance via Shared Model in Federated Learning," Mathematics, MDPI, vol. 11(10), pages 1-22, May.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Simon Pol & Paula Rojas Garcia & Fernando AntoƱanzas Villar & Maarten J. Postma & Antoinette D. I. Asselt, 2021. "Health-Economic Analyses of Diagnostics: Guidance on Design and Reporting," PharmacoEconomics, Springer, vol. 39(12), pages 1355-1363, December.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:11:y:2023:i:6:p:1489-:d:1100898. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.