PSSPNN: PatchShuffle Stochastic Pooling Neural Network for an explainable diagnosis of COVID-19 with multiple-way data augmentation

Wang, Shui-Hua ORCID logoORCID:, Zhang, Yin, Cheng, Xiaochun ORCID logoORCID:, Zhang, Xin ORCID logoORCID: and Zhang, Yu-Dong ORCID logoORCID: (2021) 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 , 6633755. pp. 1-18. ISSN 1748-670X [Article] (doi:10.1155/2021/6633755)

[img] PDF - Published version (with publisher's formatting)
Available under License Creative Commons Attribution 4.0.

Download (7MB)


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.

Item Type: Article
Keywords (uncontrolled): General Biochemistry, Genetics and Molecular Biology, Modelling and Simulation, General Immunology and Microbiology, Applied Mathematics, General Medicine
Research Areas: A. > School of Science and Technology > Computer Science
Item ID: 32820
Notes on copyright: Copyright © 2021 Shui-Hua Wang et al. This is an open access article distributed under the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited
Useful Links:
Depositing User: Jisc Publications Router
Date Deposited: 08 Apr 2021 10:20
Last Modified: 09 Nov 2022 11:53

Actions (login required)

View Item View Item


Activity Overview
6 month trend
6 month trend

Additional statistics are available via IRStats2.