Classification of Chest Radiology Images in Order to Identify Patients with COVID-19 Using Deep Learning Techniques

Abstract

Background and Aim: Due to the important role of radiological images for identifying patients with COVID-19, creating a model based on deep learning methods was the main objective of this study. Materials and Methods: 15,153 available chest images of normal, COVID-19, and pneumonia individuals which were in the Kaggle data repository was used as dataset of this research. Data preprocessing including normalizing images, integrating images and labeling into three categories, train, test and validation was performed. By Python language in the fastAI library based on convolution technique (CNN) and four architectures (ResNet, VGG MobileNet, AlexNet), nine models through transitional learning method were trained to recognize patients from healthy persons. Finally, the performance of these models was evaluated with indicators such as accuracy, sensitivity and specificity, and F-Measure. Results: Of the nine generated models, the ResNet101 model has the highest ability to distinguish COVID-19 cases from other cases with 95.29% sensitivity. Other applied models showed more than 96% accuracy in correctly diagnosis of various cases in test phase. Finally, the ResNet101 model was able to demonstrate 98.4% accuracy in distinguishing between healthy and infected cases. Conclusion: The obtained accuracy showed the accurate performance of developed model in detecting COVID-19 cases. Therefore, by implementing an application based on the developed model, physicians can be helped in accurate and early diagnosis of cases. an application based on the developed model, physicians can be helped in accurate and early diagnosis of infected cases.

Keywords

• convolutional neural network
• coronavirus
• covid-19
• machine learning
• deep learning
• transfer learning