Detection of COVID-19 Cases Based on Deep Learning with X-ray Images

被引:3
作者
Wang, Zhiqiang [1 ]
Zhang, Ke [1 ]
Wang, Bingyan [1 ]
机构
[1] Beijing Elect Sci & Technol Inst, Dept Cyberspace Secur, Beijing 100070, Peoples R China
基金
中国博士后科学基金;
关键词
COVID-19 image detection; deep learning; attention mechanism; residual neural network; DIAGNOSIS; VALIDATION; NETWORK;
D O I
10.3390/electronics11213511
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Since the outbreak of COVID-19, the coronavirus has caused a massive threat to people's lives. With the development of artificial intelligence technology, identifying key features in medical images through deep learning, infection cases can be screened quickly and accurately. This paper uses deep-learning-based approaches to classify COVID-19 and normal (healthy) chest X-ray images. To effectively extract medical X-ray image features and improve the detection accuracy of COVID-19 images, this paper extracts the texture features of X-ray images based on the gray level co-occurrence matrix and then realizes feature selection by principal components analysis (PCA) and t-distributed stochastic neighbor embedding (T-SNE) algorithms. To improve the accuracy of X-ray image detection, this paper designs a COVID-19 X-ray image detection model based on the multi-head self-attention mechanism and residual neural network. It applies the multi-head self-attention mechanism to the residual network bottleneck layer. The experimental results show that the multi-head self-attention residual network (MHSA-ResNet) detection model has an accuracy of 95.52% and a precision of 96.02%. It has a good detection effect and can realize the three classifications of COVID-19 pneumonia, common pneumonia, and normal lungs, proving the method's effectiveness and practicability in this paper.
引用
收藏
页数:26
相关论文
共 33 条
[1]   Attention Augmented Convolutional Networks [J].
Bello, Irwan ;
Zoph, Barret ;
Vaswani, Ashish ;
Shlens, Jonathon ;
Le, Quoc V. .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3285-3294
[2]   Deep learning-based model for detecting 2019 novel coronavirus pneumonia on high-resolution computed tomography [J].
Chen, Jun ;
Wu, Lianlian ;
Zhang, Jun ;
Zhang, Liang ;
Gong, Dexin ;
Zhao, Yilin ;
Chen, Qiuxiang ;
Huang, Shulan ;
Yang, Ming ;
Yang, Xiao ;
Hu, Shan ;
Wang, Yonggui ;
Hu, Xiao ;
Zheng, Biqing ;
Zhang, Kuo ;
Wu, Huiling ;
Dong, Zehua ;
Xu, Youming ;
Zhu, Yijie ;
Chen, Xi ;
Zhang, Mengjiao ;
Yu, Lilei ;
Cheng, Fan ;
Yu, Honggang .
SCIENTIFIC REPORTS, 2020, 10 (01)
[3]  
Cohen J.P., 2020, arXiv
[4]   A Deep-Learning Diagnostic Support System for the Detection of COVID-19 Using Chest Radiographs A Multireader Validation Study [J].
Fontanellaz, Matthias ;
Ebner, Lukas ;
Huber, Adrian ;
Peters, Alan ;
Lobelenz, Laura ;
Hourscht, Cynthia ;
Klaus, Jeremias ;
Munz, Jaro ;
Ruder, Thomas ;
Drakopoulos, Dionysios ;
Sieron, Dominik ;
Primetis, Elias ;
Heverhagen, Johannes T. ;
Mougiakakou, Stavroula ;
Christe, Andreas .
INVESTIGATIVE RADIOLOGY, 2021, 56 (06) :348-356
[5]  
Ghoshal B., 2020, ARXIV
[6]   Accurate Screening of COVID-19 Using Attention-Based Deep 3D Multiple Instance Learning [J].
Han, Zhongyi ;
Wei, Benzheng ;
Hong, Yanfei ;
Li, Tianyang ;
Cong, Jinyu ;
Zhu, Xue ;
Wei, Haifeng ;
Zhang, Wei .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2020, 39 (08) :2584-2594
[7]   TEXTURAL FEATURES FOR IMAGE CLASSIFICATION [J].
HARALICK, RM ;
SHANMUGAM, K ;
DINSTEIN, I .
IEEE TRANSACTIONS ON SYSTEMS MAN AND CYBERNETICS, 1973, SMC3 (06) :610-621
[8]   Deep supervised learning using self-adaptive auxiliary loss for COVID-19 diagnosis from imbalanced CT images [J].
Hu, Kai ;
Huang, Yingjie ;
Huang, Wei ;
Tan, Hui ;
Chen, Zhineng ;
Zhong, Zheng ;
Li, Xuanya ;
Zhang, Yuan ;
Gao, Xieping .
NEUROCOMPUTING, 2021, 458 :232-245
[9]  
Jin C., 2020, DEV EVALUATION AI SY, DOI [10.1101/2020.03.20.20039834, DOI 10.1101/2020.03.20.20039834]
[10]  
Jin S., 2020, medRxiv