COVID-19 CT image recognition algorithm based on transformer and CNN

被引:95
作者
Fan, Xiaole [1 ]
Feng, Xiufang [1 ]
Dong, Yunyun [1 ]
Hou, Huichao [1 ]
机构
[1] Taiyuan Univ Technol, Coll Software, Taiyuan 030024, Peoples R China
关键词
Bi-directional feature fusion; Transformer; CNN; COVID-19; NETWORK;
D O I
10.1016/j.displa.2022.102150
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Novel corona virus pneumonia (COVID-19) broke out in 2019, which had a great impact on the development of world economy and people's lives. As a new mainstream image processing method, deep learning network has been constructed to extract medical features from chest CT images, and has been used as a new detection method in clinical practice. However, due to the medical characteristics of COVID-19 CT images, the lesions are widely distributed and have many local features. Therefore, it is difficult to diagnose directly by using the existing deep learning model. According to the medical features of CT images in COVID-19, a parallel bi-branch model (Trans-CNN Net) based on Transformer module and Convolutional Neural Network module is proposed by making full use of the local feature extraction capability of Convolutional Neural Network and the global feature extraction advantage of Transformer. According to the principle of cross-fusion, a bi-directional feature fusion structure is designed, in which features extracted from two branches are fused bi-directionally, and the parallel structures of branches are fused by a feature fusion module, forming a model that can extract features of different scales. To verify the effect of network classification, the classification accuracy on COVIDx-CT dataset is 96.7%, which is obviously higher than that of typical CNN network (ResNet-152) (95.2%) and Transformer network (Deit-B) (75.8%). These results demonstrate accuracy is improved. This model also provides a new method for the diagnosis of COVID-19, and through the combination of deep learning and medical imaging, it promotes the development of real-time diagnosis of lung diseases caused by COVID-19 infection, which is helpful for reliable and rapid diagnosis, thus saving precious lives.
引用
收藏
页数:9
相关论文
共 33 条
[1]  
[Anonymous], 2015, 1511 ARXIV
[2]   Towards Optimal Outsourcing of Service Function Chain Across Multiple Clouds [J].
Chen, Huan ;
Xu, Shizhong ;
Wang, Xiong ;
Zhao, Yangming ;
Li, Ke ;
Wang, Yang ;
Wang, Wei ;
Li, Le Min .
2016 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC), 2016,
[3]  
Chen Y., ARXIV PREPRINT ARXIV, P20
[4]  
Cheng Weiyue, J FRONTIERS COMPUTER, P1
[5]  
Dosovitskiy Alexey, 2021, arXiv
[6]  
Graham B., 2021, ARXIV PREPRINT ARXIV
[7]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586
[8]  
Howard A.G., 2017, arXiv
[9]   Searching for MobileNetV3 [J].
Howard, Andrew ;
Sandler, Mark ;
Chu, Grace ;
Chen, Liang-Chieh ;
Chen, Bo ;
Tan, Mingxing ;
Wang, Weijun ;
Zhu, Yukun ;
Pang, Ruoming ;
Vasudevan, Vijay ;
Le, Quoc V. ;
Adam, Hartwig .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :1314-1324
[10]   FaNet: fast assessment network for the novel coronavirus (COVID-19) pneumonia based on 3D CT imaging and clinical symptoms [J].
Huang, Zhenxing ;
Liu, Xinfeng ;
Wang, Rongpin ;
Zhang, Mudan ;
Zeng, Xianchun ;
Liu, Jun ;
Yang, Yongfeng ;
Liu, Xin ;
Zheng, Hairong ;
Liang, Dong ;
Hu, Zhanli .
APPLIED INTELLIGENCE, 2021, 51 (05) :2838-2849