TY - JOUR
T1 - PSCNN
T2 - PatchShuffle Convolutional Neural Network for COVID-19 Explainable Diagnosis
AU - Wang, Shui Hua
AU - Zhu, Ziquan
AU - Zhang, Yu Dong
N1 - Publisher Copyright:
© Copyright © 2021 Wang, Zhu and Zhang.
PY - 2021/10/29
Y1 - 2021/10/29
N2 - Objective: COVID-19 is a sort of infectious disease caused by a new strain of coronavirus. This study aims to develop a more accurate COVID-19 diagnosis system. Methods: First, the n-conv module (nCM) is introduced. Then we built a 12-layer convolutional neural network (12l-CNN) as the backbone network. Afterwards, PatchShuffle was introduced to integrate with 12l-CNN as a regularization term of the loss function. Our model was named PSCNN. Moreover, multiple-way data augmentation and Grad-CAM are employed to avoid overfitting and locating lung lesions. Results: The mean and standard variation values of the seven measures of our model were 95.28 ± 1.03 (sensitivity), 95.78 ± 0.87 (specificity), 95.76 ± 0.86 (precision), 95.53 ± 0.83 (accuracy), 95.52 ± 0.83 (F1 score), 91.7 ± 1.65 (MCC), and 95.52 ± 0.83 (FMI). Conclusion: Our PSCNN is better than 10 state-of-the-art models. Further, we validate the optimal hyperparameters in our model and demonstrate the effectiveness of PatchShuffle.
AB - Objective: COVID-19 is a sort of infectious disease caused by a new strain of coronavirus. This study aims to develop a more accurate COVID-19 diagnosis system. Methods: First, the n-conv module (nCM) is introduced. Then we built a 12-layer convolutional neural network (12l-CNN) as the backbone network. Afterwards, PatchShuffle was introduced to integrate with 12l-CNN as a regularization term of the loss function. Our model was named PSCNN. Moreover, multiple-way data augmentation and Grad-CAM are employed to avoid overfitting and locating lung lesions. Results: The mean and standard variation values of the seven measures of our model were 95.28 ± 1.03 (sensitivity), 95.78 ± 0.87 (specificity), 95.76 ± 0.86 (precision), 95.53 ± 0.83 (accuracy), 95.52 ± 0.83 (F1 score), 91.7 ± 1.65 (MCC), and 95.52 ± 0.83 (FMI). Conclusion: Our PSCNN is better than 10 state-of-the-art models. Further, we validate the optimal hyperparameters in our model and demonstrate the effectiveness of PatchShuffle.
KW - Grad-CAM
KW - PatchShuffle
KW - convolutional neural network
KW - data augmentation
KW - deep learning
KW - stochastic pooling
UR - http://www.scopus.com/inward/record.url?scp=85119058770&partnerID=8YFLogxK
U2 - 10.3389/fpubh.2021.768278
DO - 10.3389/fpubh.2021.768278
M3 - Article
C2 - 34778194
AN - SCOPUS:85119058770
SN - 2296-2565
VL - 9
JO - Frontiers in Public Health
JF - Frontiers in Public Health
M1 - 768278
ER -