CGNet: A graph-knowledge embedded convolutional neural network for detection of pneumonia

Xiang Yu; Shui-Hua Wang; Yu-Dong Zhang

doi:10.1016/j.ipm.2020.102411

CGNet: A graph-knowledge embedded convolutional neural network for detection of pneumonia

Inf Process Manag. 2021 Jan;58(1):102411. doi: 10.1016/j.ipm.2020.102411. Epub 2020 Oct 19.

Authors

Xiang Yu¹, Shui-Hua Wang², Yu-Dong Zhang^{1

3}

Affiliations

¹ School of Informatics, University of Leicester, Leicester, LE1 7RH, UK.
² School of Architecture Building and Civil engineering, Loughborough University, Loughborough, LE11 3TU, UK.
³ Department of Information Systems, Faculty of Computing and Information Technology, King Abdulaziz University, Jeddah 21589, Saudi Arabia.

Abstract

Pneumonia is a global disease that causes high children mortality. The situation has even been worsening by the outbreak of the new coronavirus named COVID-19, which has killed more than 983,907 so far. People infected by the virus would show symptoms like fever and coughing as well as pneumonia as the infection progresses. Timely detection is a public consensus achieved that would benefit possible treatments and therefore contain the spread of COVID-19. X-ray, an expedient imaging technique, has been widely used for the detection of pneumonia caused by COVID-19 and some other virus. To facilitate the process of diagnosis of pneumonia, we developed a deep learning framework for a binary classification task that classifies chest X-ray images into normal and pneumonia based on our proposed CGNet. In our CGNet, there are three components including feature extraction, graph-based feature reconstruction and classification. We first use the transfer learning technique to train the state-of-the-art convolutional neural networks (CNNs) for binary classification while the trained CNNs are used to produce features for the following two components. Then, by deploying graph-based feature reconstruction, we, therefore, combine features through the graph to reconstruct features. Finally, a shallow neural network named GNet, a one layer graph neural network, which takes the combined features as the input, classifies chest X-ray images into normal and pneumonia. Our model achieved the best accuracy at 0.9872, sensitivity at 1 and specificity at 0.9795 on a public pneumonia dataset that includes 5,856 chest X-ray images. To evaluate the performance of our proposed method on detection of pneumonia caused by COVID-19, we also tested the proposed method on a public COVID-19 CT dataset, where we achieved the highest performance at the accuracy of 0.99, specificity at 1 and sensitivity at 0.98, respectively.

Keywords: COVID-19; Chest X-ray images; Feature reconstruction; Graph; Transfer learning.