Automatic diagnosis of COVID-19 from CT images using CycleGAN and transfer learning

被引:18
作者
Ghassemi, Navid [1 ,2 ]
Shoeibi, Afshin [1 ,2 ,10 ]
Khodatars, Marjane [3 ]
Heras, Jonathan [4 ]
Rahimi, Alireza [2 ]
Zare, Assef [5 ]
Zhang, Yu-Dong [6 ]
Pachori, Ram Bilas [7 ]
Gorriz, Manuel [8 ,9 ]
机构
[1] KN Toosi Univ Technol, Fac Elect Engn, Nanostruct Elect Devices Lab, Tehran, Iran
[2] Ferdowsi Univ Mashhad, Comp Engn Dept, Mashhad, Iran
[3] Islamic Azad Univ, Dept Med Engn, Mashhad Branch, Mashhad, Iran
[4] Univ La Rioja, Dept Math & Comp Sci, La Rioja, Spain
[5] Islamic Azad Univ, Fac Elect Engn, Gonabad Branch, Gonabad, Iran
[6] Univ Leicester, Sch Informat, Leicester LE1 7RH, England
[7] Indian Inst Technol Indore, Dept Elect Engn, Indore 453552, India
[8] Univ Granada, Dept Signal Theory Networking & Commun, Granada, Spain
[9] Univ Cambridge, Dept Pathol, Cambridge, England
[10] KN Toosi Univ Technol, Fac Elect Engn, FPGA Lab, Tehran, Iran
关键词
COVID-19; CT scan; Deep learning; CycleGAN; Transfer learning; ARTIFICIAL-INTELLIGENCE; CHEST CT; CLASSIFICATION; NETWORK;
D O I
10.1016/j.asoc.2023.110511
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The outbreak of the corona virus disease (COVID-19) has changed the lives of most people on Earth. Given the high prevalence of this disease, its correct diagnosis in order to quarantine patients is of the utmost importance in the steps of fighting this pandemic. Among the various modalities used for diagnosis, medical imaging, especially computed tomography (CT) imaging, has been the focus of many previous studies due to its accuracy and availability. In addition, automation of diagnostic methods can be of great help to physicians. In this paper, a method based on pre-trained deep neural networks is presented, which, by taking advantage of a cyclic generative adversarial net (CycleGAN) model for data augmentation, has reached state-of-the-art performance for the task at hand, i.e., 99.60% accuracy. Also, in order to evaluate the method, a dataset containing 3163 images from 189 patients has been collected and labeled by physicians. Unlike prior datasets, normal data have been collected from people suspected of having COVID-19 disease and not from data from other diseases, and this database is made available publicly. Moreover, the method's reliability is further evaluated by calibration metrics, and its decision is interpreted by Grad-CAM also to find suspicious regions as another output of the method and make its decisions trustworthy and explainable.& COPY; 2023 Elsevier B.V. All rights reserved.
引用
收藏
页数:15
相关论文
共 102 条
[61]   Dual-Sampling Attention Network for Diagnosis of COVID-19 From Community Acquired Pneumonia [J].
Ouyang, Xi ;
Huo, Jiayu ;
Xia, Liming ;
Shan, Fei ;
Liu, Jun ;
Mo, Zhanhao ;
Yan, Fuhua ;
Ding, Zhongxiang ;
Yang, Qi ;
Song, Bin ;
Shi, Feng ;
Yuan, Huan ;
Wei, Ying ;
Cao, Xiaohuan ;
Gao, Yaozong ;
Wu, Dijia ;
Wang, Qian ;
Shen, Dinggang .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2020, 39 (08) :2595-2605
[62]  
Ozkaya U, 2020, Big Data Analytics and Artificial Intelligence Against COVID-19: Innovation Vision and Approach, P281
[63]  
™zkaya U, 2020, Arxiv, DOI arXiv:2011.05746
[64]   Automated detection of COVID-19 cases using deep neural networks with X-ray images [J].
Ozturk, Tulin ;
Talo, Muhammed ;
Yildirim, Eylul Azra ;
Baloglu, Ulas Baran ;
Yildirim, Ozal ;
Acharya, U. Rajendra .
COMPUTERS IN BIOLOGY AND MEDICINE, 2020, 121
[65]   Deep Transfer Learning Based Classification Model for COVID-19 Disease [J].
Pathak, Y. ;
Tiwari, A. ;
Stalin, S. ;
Singh, S. ;
Shukla, P. K. .
IRBM, 2022, 43 (02) :87-92
[66]  
Perchetti GA, 2020, J CLIN VIROL, V128, DOI 10.1016/j.jcv.2020.104438
[67]  
Pereyra G., 2017, arXiv, DOI DOI 10.48550/ARXIV.1701.06548
[68]   A light CNN for detecting COVID-19 from CT scans of the chest [J].
Polsinelli, Matteo ;
Cinque, Luigi ;
Placidi, Giuseppe .
PATTERN RECOGNITION LETTERS, 2020, 140 :95-100
[69]   Any unique image biomarkers associated with COVID-19? [J].
Pu, Jiantao ;
Leader, Joseph ;
Bandos, Andriy ;
Shi, Junli ;
Du, Pang ;
Yu, Juezhao ;
Yang, Bohan ;
Ke, Shi ;
Guo, Youmin ;
Field, Jessica B. ;
Fuhrman, Carl ;
Wilson, David ;
Sciurba, Frank ;
Jin, Chenwang .
EUROPEAN RADIOLOGY, 2020, 30 (11) :6221-6227
[70]   CNN Features off-the-shelf: an Astounding Baseline for Recognition [J].
Razavian, Ali Sharif ;
Azizpour, Hossein ;
Sullivan, Josephine ;
Carlsson, Stefan .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2014, :512-519