ai-corona: Radiologist-assistant deep learning framework for COVID-19 diagnosis in chest CT scans
Mehdi Yousefzadeh, Parsa Esfahanian, M. Sadegh Movahed, Saeid Gorgin, Dara Rahmati, Atefeh Abedini, Seyed Alireza Nadji, Sara Haseli, Mehrdad Bakhshayesh Karam, Arda Kiani, Meisam Hoseinyazdi, Jafar Roshandel, Reza Lashgari
Abstract
The development of medical assisting tools based on artificial intelligence advances is essential in the global fight against COVID-19 outbreak and the future of medical systems. In this study, we introduce ai-corona, a radiologist-assistant deep learning framework for COVID-19 infection diagnosis using chest CT scans. Our framework incorporates an EfficientNetB3-based feature extractor. We employed three datasets; the CC-CCII set, the MasihDaneshvari Hospital (MDH) cohort, and the MosMedData cohort. Overall, these datasets constitute 7184 scans from 5693 subjects and include the COVID-19, non-COVID abnormal (NCA), common pneumonia (CP), non-pneumonia, and Normal classes. We evaluate ai-corona on test sets from the CC-CCII set, MDH cohort, and the entirety of the MosMedData cohort, for which it gained AUC scores of 0.997, 0.989, and 0.954, respectively. Our results indicates ai-corona outperforms all the alternative models. Lastly, our framework's diagnosis capabilities were evaluated as assistant to several experts. Accordingly, We observed an increase in both speed and accuracy of expert diagnosis when incorporating ai-corona's assistance.