PSSPNN: PatchShuffle Stochastic Pooling Neural Network for an Explainable Diagnosis of COVID-19 with Multiple-Way Data Augmentation.

Shui-Hua Wang,Yin Zhang,Xiaochun Cheng,Xin Zhang,Yu-Dong Zhang
DOI: https://doi.org/10.1155/2021/6633755
IF: 2.809
2021-01-01
Computational and Mathematical Methods in Medicine
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.
What problem does this paper attempt to address?