ResGNet-C: A graph convolutional neural network for detection of COVID-19

Xiang Yu, Siyuan Lu, Lili Guo, Shui Hua Wang*, Yu Dong Zhang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

55 Citations (Scopus)

Abstract

The widely spreading COVID-19 has caused thousands of hundreds of mortalities over the world in the past few months. Early diagnosis of the virus is of great significance for both of infected patients and doctors providing treatments. Chest Computerized tomography (CT) screening is one of the most straightforward techniques to detect pneumonia which was caused by the virus and thus to make the diagnosis. To facilitate the process of diagnosing COVID-19, we therefore developed a graph convolutional neural network ResGNet-C under ResGNet framework to automatically classify lung CT images into normal and confirmed pneumonia caused by COVID-19. In ResGNet-C, two by-products named NNet-C, ResNet101-C that showed high performance on detection of COVID-19 are simultaneously generated as well. Our best model ResGNet-C achieved an averaged accuracy at 0.9662 with an averaged sensitivity at 0.9733 and an averaged specificity at 0.9591 using five cross-validations on the dataset, which is comprised of 296 CT images. To our best knowledge, this is the first attempt at integrating graph knowledge into the COVID-19 classification task. Graphs are constructed according to the Euclidean distance between features extracted by our proposed ResNet101-C and then are encoded with the features to give the prediction results of CT images. Besides the high-performance system, which surpassed all state-of-the-art methods, our proposed graph construction method is simple, transferrable yet quite helpful for improving the performance of classifiers, as can be justified by the experimental results.

Original languageEnglish
Pages (from-to)592-605
Number of pages14
JournalNeurocomputing
Volume452
DOIs
Publication statusPublished - 10 Sept 2021
Externally publishedYes

Keywords

  • COVID-19
  • Deep learning
  • Graph neural network
  • Pneumonia
  • ResGNet-C

Fingerprint

Dive into the research topics of 'ResGNet-C: A graph convolutional neural network for detection of COVID-19'. Together they form a unique fingerprint.

Cite this