Next Article in Journal
The Vitamin D Role in Preventing Primary Headache in Adult and Pediatric Population
Next Article in Special Issue
Acute Arterial Thrombosis of Lower Extremities in COVID-19 Patients
Previous Article in Journal
Systemic Treatment of Immune-Mediated Keratoconjunctivitis Sicca with Allogeneic Stem Cells Improves the Schirmer Tear Test Score in a Canine Spontaneous Model of Disease
Previous Article in Special Issue
COVID-19 Pandemic and Upcoming Influenza Season—Does an Expert’s Computed Tomography Assessment Differentially Identify COVID-19, Influenza and Pneumonias of Other Origin?
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

A Deep Learning Ensemble Approach for Automated COVID-19 Detection from Chest CT Images

1
CIRA—Italian Aerospace Research Centre, 81043 Capua, Italy
2
IRCCS Neuromed, 86077 Pozzilli, Italy
*
Author to whom correspondence should be addressed.
J. Clin. Med. 2021, 10(24), 5982; https://0-doi-org.brum.beds.ac.uk/10.3390/jcm10245982
Submission received: 15 October 2021 / Revised: 17 November 2021 / Accepted: 16 December 2021 / Published: 20 December 2021
(This article belongs to the Special Issue COVID-19: Diagnostic Imaging and Beyond - Part I)

Abstract

:
Background: The aim of this study was to evaluate the performance of an automated COVID-19 detection method based on a transfer learning technique that makes use of chest computed tomography (CT) images. Method: In this study, we used a publicly available multiclass CT scan dataset containing 4171 CT scans of 210 different patients. In particular, we extracted features from the CT images using a set of convolutional neural networks (CNNs) that had been pretrained on the ImageNet dataset as feature extractors, and we then selected a subset of these features using the Information Gain filter. The resulting feature vectors were then used to train a set of k Nearest Neighbors classifiers with 10-fold cross validation to assess the classification performance of the features that had been extracted by each CNN. Finally, a majority voting approach was used to classify each image into two different classes: COVID-19 and NO COVID-19. Results: A total of 414 images of the test set (10% of the complete dataset) were correctly classified, and only 4 were misclassified, yielding a final classification accuracy of 99.04%. Conclusions: The high performance that was achieved by the method could make it feasible option that could be used to assist radiologists in COVID-19 diagnosis through the use of CT images.

1. Introduction

In March 2020, the new coronavirus (COVID-19) pandemic was declared by the World Health Organization (WHO). As of now, there have been about 239 million confirmed cases of COVID-19, including more than 4 million deaths, according to the WHO [1]. The main common symptoms of COVID-19 include fever, dry cough, and tiredness. Since the virus may cause pneumonia as well as breathing difficulties or shortness of breath, chest pain or pressure, and loss of speech or movement in more aggressive infections, many healthcare systems around the world have suffered a breakdown, especially in terms of their intensive care units [2,3]. The gold standard for COVID-19 diagnosis is the nucleic acid kit for reverse transcription-polymerase chain reaction (RT-PCR) [4]. However, this method has several limitations, including false negatives (low sensitivity) [5,6], variability in diagnostic accuracy over the disease course [7], and a limited testing capacity in many countries [8]. Additionally, the ability of RT-PCR to detect COVID-19 strictly depends on the viral load. Medical imaging methods such as chest X-rays (CXR) and computer tomography (CT) can play a significant role in the diagnosis of COVID-19 [9,10], especially when they are used in combination with RT-PCR. In fact, they are very useful for monitoring disease progression and thus for optimizing the treatment strategy for the patient. CXR is a fast, cheap imaging method that is commonly used for the diagnosis of pneumonia worldwide [8,9]. It is less invasive than CT since it requires a lower dose of radiation. CXR is essential to evaluate pneumonia, pleural effusion, or pulmonary edema in COVID-19 patients, but its sensitivity in diagnosing COVID-19 is quite low [11]. CT is a medical imaging method that is based on X-rays and consists of many two-dimensional slices that allow high-resolution 3D images of the investigated body tract to be obtained through the application of a reconstruction algorithm. Chest CT is also widely used to diagnose COVID-19 because it seems to provide better diagnostic accuracy compared to CXR [12]. Although it is a new emerging disease, the intense research activity on the imaging data that can be obtained from COVID-19 patients that has been performed since the beginning of this pandemic has identified the imaging characteristics of COVID-19 [10]. CT findings include bilateral pulmonary parenchymal ground-glass and consolidative pulmonary opacities, sometimes with a rounded morphology. Furthermore, chest-CT scans have revealed incidental findings that are consistent with COVID-19, even in patients without respiratory symptoms [13]. The analysis of a CT scan requires an expert radiologist and takes about 10 min. This is a tedious and repetitive task that may cause doctors to experience mental fatigue, and it does not allow fast detection or screening in large-scale investigations. Furthermore, it has been demonstrated that the interpretation of CT images for COVID-19 diagnosis that have been made by radiologists do not have high sensitivity [14]. For these reasons, novel approaches have been proposed in order to find automated methods to detect COVID-19 in CT images. All of these approaches use artificial intelligence (AI) techniques, particularly those that are derived from machine learning (ML), which are considered to be a prominent tool for the prediction and diagnosis of numerous diseases [15]. In recent years, many research groups have tried to address the need for automated COVID-19 detection by proposing machine learning approaches that are based on clinical neuroimaging data. Although there are many studies that make use of CXR [16,17,18] or both image types (CXR and CT) [19,20], we only reported studies that make use of chest CT images, because these images are more accurate in COVID-19 diagnosis [12]. Shan et al. [21] proposed a deep learning-based segmentation system for quantitative infection assessment. The method includes the auto contouring of infection regions and the estimation of shape, volume, and percentage of infection, achieving Dice similarity coefficients of 91.6 ± 10.0% between automatic and manual segmentations, demonstrating a dramatic reduction in the time needed to delineate the infection compared to the manual approach. Alshazly et al. [22] used a number of different deep network architecture with transfer learning and conducted extensive sets of experiments to optimize the performance of the models on two different CT image datasets; the best model achieved an average accuracy of 99.4% and 92.9% on the two datasets. Xu et al. [23] used multiple CNN models to classify CT image datasets in two classes, namely COVID-19 and Influenza-A viral pneumonia, and to calculate the infection probability of COVID-19, achieving an overall accuracy of 86.7%. Wang et al. [24] proposed a deep learning algorithm using CT images to screen for COVID-19 during the influenza season using a transfer learning neural network that was based on the Inception network. Gozes et al. [25] presented a system that utilized both 2D and 3D deep learning models, modifying and adapting existing AI models to classify coronavirus and non-coronavirus cases. These authors achieved classification results of 0.996 AUC (Area Under the ROC Curve), with a sensitivity ranging from 96.4% to 98.2% and specificity from 92.2% to 98%. Hasan et al. [26] proposed a DenseNet-121 Convolutional Neural Network to classify and identify COVID-19 patients from CT images, achieving an accuracy of 92% with 95% recall. Rohila et al. [27] presented a method that could be used to detect varying degrees of COVID-19 infection from full chest CT scans by using a deep CNN model with ResNet-101, achieving an overall accuracy of 94.9%. Soares et al. [28] proposed an eXplainable Deep Learning approach (xDNN) that used a dataset that contained 2482 CT scans in total, 1252 CT scans that were positive for SARS-CoV-2 infection, and 1230 CT scans for patients who were not infected with SARS-CoV-2, achieving an F1 score of 97.31%. Loddo et al. [29] presented a method in which they first compared different architectures on a public and extended reference dataset to find the most suitable one, and then proposed a patient-oriented investigation to determine which network had the best performance. Finally, they evaluated their robustness in a real-world scenario, which was represented by cross-dataset experiments. They achieved 98.87% accuracy in their network comparison, 95.91% accuracy for patient status classification, and 70.15% accuracy in the real-world scenario. In this paper, we present a method that makes use of chest CT images taken from a publicly available dataset [28] in order to conduct an automated classification of the images in two classes, namely COVID-19 and NO_COVID-19. The method consists of the following steps:
  • A feature extraction step using a set of convolutional neural networks that have been pretrained on the ImageNet dataset [30]. ImageNet is a large, publicly available database of natural images with 1000 object classes that was specifically created for computer vision and, more recently, has been widely used for deep learning and transfer learning research. There are more than 1.2 million training images, 50,000 validation images, and 100,000 test images that are available in the database with relative annotations. The images are mostly in the JPG file format and vary in size.
  • A feature selection step that uses the information gain filter.
  • Training of the generated models using machine learning approaches.
  • A model selection step.
  • Classification of CT images into one of two classes using a majority voting approach.

2. Materials and Methods

2.1. Dataset

In this study, we used a publicly available multiclass CT scan dataset [28] that contained 4171 CT scans of 210 different patients, out of which 2167 correspond to 80 patients who were infected with SARS-CoV-2, a diagnosis that had been confirmed by RT-PCR. These data were collected at the Public Hospital of the Government Employees of Sao Paulo (HSPM) and the Metropolitan Hospital of Lapa, both in Sao Paulo, Brazil. The dataset is composed of CT scans in the png format that have been divided into 757 CT scans from healthy patients (15 CT scans per patient on average), 2167 CT scans from patients infected by SASR-CoV-2 (27 CT scans per patient on average), and 1247 CT scans from patients with other pulmonary directions (16 CT scans per patient on average). As our goal was to develop a method that would be able to distinguish CT images of COVID-19 patients from those of NO_COVID-19 patients, we decided to gather CT scans of healthy patients and of patients with other pulmonary diseases together in one group in order to simplify the training step. As a result, we created a balanced dataset that composed of 2167 CT scans with the COVID-19 label and 2004 CT scans with the NO_COVID-19 label.

2.2. Proposed Approach

The analysis of the whole dataset with all the 4171 CT images began with the separate use of N Deep Neural Architectures that had been pretrained on the ImageNet dataset, which is freely available through the TensorFlow framework [31], with the main goal of extracting different N sets of features. We chose to use the ImageNet dataset and transfer learning due to their increasing use in deep learning research and in many papers describing automated COVID-19 detection methods [25,28]. Once the N sets of features had been obtained, we performed a feature selection step, which was useful to reduce the dimensions of the dataset, by using the information gain filter [32], which is based on information entropy, which is widely used in ML. This filter will be described in detail in Section 3.1.
Then, a model was induced, where 90% of each set of features was used for the training set; its performances, e.g., its accuracy, was evaluated by applying the cross-validation method [32] with 10 folds, which allows models to be trained and evaluated when using small datasets. The remaining 10% of the examples was used for the final evaluation. In order to select the best deep neural architectures, the k-nearest neighbors (k = 1) (k-NN) algorithm was applied to the different N datasets of features by using the 10-fold cross validation technique. The M (M ≤ N) selected k-NN results were thus used for the final tests on the M Test Sets. Finally, the M k-NN classifications were included in an ensemble for a majority vote, and the performances of a set of independent images (the remaining 10% of the whole dataset) was presented for final testing; in particular, each image of the test set was classified according to the M k-NN, and it was assigned to the cored class (COVID-19 or NO_COVID-19) by majority. A summary of the steps that were used in our method is shown in Figure 1.

2.3. Transfer Learning

Transfer learning (TL), also called inductive transfer, knowledge transfer, or learning to learn [33], is an ML method that allows the domains, tasks, and/or data distributions that are used in training and testing phases to be differentiated. TL is a method that utilizes the knowledge that is achieved by a learning model after considering a specific problem to solve a distinct but similar task [34]. This transferred knowledge can be applied to a new dataset, the size of which is usually insufficient to train a new accurate model from scratch. Some authors [35] have defined three main different TL sub fields that are based on different conditions among the sources and the target domains and tasks, as reported in Table 1.
TL approaches can be divided into four main groups [33]:
  • Instance-based: Mainly refers to instance weighting strategy.
  • Feature-based: Transforms the original features to create a new feature representation.
  • Parameter-based: Transfers the knowledge at the model/parameter level.
  • Relational-based: Focuses on the problems in relational domains. This approach transfers the logical relationship or rules learned in the source domain to the target domain.
TL has been applied in many domains and application fields for many tasks. In recent studies, TL has been used in combination with the use of deep learning [36], for example, by means of convolutional networks (CNNs), where the method needs an initial training of a net for a given task using a large dataset. For example, in [17], the authors proposed an application of a TL method in the medical field that was aimed at the automatic detection of a COVID-19 infection from chest X-ray images. Since few chest images from infected patients were publicly available, the authors used different architectures of CNNs to apply a TL method.
The architectures of the networks were trained on ImageNet [30], and after being adapted to be feature extractors for the chest X-ray images, the proposed method achieved an accuracy of 98.5% by combining the extracted features with a classifier based on a support vector machine algorithm. This approach is called feature extraction for TL [37], where the extracted features are used in a new model later on that will process its classification. As such, the accessibility of a considerable set of data is the main factor that can guarantee the success of the training phase, and the use of large datasets for the initial training of the neural networks enables high performance in smaller or poorer datasets. Moreover, the feature extraction that is achieved with TL allows a large number of features to be extracted by generalizing the problem and by avoiding excessive customizations and adjustments.

2.4. Convolutional Neural Networks (CNNs)

Currently, CNNs have achieved state-of-the-art performance in computer vision tasks such as object recognition, image classification, and image segmentation. In these networks, the analysis of the input image proceeds through several layers of convolutional filters. The elements of the matrices that are implementing these filters are calculated during the training process, maximizing the performance of the network on the training dataset. In this adaptation lies the strength and flexibility of convolutional networks. A convolutional layer extracts those characteristic elements, called features, which are useful in the analysis process, from its input. A layer of the network implements many different convolutional filters and collects them into a so-called feature map. A convolutional layer may be followed by a pooling layer: a window of a predetermined size is scrolled along the layer output, and the values that are contained in it are replaced by their average (average pooling) or by the maximum element (max pooling). The pooling layers make the network more robust with respect to translations and reduce its output size. After the last convolutional or pooling layer, the feature maps are collected into a single one-dimensional vector of features that can be used as a unique descriptor of the content of the entire image. This vector is then used as the input of a classical, fully connected neural network that can be used for classification. The typical architecture of a CNN is showed in Figure 2.
Varying the number of convolutional and pooling layers, the number of filters in the feature maps and the way in which the layers are connected results in a wide variety of CNN architectures [38,39,40,41,42,43,44,45].
In general, training a CNN from scratch requires a large number of labelled images that can be used for training that may not be available for the problem at hand. This limitation can be at least partially overcome by using a network that has already been trained on a large dataset and that has been deprived of the final classification layers as a feature extractor (transfer learning).
In this work, we use a number of different CNN architectures that have been pre-trained as features extractors on the ImageNet dataset for the classification of CT scans in COVID and non-COVID classes. The dataset that we used contains images of different sizes. This is not a problem for convolutional and pooling layers, but it does produce vectors of features with different lengths as the output. Since classification algorithms expect the input vectors to be of a fixed size, we added a global max-pooling layer to every CNN that we used after removing the classification layers, which gives the max output values of the feature maps in the last layer of the convolutional network, fixing the size of the output vectors of the features.

2.5. K-Nearest Neighbor and Majority Voting Approach

The classifications were compared, and a majority voting rule was defined for the assignment of the final class. Therefore, instead of running the risk of picking an unsuitable or not very accurate classifier, an ensemble model can be used to achieve “better” results. The idea is that no single model or criterion can truly capture the optimal classification (or class separation rules), but a collective of models can provide a more robust final classification. As is often quoted in the literature [32,46,47], a classifier that is based on the majority voting method often achieves better performance than the performance of a single component method.
In this work, the classifiers (or the ensemble components) were trained using the k-NN algorithm, fixing k = 1, by considering different datasets of extracted features. For the sake of clarity, k-NN is instance-based or a lazy learner, and it does not require model training: all pf the samples that belong to the training set are memorized, and all of them are considered to classify every test sample, computing the distances individually between the test and all of the training samples. For this reason, k-NN is also classified as a memory-based learner, and the procedure does not gain a classifier as output. In more detail, it only starts working during the testing phase to compare the given test observations with the nearest training observations. The k-NN algorithm is one of the most commonly used methods in data mining (DM) due to its simplicity and high performance in many applications. It has gained popularity through the work of Aha [48]. k-NN is widely used in many ML and DM tasks, such as classification, motif discovery, and anomaly detection. It has shown excellent results in several application domains and in a large number of classification problems, including satellite image scenes, handwritten digits, ECG patterns, web search, spell checking, and fraud detection. It is often successful cases where the decision boundaries are very irregular [49]. The algorithm represents a classification method that is based on learning by analogy, in which a new object is labeled based on its closest (k) neighboring objects (points). It is based on the simple assumption that similar inputs are usually related to similar outputs. In other words, in the simplest case where k = 1, the class of the instance that is most similar, or close, to the new vector is used as the output class. If k > 1 and k is odd, then the output class is assigned to the new instance by considering the majority of the k class of the k nearest instances. The algorithm computes the distances between each point in the test set and all of the points of the training set in order to achieve its nearest-neighbor list. “Analogy” (or “closeness”, or “nearness”) is usually defined in terms of Euclidean distance, but other choices are possible [50].

3. Results

3.1. Feature Extraction and Selection

The analysis of the whole dataset containing the 4171 CT images begins with the separate use of 26 deep neural architectures, which are listed in Table 2, with the main goal of extracting 26 different sets of features. Once we obtained these 26 sets of features, we performed a feature selection step, which was useful to reduce the dimensions of the datasets. This was performed using the information gain (IG) filter, which evaluates the worth of an attribute by measuring the information gain with respect to the class. We chose all of the features X for which IG(X) > 0. The IG can be calculated by the following formula [32]:
IG(Class,Attribute) = H(Class) − H(Class|Attribute)
where H is the information entropy (or Shannon entropy), which can be calculated with the following formula:
H ( X ) = i = 1 n p ( x i ) log ( p ( x i ) )
The minimum value of IG for each dataset was equal to 0. The reduction in the number of features varied from 6.17% for EfficientNetB0 to 61.1% for ResNet101V2. Table 2 summarizes the number of features that was selected by applying the IG for each deep neural network (and therefore for each dataset).

3.2. Model Training

The hold-out method was applied. In this strategy, each of 26 feature datasets with labeled examples was partitioned into two disjoint subsets, called the training set and the test set. A model was induced from each training set, which comprised 90% of the whole dataset; its performances, e.g., its accuracy, which is the ratio between the number of correctly classified instances and the total amount of instances, was evaluated by applying the cross-validation method with 10 folds, which allows models with small datasets to be evaluated. The remaining 10% of the examples was used for the final evaluation. For the sake of clarity, the test sets were made up of feature vectors that resulted from the same set of images. In order to select the best deep neural architectures, the k-NN (fixing k = 1) algorithm was applied for the training phase on the 26 different datasets of features by using the 10-fold cross validation technique.
The average accuracies of the 26 classifiers are showed in Table 3.

3.3. Model Selection

The k-NN algorithm was used also to select the best M (M ≤ N) neural architectures or, equivalently, the M datasets of the features. Thus, the M datasets were selected through the use of an arbitrarily selected accuracy threshold of 80% of correctly classified instances. As such, the k-NN algorithm was applied with a dual purpose: to obtain the M classification rules using the concept of proximity and also to select the best M neural architectures and therefore the M datasets of features. Only one of the neural architectures (InceptionResNetV2) among those that were considered did not meet the chosen criterion (accuracy threshold of 80%). The accuracy of the remaining 25 classifiers on the independent test set (10% of the whole dataset of CT images) varied from 81.5789% to 96.6507%, with an average accuracy of 91.5502%. These 25 k-NNs were used for final tests on the 25 test sets.

3.4. Classification

Finally, the 25 k-NN classifiers were included in an ensemble for a majority vote, and the performances of a set of 414 independent images (the remaining 10% of the whole dataset) were presented for final testing; in particular, each image of the test set was classified by each of the 25 k-NNs, and its final class (COVID-19 or NO_COVID-19) was assigned by majority. Considering the 25 k-NNs as components of an ensemble classification model and by assigning the class label with the majority voting method, we obtained a meta-classifier (ensemble classifier) of 99.0431% accuracy, a value that was higher than those of the accuracies of the singular components (Table 3). In particular, 414 images from the test set (10% of the complete dataset) were correctly classified, and only 4 were misclassified. In particular, the final model showed two false positives (two false alarms or two CT scans incorrectly classified as containing COVID-19 infection) and two false negatives (two CT scans incorrectly classified as not containing COVID-19 infection). The confusion matrix of the final ensemble model is shown in Table 4, where TP is the number of true positives, FP is the number of false positives, FN is the number of false negatives and, finally, TN is the number of true negatives. As reported in Table 3 and Table 4, the ensemble meta-classifier produces better results than the individual classification components do. Many performance metrics [32] can be calculated from the confusion matrix. The most common as well as the most useful for model comparisons are shown in Table 5, where the values of our meta-classifier are also reported.

4. Discussion

The described study presents an automated method that can be used to detect COVID-19 infections from chest CT scans by using deep learning-based approaches. Specifically, we used 26 pre-trained deep neural architectures for feature extractions, an information gain filter to select a subset of the previously extracted features for each dataset, and the k nearest neighbors (k = 1) algorithm for target class detection and model comparison and selection. A total of 25 of the 26 models achieved the arbitrarily determined accuracy threshold for correctly classified instances of 80%. The selected 25 k nearest neighbors classifications were also used as the ensemble components for a majority voting approach that was able to classify each input image into two different classes: COVID-19 and NO_COVID-19. Although the dataset contained three different classes of subjects (COVID-19, healthy, and subjects with other pulmonary diseases), we decided to group the images of healthy subject and those of patients with other pulmonary diseases into one class (NO_COVID-19) in order to have a balanced dataset. Thus, we did not test the ability of this method to discriminate between COVID-19 and other pulmonary disease. We achieved an ensemble classification accuracy of 99.04%, which is greater than the accuracies of each of the models 25 individual components. Moreover, the results that were achieved by the proposed method (Table 5) exceed the results of all of the other works that make use of chest CT images. In particular, our technique appears to be better since it exhibits a very low percentage of misclassified images (see Table 6).
Our approach is different with respect to other similar approaches since we used CNN models that had been pretrained using the ImageNet dataset alone and without any fine tuning and simple KNN models in an ensemble, which greatly reduced the computational burden of the training phase, which is a critical parameter in a clinical setting. In fact, the time needed for the analysis of a single CT image was about 19 s on average, with a minimum of 8 s for the smallest images to a maximum of 28 s for the largest. The CNNs that were used were implemented using the TensorFlow framework on a laptop with Intel(R) Core (TM) i7-8665U CPU, 16 GB RAM, with no discrete graphic card. Considering the very high performance that was achieved by our method and the very low computational times, we strong believe that with a few improvements, it could provide a reliable and accurate method that could help radiologists and clinicians during the diagnostic process. First, our method does not perform any kind of pre-processing on the input images, and this could affect the performance of the method because it is well-known that it is important to pre-process imaging data in order for the model to provide more efficient analysis and better consistency. Furthermore, the proposed method did not perform any fine tuning, which could greatly improve the performance of the model; thus, in future developments of our methodology, we will include an ad hoc fine-tuning feature for all or for a subset of the considered neural architectures. We did not perform any fine tuning in order to reduce the computational time that was required to make a decision because our goal was to search for a method with a good trade-off between performance and computational burden. Since our method is useful for the selection of the most promising neural architectures that are able to detect COVID-19 infection, we plan to only consider the neural architectures with the highest performances in future work in order to apply a fine-tuning step, which would also serve to evaluate the use of a different machine learning algorithm, instead of the KNN approach. In our work, the ensemble of pre-trained neural architectures showed better performances (Table 4 and Table 5) than those of the singular components (Table 3). This result could suggest that the high performances that were achieved by our methodology could also be due to the choices that were made via majority voting; moreover, this approach also differentiates our method from those that were examined in related works (Table 6). Finally, the dataset that was used in this study is not big since it contains about 4000 CT images, and the number of the patients to which such images belong to is limited. Increasing the amount of data would improve any deep learning model. Thus, in the future, we plan to test our method in larger COVID-19 research databases acquired from different sources in order to improve the performance of the model and to make the methods more generalized. In fact, for the feature extraction step, we used 26 deep neural architectures that had been trained on the ImageNet dataset. Since this dataset is not specific for the diagnosis of COVID-19, we think that training these architectures on a large dataset that is specifically designed for COVID-19 diagnosis would improve the performance of our method. Finally, the lack of clinical data that are associated with the images did not allow us to investigate the effects of the clinical characteristics of the patients on the performance of the method, and we plan to carry out such an investigation in future studies.

Author Contributions

Conceptualization, G.Z., F.M., G.R. and L.P.; methodology, G.Z., F.M. and L.P.; validation, G.Z., F.M., G.R. and L.P.; formal analysis, G.Z., F.M. and L.P.; investigation, G.Z., F.M. and L.P.; data curation, G.Z., F.M. and L.P.; writing—original draft preparation, G.Z., F.M., G.R. and L.P.; writing—review and editing, G.Z. and L.P. All authors have read and agreed to the published version of the manuscript.

Funding

This work was funded by a grant from Ministero della Salute (Ricerca Corrente 2021). The research was performed as part of the employment of the authors with Italian Aerospace Research Centre (CIRA) and with IRCCS Neuromed.

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

The CT data used in this study are publicly available, while the data coming from deep learning analysis are available upon request by contacting the corresponding author.

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. World Health Organization. Available online: https://covid19.who.int/ (accessed on 11 October 2021).
  2. Armocida, B.; Formenti, B.; Ussai, S.; Palestra, F.; Missoni, E. The Italian health system and the COVID-19 challenge. Lancet Public Health 2020, 5, e253. [Google Scholar] [CrossRef]
  3. Narin, A.; Kaya, C.; Pamuk, Z. Automatic detection of coronavirus disease (COVID-19) using X-ray images and deep convolutional neural networks. arXiv 2020, arXiv:2003.10849. [Google Scholar] [CrossRef] [PubMed]
  4. Huang, P.; Liu, T.; Huang, L.; Liu, H.; Lei, M.; Xu, W.; Hu, X.; Chen, J.; Liu, B. Use of chest CT in combination with negative RT-PCR assay for the 2019 novel coronavirus but high clinical suspicion. Radiology 2020, 295, 22–23. [Google Scholar] [CrossRef]
  5. Winichakoon, P.; Chaiwarith, R.; Liwsrisakun, C.; Salee, P.; Goonna, A.; Limsukon, A.; Kaewpoowat, Q. Negative Nasopharyngeal and Oropharyngeal Swabs Do Not Rule Out COVID-19. J. Clin. Microbiol. 2020, 58, e00297-20. [Google Scholar] [CrossRef] [Green Version]
  6. Chen, Z.; Li, Y.; Wu, B.; Hou, Y.; Bao, J.; Deng, X. A patient with covid-19 presenting a false-negative reverse transcriptase polymerase chain reaction result. Korean J. Radiol. 2020, 21, 623–624. [Google Scholar] [CrossRef] [Green Version]
  7. Sethuraman, N.; Jeremiah, S.S.; Ryo, A. Interpreting Diagnostic Tests for SARS-CoV-2. JAMA 2020, 323, 2249–2251. [Google Scholar] [CrossRef]
  8. American Society for Microbiology. ASM Expresses Concern about Coronavirus Test Reagent Shortages. 2020. Available online: https://asm.org/Articles/Policy/2020/March/ASM-Expresses-Concern-about-Test-Reagent-Shortages (accessed on 11 October 2021).
  9. Liu, H.; Liu, F.; Li, J.; Zhang, T.; Wang, D.; Lan, W. Clinical and CT imaging features of the COVID-19 pneumonia: Focus on pregnant women and children. J. Infect. 2020, 80, e7–e13. [Google Scholar] [CrossRef]
  10. Chung, M.; Bernheim, A.; Mei, X.; Zhang, N.; Cui, J.; Jacobi, A.; Li, K.; Li, S.; Shan, H.; Xu, W.; et al. CT imaging features of 2019 novel coronavirus (2019-nCoV). Radiology 2020, 295, 202–207. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  11. Chen, S.-G.; Chen, J.-Y.; Yang, Y.-P.; Chien, C.-S.; Wang, M.-L.; Lin, L.-T. Use of radiographic features in COVID-19 diagnosis: Challenges and perspectives. J. Chin. Med. Assoc. 2020, 83, 644–647. [Google Scholar] [CrossRef]
  12. Borakati, A.; Perera, A.; Johnson, J.; Sood, T. Diagnostic accuracy of X-ray versus CT in COVID-19: A propensity-matched database study. BMJ Open 2020, 10, e042946. [Google Scholar] [CrossRef] [PubMed]
  13. Khazaei, M.; Asgari, R.; Zarei, E.; Moharramzad, Y.; Haghighatkhah, H.; Taheri, M.S. Incidentally Diagnosed COVID-19 Infection in Trauma Patients; a Clinical Experience. Arch. Acad. Emerg. Med. 2020, 8, e31. [Google Scholar] [PubMed]
  14. Bai, H.X.; Hsieh, B.; Xiong, Z.; Halsey, K.; Choi, J.W.; Tran, T.M.L.; Pan, I.; Shi, L.-B.; Wang, D.-C.; Mei, J.; et al. Performance of radiologists in differentiating COVID-19 from viral pneumonia on chest CT. Radiology 2020, 296, E46–E54. [Google Scholar] [CrossRef]
  15. Gao, L.; Zhang, L.; Liu, C.; Wu, S. Handling imbalanced medical image data: A deep-learning-based one-class classification approach. Artif. Intell. Med. 2020, 108, 101935. [Google Scholar] [CrossRef]
  16. Apostolopoulos, I.D.; Mpesiana, T.A. Covid-19: Automatic detection from X-ray images utilizing transfer learning with convolutional neural networks. Phys. Eng. Sci. Med. 2020, 43, 635–640. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  17. Ohata, E.F.; Bezerra, G.M.; das Chagas, J.V.S.; Neto, A.V.L.; Albuquerque, A.B.; de Albuquerque, V.H.C.; Filho, P.P.R. Automatic detection of COVID-19 infection using chest X-ray images through transfer learning. IEEE/CAA J. Autom. Sin. 2020, 8, 239–248. [Google Scholar] [CrossRef]
  18. Jain, R.; Gupta, M.; Taneja, S.; Hemanth, D.J. Deep learning based detection and analysis of COVID-19 on chest X-ray images. Appl. Intell. 2020, 51, 1690–1700. [Google Scholar] [CrossRef]
  19. Kassani, H.; Sara, P.H.K.; Wesolowski, M.J.; Schneider, K.A.; Deters, R. Automatic detection of coronavirus disease (COVID-19) in X-ray and CT images: A machine learning based approach. arXiv 2004, arXiv:2004.10641. [Google Scholar] [CrossRef]
  20. Maghdid, H.S.; Asaad, A.T.; Ghafoor, K.Z.G.; Sadiq, A.S.; Mirjalili, S.; Khan, M.K.K. Diagnosing COVID-19 Pneumonia from X-Ray and CT Images using Deep Learning and Transfer Learning Algorithms. SPIE 2021, 11734, 117340E. [Google Scholar] [CrossRef]
  21. Shan, F.; Gao, Y.; Wang, J.; Shi, W.; Shi, N.; Han, M.; Xue, J.; Shen, D.; Shi, Y. Lung Infection Quantification of COVID-19 in CT Images with Deep Learning. arXiv 2020, arXiv:2003.04655. [Google Scholar]
  22. Alshazly, H.; Linse, C.; Barth, E.; Martinetz, T. Explainable COVID-19 Detection Using Chest CT Scans and Deep Learning. Sensors 2021, 21, 455. [Google Scholar] [CrossRef]
  23. Xu, X.; Jiang, X.; Ma, C.; Du, P.; Li, X.; Lv, S.; Yu, L.; Ni, Q.; Chen, Y.; Su, J.; et al. A Deep Learning System to Screen Novel Coronavirus Disease 2019 Pneumonia. Engineering 2020, 6, 1122–1129. [Google Scholar] [CrossRef] [PubMed]
  24. Wang, S.; Kang, B.; Ma, J.; Zeng, X.; Xiao, M.; Guo, J.; Cai, M.; Yang, J.; Li, Y.; Meng, X.; et al. A deep learning algorithm using CT images to screen for Corona virus disease (COVID-19). Eur. Radiol. 2021, 31, 6096–6104. [Google Scholar] [CrossRef]
  25. Gozes, O.; Frid-Adar, M.; Greenspan, H.; Browning, P.D.; Zhang, H.; Ji, W.; Bernheim, A.; Siegel, E. Rapid AI development cycle for the coronavirus (COVID-19) pandemic: Initial results for automated detection & patient monitoring using deep learning ct image analysis. arXiv 2020, arXiv:2003.05037. [Google Scholar]
  26. Hasan, N.; Bao, Y.; Shawon, A.; Huang, Y. DenseNet Convolutional Neural Networks Application for Predicting COVID-19 Using CT Image. SN Comput. Sci. 2021, 2, 389. [Google Scholar] [CrossRef]
  27. Rohila, V.S.; Gupta, N.; Kaul, A.; Sharma, D.K. Deep learning assisted COVID-19 detection using full CT-scans. Internet Things 2021, 14, 100377. [Google Scholar] [CrossRef]
  28. Soares, E.; Angelov, P.; Biaso, S.; Froes, M.H.; Abe, D.K. SARS-CoV-2 CT-scan dataset: A large dataset of real patients CT scans for SARS-CoV-2 identification. medRxiv 2020. [Google Scholar] [CrossRef]
  29. Loddo, A.; Pili, F.; Di Ruberto, C. Deep Learning for COVID-19 Diagnosis from CT Images. Appl. Sci. 2021, 11, 8227. [Google Scholar] [CrossRef]
  30. Deng, J.; Dong, W.; Socher, R.; Li, L.J.; Li, K.; Li, F.-F. Imagenet: A large-scale hierarchical image database. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Miami, FL, USA, 20–25 June 2009; pp. 248–255. [Google Scholar]
  31. Abadi, M.; Barham, P.; Chen, J.; Chen, Z.; Davis, A.; Dean, J.; Devin, M.; Ghemawat, S.; Irving, G.; Isard, M.; et al. TensorFlow: A System for Large-Scale Machine Learning. In Proceedings of the 12th USENIX Symposium on Operating Systems Design and Implementation (OSDI ’16), Savannah, GA, USA, 2–4 November 2016. [Google Scholar]
  32. Tan, P.; Steinbach, M.; Kumar, V. Introduction to Data Mining; Pearson Addison Wesley: Boston, MA, USA, 2005. [Google Scholar]
  33. Pan, S.J.; Yang, Q. A Survey on Transfer Learning. IEEE Trans. Knowl. Data Eng. 2010, 22, 1345–1359. [Google Scholar] [CrossRef]
  34. Sarkar, D.; Bali, R.; Ghosh, T. Hands-On Transfer Learning with Python; Packt Publishing Ltd.: Birmingham, UK, 2018. [Google Scholar]
  35. Zhuang, F.; Qi, Z.; Duan, K.; Xi, D.; Zhu, Y.; Zhu, H.; Xiong, H.; He, Q. A Comprehensive Survey on Transfer Learning. Proc. IEEE 2021, 109, 43–76. [Google Scholar] [CrossRef]
  36. Yang, Q.; Zhang, Y.; Dai, W.; Pan, S.J. Transfer Learning; Cambridge University Press: Cambridge, UK, 2020. [Google Scholar]
  37. Huh, M.; Agrawal, P.; Efros, A.A. What makes imagenet good for transfer learning? arXiv 2016, arXiv:1608.08614. [Google Scholar]
  38. Huang, G.; Liu, Z.; Van Der Maaten, L.; Weinberger, K.Q. Densely Connected Convolutional Networks. In Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA, 21–26 July 2017; pp. 2261–2269. [Google Scholar] [CrossRef] [Green Version]
  39. Tan, M.; Le, Q. EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks. In Proceedings of the 36th International Conference on Machine Learning, Long Beach, CA, USA, 9–15 June 2019; Volume 97, pp. 6105–6114. Available online: http://proceedings.mlr.press/v97/tan19a.html (accessed on 11 October 2021).
  40. Szegedy, C.; Ioffe, S.; Vanhoucke, V.; Alemi, A.A. Inception-v4, inception-ResNet and the impact of residual connections on learning. In Proceedings of the Thirty-First AAAI Conference on Artificial Intelligence (AAAI’17), San Francisco, CA, USA, 4–9 February 2017; AAAI Press: Cambridge, MA, USA, 2017; pp. 4278–4284. [Google Scholar]
  41. Szegedy, C.; Liu, W.; Jia, Y.; Sermanet, P.; Reed, S.; Anguelov, D.; Erhan, D.; Vanhoucke, V.; Rabinovich, A. Going deeper with convolutions. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Boston, MA, USA, 7–12 June 2015; pp. 1–9. [Google Scholar] [CrossRef] [Green Version]
  42. Howard, A.G.; Zhu, M.; Chen, B.; Kalenichenko, D.; Wang, W.; Weyand, T.; Andreetto, M.; Adam, H. MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications. arXiv 2017, arXiv:1704.04861. [Google Scholar]
  43. He, K.; Zhang, X.; Ren, S.; Sun, J. Deep Residual Learning for Image Recognition. In Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, 27–30 June 2016; pp. 770–778. [Google Scholar] [CrossRef] [Green Version]
  44. Simonyan, K.; Zisserman, A. Very Deep Convolutional Networks for Large-Scale Image Recognition. In Proceedings of the International Conference on Learning Representations, San Diego, CA, USA, 7–9 May 2015. [Google Scholar]
  45. Chollet, F. Xception: Deep Learning with Depthwise Separable Convolutions. In Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA, 21–26 July 2017; pp. 1800–1807. [Google Scholar] [CrossRef] [Green Version]
  46. Leon, F.; Floria, S.; Bădică, C. Evaluating the effect of voting methods on ensemble-based classification. In Proceedings of the 2017 IEEE International Conference on INnovations in Intelligent SysTems and Applications (INISTA), Gdynia, Poland, 3–5 July 2017; pp. 1–6. [Google Scholar] [CrossRef]
  47. Kuncheva, L.I. Combining Pattern Classifiers: Methods and Algorithms, 2nd ed.; John Wiley & Sons, Inc.: Hoboken, NJ, USA, 2014. [Google Scholar]
  48. Aha, D.; Kibler, D. Instance-based learning algorithms. Mach. Learn. 1991, 6, 37–66. [Google Scholar] [CrossRef] [Green Version]
  49. Bhatia, N.; Vandana. Survey of Nearest Neighbor Techniques. IJCSIS Int. J. Comput. Sci. Inf. Secur. 2010, 8, 302–305. [Google Scholar]
  50. Lubis, A.R.; Lubis, M.; Al-Khowarizmi, A.-K. Optimization of distance formula in K-Nearest Neighbor method. Bull. Electr. Eng. Inform. 2020, 9, 326–338. [Google Scholar] [CrossRef]
Figure 1. Summary of the pipeline used in this study.
Figure 1. Summary of the pipeline used in this study.
Jcm 10 05982 g001
Figure 2. The typical architecture of a convolutional neural network.
Figure 2. The typical architecture of a convolutional neural network.
Jcm 10 05982 g002
Table 1. Transfer learning subsettings.
Table 1. Transfer learning subsettings.
Subsetting NameDescriptionLabel Information
1Inductive TLthe target task is different, but related,
from the source task
comes from the target domain
2Transductive TLthe source and target tasks are the same, while the source and target domains are
different
comes from the source domain
3Unsupervised TLthe target is different from, but related to, the source task, and the focus is on solving unsupervised learning tasks in the target
domain
is always unknown for both the source and the target domains
Table 2. Number of selected features by information gain for each deep neural network.
Table 2. Number of selected features by information gain for each deep neural network.
NDeep Neural NetworkMaximum Value of Information GainN. of Original FeaturesN. of Selected FeaturesPercentage Reduction of Features by IG
1DenseNet1210.14510259309.27%
2DenseNet1690.1621665148810.63%
3DenseNet2010.1651921166913.12%
4EfficientNetB00.203128111599.52%
5EfficientNetB10.160128112026.17%
6EfficientNetB20.187140913146.74%
7EfficientNetB30.207153714316.90%
8EfficientNetB40.1411793159810.88%
9EfficientNetB50.164204919037.13%
10EfficientNetB60.1812305207410.02%
11EfficientNetB70.157256123159.61%
12InceptionResNetV20.1261737153311.74%
13InceptionV30.186204918997.32%
14MobileNet0.175102582919.12%
15MobileNetV20.144128188930.60%
16MobileNetV3Large0.1161281108015.69%
17MobileNetV3Small0.111102584917.17%
18ResNet500.3332049173515.32%
19ResNet50V20.150204995553.39%
20ResNet1010.2042049171716.20%
21ResNet101V20.183204979761.10%
22ResNet1520.2842049166518.74%
23ResNet152V20.2062049151825.92%
24VGG160.15651340421.25%
25VGG190.18451344014.23%
26Xception0.1382049107147.73%
Table 3. Average accuracies of the 26 classifiers.
Table 3. Average accuracies of the 26 classifiers.
Average Accuracy
= 90.4548%
Average Accuracy
= 91.5502% (without InceptionResNetV2)
NNameAccuracy of k-NN (k = 1) 10-Fold Cross ValidatedAccuracy of k-NN (k = 1) on 10% Test Set
1DenseNet12193.4186%93.3014%
2DenseNet16992.5926%94.7368%
3DenseNet20191.3403%90.1914%
4EfficientNetB095.2038%95.933%
5EfficientNetB196.7493%96.6507%
6EfficientNetB293.6318%94.7368%
7EfficientNetB393.7393%96.1722%
8EfficientNetB492.8058%93.5407%
9EfficientNetB591.8732%91.3876%
10EfficientNetB688.5159%87.5598%
11EfficientNetB792.3261%94.0191%
12InceptionResNetV278.737%77.2727%
13InceptionV388.5425%88.756%
14MobileNet90.9406%90.6699%
15MobileNetV291.3669%93.7799%
16MobileNetV3Large88.5425%90.1914%
17MobileNetV3Small84.3858%83.7321%
18ResNet5095.3637%96.1722%
19ResNet50V281.1617%81.5789%
20ResNet10194.1913%93.5407%
21ResNet101V286.2776%84.9282%
22ResNet15294.1114%96.89%
23ResNet152V284.0927%85.6459%
24VGG1694.7509%93.3014%
25VGG1994.5377%95.933%
26Xception82.6272%85.4067%
Table 4. Confusion matrix on test set.
Table 4. Confusion matrix on test set.
YESNOClassified as
YESTP = 215FN = 2
NOFP = 2TN = 199
Meta-Classifier Accuracy 99.04%
Table 5. Performance metrics of the meta-classifier on test set.
Table 5. Performance metrics of the meta-classifier on test set.
SymbolPerformance MetricDefinition asWhat Does It Measure?Value
CCRCorrectly Classified instance Rate—Accuracy(TP + TN)/(TP + TN + FP + FN)How good the model is at correctly predicting both positive and negative cases0.9904
TPRTrue Positive Rate—Sensitivity—RecallTP/(TP + FN)How good the model is at correctly predicting positive cases0.9908
FPRFalse Positive Rate—Fall-outFP/(FP + TN)Proportion of incorrectly classified negative cases0.010
PPVPositive Predictive Value—PrecisionTP/(TP + FP)Proportion of correctly classified positive cases out of total positive predictions0.9908
AUCROC AreaArea under the ROC curveArea under plot of TPR against FPR0.997
Table 6. Related works for COVID-19 infection detection.
Table 6. Related works for COVID-19 infection detection.
AuthorML ApproachData SourceTransfer
Learning
Achieved
Performance
Alshazly et al. [22]Pre-trained SqueezeNet, Inception, ResNet, ResNeXt, Xception, ShuffleNet and DenseNet CNN with fine tuning2482 CT images + 746 CT imagesNot DeclaredAccuracy: 99.4% and 92.9% on the two datasets
Xu et al. [23]ROI segmentation with 3D CNN + Classification with ad hoc ResNet-18 CNN618 CT imagesNoAccuracy: 86.7%
Wang et al. [24]Pre-trained Inception CNN with fine tuning1065 CT imagesNot DeclaredAccuracy: 79.3%
Recall: 83%
Specificity: 67%
AUC: 0.81
Gozes et al. [25]Pre-trained ResNet-50 CNN with fine tuning206 patients CT scansImageNetAUC: 0.996
Hasan et al. [26]DenseNet-121 CNN2482 CT imagesNoAccuracy: 92%
Recall: 95%
Rohila et al. [27]Ad hoc deep learning network based on ResNet-1011110 patients CT scansYes, but no ImageNetAccuracy: 94.9%
Soares et al. [28]xDNN (eXplainable Deep Neural Network)2482 CT imagesImageNetAccuracy: 97.4%
Recall: 95.53%
Precision: 99.16%
AUC: 0.9736
Loddo et al. [29]Pre-trained AlexNet, Residual Networks, ResNet18, ResNet50, ResNet101, GoogLeNet, ShuffleNet, MobileNetV2, InceptionV3, VGG16 and VGG19470 + 194,122 Chest CT imagesNoAccuracy: 98.87% (nets comparison)
95.91% (patient status classification)
Our
approach
Pre-trained CNNs, k Nearest Neighbors with 10-fold cross validation, majority voting approach2482 CT imagesYes ImageNetAccuracy: 99.04%
Recall: 99.08%
Precision: 99.08%
AUC: 0.997
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Zazzaro, G.; Martone, F.; Romano, G.; Pavone, L. A Deep Learning Ensemble Approach for Automated COVID-19 Detection from Chest CT Images. J. Clin. Med. 2021, 10, 5982. https://0-doi-org.brum.beds.ac.uk/10.3390/jcm10245982

AMA Style

Zazzaro G, Martone F, Romano G, Pavone L. A Deep Learning Ensemble Approach for Automated COVID-19 Detection from Chest CT Images. Journal of Clinical Medicine. 2021; 10(24):5982. https://0-doi-org.brum.beds.ac.uk/10.3390/jcm10245982

Chicago/Turabian Style

Zazzaro, Gaetano, Francesco Martone, Gianpaolo Romano, and Luigi Pavone. 2021. "A Deep Learning Ensemble Approach for Automated COVID-19 Detection from Chest CT Images" Journal of Clinical Medicine 10, no. 24: 5982. https://0-doi-org.brum.beds.ac.uk/10.3390/jcm10245982

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop