City Research Online

COVID-CT-Mask-Net: Prediction of COVID-19 from CT Scans Using Regional Features

Ter-Sarkisov, A. ORCID: 0000-0002-1300-6132 (2021). COVID-CT-Mask-Net: Prediction of COVID-19 from CT Scans Using Regional Features. Applied Intelligence,

Abstract

We present COVID-CT-Mask-Net model that predicts COVID-19 in chest CT scans. The model works in two stages: in the first stage, Mask R-CNN is trained to localize and detect two types of lesions in images. In the second stage, these detections are fused to classify the whole input image. To develop the solution for the three-class problem (COVID-19, Common Pneumonia and Control), we used the COVIDx-CT data split derived from the dataset of chest CT scans collected by China National Center for Bioinformation.We use 3000 images (about 5% of the train split of COVIDx-CT) to train the model. Without any complicated data normalization, balancing and regularization, and training only a small fraction of the model's parameters, we achieve a 90:80% COVID-19 sensitivity, 91:62% Common Pneumonia sensitivity and 92:10% true negative rate (Control sensitivity), an overall accuracy of 91:66% and F1-score of 91:50% on the test data split with 21192 images, bringing the ratio of test to train data to 7.06.We also establish an important result that regional predictions (bounding boxes with confidence scores) detected by Mask R-CNN can be used to classify whole images. The full source code, models and pretrained weights are available on https://github.com/AlexTS1980/COVID-CT-Mask-Net.

Publication Type: Article
Additional Information: This version of the article has been accepted for publication, after peer review (when applicable) and is subject to Springer Nature’s AM terms of use, but is not the Version of Record and does not reflect post-acceptance improvements, or any corrections. The Version of Record will be available online at: http://link.springer.com/journal/10489
Subjects: H Social Sciences > HN Social history and conditions. Social problems. Social reform
Q Science > QR Microbiology
R Medicine > RA Public aspects of medicine > RA0421 Public health. Hygiene. Preventive Medicine
Departments: School of Mathematics, Computer Science & Engineering > Computer Science
[img] Text - Accepted Version
This document is not freely accessible due to copyright restrictions.

To request a copy, please use the button below.

Request a copy

Export

Downloads

Downloads per month over past year

View more statistics

Actions (login required)

Admin Login Admin Login