Wang SH, Zhang Y, Cheng X, Zhang X, Zhang YD. PSSPNN: PatchShuffle Stochastic Pooling Neural Network for an Explainable Diagnosis of COVID-19 with Multiple-Way Data Augmentation.
COMPUTATIONAL AND MATHEMATICAL METHODS IN MEDICINE 2021;
2021:6633755. [PMID:
33777167 PMCID:
PMC7945676 DOI:
10.1155/2021/6633755]
[Citation(s) in RCA: 24] [Impact Index Per Article: 6.0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Track Full Text] [Download PDF] [Figures] [Subscribe] [Scholar Register] [Received: 11/09/2020] [Revised: 12/23/2020] [Accepted: 02/18/2021] [Indexed: 12/31/2022]
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.
Collapse