Abstract

Aim COVID-19 has caused large death tolls all over the world. Accurate diagnosis is of significant importance for early treatment. Methods In this study, we proposed a novel PSSPNN model for classification between COVID-19, secondary pulmonary tuberculosis, community-captured pneumonia, and healthy subjects. PSSPNN entails five improvements: we first proposed the n-conv stochastic pooling module. Second, a novel stochastic pooling neural network was proposed. Third, PatchShuffle was introduced as a regularization term. Fourth, an improved multiple-way data augmentation was used. Fifth, Grad-CAM was utilized to interpret our AI model. Results The 10 runs with random seed on the test set showed our algorithm achieved a microaveraged F1 score of 95.79%. Moreover, our method is better than nine state-of-the-art approaches. Conclusion This proposed PSSPNN will help assist radiologists to make diagnosis more quickly and accurately on COVID-19 cases.

Highlights

  • COVID-19 is a type of disease caused by a new strain of coronavirus

  • We proposed a PatchShuffle SPNN (PSSPNN), which entails five improvements: (i) proposed n-conv stochastic pooling module (NCSPM) module, (ii) usage of stochastic pooling, (iii) usage of PatchShuffle, (iv) improved multiple-way data augmentation, and (v) explainability via Grad-CAM

  • Those five improvements enable our artificial intelligence (AI) model to deliver improved performances compared to 9 state-ofthe-art approaches

Read more

Summary

Introduction

COVID-19 is a type of disease caused by a new strain of coronavirus. “CO” means corona, “VI” virus, and “D” disease. The additional advantage of CCT is that it can provide highquality, three-dimensional chest data where radiologists can clearly view the COVID-19 lesions, which may be obscure in the other two CI approaches. Jiang and Zhang [8] proposed a 6-level convolutional neural network (6L-CNN) for therapy and rehabilitation Their performances were improved by replacing the traditional rectified linear unit with a leaky rectified linear unit. Wang et al [13] proposed a 3D deep convolutional neural network to detect COVID-19 (DeCovNet). Zhang et al [14] proposed a seven-layer convolutional neural network for COVID-19 diagnosis (7L-CCD). Their performance achieved an accuracy of 94:03 ± 0:80 for the binary classification task (COVID-19 against healthy subjects). (iii) A more advanced neural network, PatchShuffle SPNN (PSSPNN), is proposed where PatchShuffle is introduced as the regularization term in the loss function of SPNN (iv) An improved multiple-way data augmentation is utilized to help the network avoid overfitting (v) Grad-CAM is used to show the explainable heatmap, which displays association with lung lesions

Dataset
Methodology
Improvement II
Improvement III
Improvement IV
C2 C3 C4 MA
95 Optimal
Findings
Conclusion
Full Text
Published version (Free)

Talk to us

Join us for a 30 min session where you can share your feedback and ask us any queries you have

Schedule a call