| Literature DB >> 34898851 |
Abstract
In image classification applications, the most important thing is to obtain useful features. Convolutional neural networks automatically learn the extracted features during training. The classification process is carried out with the obtained features. Therefore, obtaining successful features is critical to achieving high classification success. This article focuses on providing effective features to enhance classification performance. For this purpose, the success of the process of concatenating features in classification is taken as basis. At first, the features acquired by feature transfer method are extracted from AlexNet, Xception, NASNETLarge, and EfficientNet-B0 architectures, which are known to be successful in classification problems. Concatenating the features results in the creation of a new feature set. The method is completed by subjecting the features to various classification algorithms. The proposed pipeline is applied to the three datasets: "COVID-19 Image Dataset," "COVID-19 Pneumonia Normal Chest X-ray (PA) Dataset," and "COVID-19 Radiography Database" for COVID-19 disease detection. The whole datasets contain three classes (normal, COVID, and pneumonia). The best classification accuracies for the three datasets are 98.8%, 95.9%, and 99.6%, respectively. Performance metrics are given such as: sensitivity, precision, specificity, and F1-score values, as well. Contribution of paper is as follows: COVID-19 disease is similar to other lung infections. This situation makes diagnosis difficult. Furthermore, the virus's rapid spread necessitates the need to detect cases as soon as possible. There has been an increased curiosity in computer-aided deep learning models to provide the requirements. The use of the proposed method will be beneficial as it provides high accuracy.Entities:
Keywords: COVID‐19; classification; convolutional neural network (CNN); features concatenation; machine learning algorithms
Year: 2021 PMID: 34898851 PMCID: PMC8653237 DOI: 10.1002/ima.22659
Source DB: PubMed Journal: Int J Imaging Syst Technol ISSN: 0899-9457 Impact factor: 2.177
FIGURE 1Simple block diagram of the proposed pipeline
Class names and number of images included in datasets used in training and validation processes
| Dataset name | Class name | Number of image |
|---|---|---|
| COVID‐19 Image Dataset | COVID | 111 |
| Normal | 70 | |
| Viral pneumonia | 70 | |
| COVID‐19 Pneumonia Normal Chest X‐ray (PA) Dataset | COVID | 1525 |
| Normal | 1525 | |
| Pneumonia | 1525 | |
| COVID‐19 Radiography Database | COVID | 3616 |
| Normal | 10 192 | |
| Viral pneumonia | 1345 |
FIGURE 2Sample images from used datasets
FIGURE 4A view of a decision tree diagram
FIGURE 3(A) SVM for three class classification, (B) The LDA's guiding principle. Two‐dimensional data samples are projected into a lower‐dimensional space (line)
FIGURE 5Graphical abstract of proposed method
Formulas for calculating performance measures
| ACC | SN | SP | PREC | F1‐score |
|---|---|---|---|---|
|
|
|
|
|
|
Classification accuracy (%) of models according to the datasets to which they are applied
| Dataset | Features | LDA | SVM | KNN | DT | NB |
|---|---|---|---|---|---|---|
| COVID‐19 image Dataset | AlexNet | 90.8 | 89.6 | 86.1 | 80.9 | 89.2 |
| EfficientNet‐b0 | 94.4 | 94.4 | 92 | 80.9 | 89.2 | |
| NASNetLarge | 89.2 | 93.2 | 87.3 | 78.9 | 83.7 | |
| Xception | 80.5 | 89.2 | 87.6 | 80.5 | 85.3 | |
| COVID‐19 Pneumonia Normal Chest X‐ray (PA) Dataset | AlexNet | 93.4 | 93.2 | 89.8 | 84.2 | 84.9 |
| EfficientNet‐b0 | 94.7 | 93.9 | 91.8 | 84.1 | 87.1 | |
| NASNetLarge | 93.0 | 92.3 | 87.3 | 82.8 | 77.9 | |
| Xception | 93.4 | 93.2 | 89.8 | 83.0 | 85.9 | |
| COVID‐19 Radiography Database | AlexNet | 95.4 | 92.7 | 90.5 | 83.5 | 71.2 |
| EfficientNet‐b0 | 96.7 | 95.7 | 94.5 | 85.6 | 77.9 | |
| NASNetLarge | 93.2 | 92.5 | 86.8 | 82.0 | 72.1 | |
| Xception | 93.9 | 92.6 | 88.8 | 81.7 | 70.3 |
FIGURE 6Confusion matrix and performance metric values for COVID‐19 image Dataset with EfficientNet‐B0 + LDA
FIGURE 7Confusion matrix and performance metric values for COVID‐19 Pneumonia Normal Chest X‐ray (PA) Dataset with EfficientNet‐B0 + LDA
FIGURE 8Confusion matrix and performance metric values for COVID‐19 radiography database with EfficientNet‐B0 + LDA
Experimental results of the feature set created by concatenating features from two architectures (COVID‐19 image Dataset)
| Feature extractor | Classifier | Acc. (%) | SN (%) | SP (%) | PREC (%) | Time(s) (%) |
|---|---|---|---|---|---|---|
| AlexNet + EfficientNet‐b0 | LDA | 98.0 | 99.0 | 100 | 100 | 4.651 |
| SVM | 97.2 | 98.0 | 99.2 | 99.0 | 4.589 | |
| KNN | 94.0 | 91.8 | 100 | 100 | 3.864 | |
| DT | 81.3 | 88.2 | 90.0 | 87.5 | 4.958 | |
| NB | 94.0 | 98.1 | 97.1 | 96.4 | 10.757 | |
| AlexNet + NASNetLarge | LDA | 94.0 | 94.5 | 98.5 | 98.1 | 4.003 |
| SVM | 96.0 | 98.1 | 98.5 | 98.1 | 3.791 | |
| KNN | 90.4 | 94.5 | 97.1 | 96.3 | 3.542 | |
| DT | 80.5 | 90.0 | 90.7 | 88.4 | 3.661 | |
| NB | 93.2 | 96.3 | 96.4 | 95.5 | 10.538 | |
| AlexNet + Xception | LDA | 92.8 | 94.5 | 100 | 100 | 4.053 |
| SVM | 96.8 | 98.1 | 100 | 100 | 3.997 | |
| KNN | 92.0 | 92.7 | 100 | 100 | 3.883 | |
| DT | 83.9 | 90.9 | 95.7 | 94.3 | 3.854 | |
| NB | 95.6 | 97.2 | 100 | 100 | 11.171 | |
| EfficientNet‐b0 + NASNetLarge | LDA | 97.2 | 98.1 | 99.2 | 99.0 | 4.052 |
| SVM | 97.2 | 97.2 | 99.2 | 99.0 | 4.321 | |
| KNN | 91.6 | 92.7 | 98.5 | 98.0 | 4.038 | |
| DT | 76.5 | 84.6 | 87.8 | 84.6 | 3.956 | |
| NB | 93.6 | 97.2 | 97.8 | 97.2 | 11.139 | |
| EfficientNet‐b0 + Xception | LDA | 96.0 | 97.2 | 100 | 100 | 4.217 |
| SVM | 97.2 | 97.2 | 100 | 100 | 4.153 | |
| KNN | 96.4 | 95.4 | 100 | 100 | 4.019 | |
| DT | 80.1 | 87.3 | 93.3 | 90.6 | 4.010 | |
| NB | 93.6 | 95.4 | 100 | 100 | 11.475 | |
| NASNetLarge + Xception | LDA | 94.8 | 95.4 | 97.8 | 97.2 | 4.760 |
| SVM | 97.6 | 99.0 | 99.2 | 99.0 | 4.485 | |
| KNN | 90.8 | 91.8 | 97.1 | 96.2 | 3.862 | |
| DT | 80.1 | 90.9 | 90.0 | 87.8 | 3.942 | |
| NB | 94.8 | 97.2 | 98.5 | 97.2 | 12.314 |
Experimental results of the feature set created by concatenating features from three architectures (COVID‐19 image Dataset)
| Feature extractor | Classifier | Acc. (%) | SN (%) | SP (%) | PREC (%) | Time(s) |
|---|---|---|---|---|---|---|
| AlexNet + EfficientNet‐b0 + NASNetLarge | LDA | 98.0 | 99.0 | 100 | 100 | 8.044 |
| SVM | 98.0 | 99.0 | 99.2 | 99.0 | 7.576 | |
| KNN | 94.4 | 95.4 | 100 | 100 | 7.180 | |
| DT | 84.1 | 91.8 | 90.0 | 87.9 | 7.027 | |
| NB | 95.2 | 97.2 | 97.1 | 96.4 | 20.428 | |
| AlexNet + EfficientNet‐b0 + Xception | LDA | 98.4 | 99.0 | 100 | 100 | 8.673 |
| SVM | 98.4 | 99.0 | 100 | 100 | 7.886 | |
| KNN | 96.4 | 97.2 | 100 | 100 | 7.301 | |
| DT | 83.7 | 86.4 | 95.7 | 94.1 | 7.719 | |
| NB | 97.2 | 97.2 | 100 | 100 | 20.97 | |
| EfficientNet‐b0 + NASNetLarge + Xception | LDA | 97.6 | 99.0 | 100 | 100 | 8.830 |
| SVM | 98.0 | 99.0 | 99.2 | 99.0 | 8.057 | |
| KNN | 94.8 | 95.4 | 100 | 100 | 8.360 | |
| DT | 80.5 | 86.4 | 92.8 | 90.5 | 7.560 | |
| NB | 96.0 | 98.1 | 98.5 | 98.1 | 22.17 |
Experimental results of the feature set created by concatenating features from four architectures
| Dataset | Feature extractor | Classifier | Acc. (%) | SN (%) | SP (%) | PREC (%) | Time(s) |
|---|---|---|---|---|---|---|---|
| COVID‐19 image Dataset | AlexNet + EfficientNet‐b0 + NASNetLarge + Xception | LDA | 98.8 | 99.0 | 99.2 | 99.0 | 10.827 |
| SVM | 98.8 | 98.1 | 100 | 100 | 11.1 | ||
| KNN | 97.2 | 97.2 | 99.2 | 99.0 | 9.8717 | ||
| DT | 84.5 | 93.6 | 92.8 | 91.2 | 10.779 | ||
| NB | 98.0 | 97.2 | 100 | 100 | 28.869 | ||
| COVID‐19 Pneumonia Normal Chest X‐ray (PA) Dataset | AlexNet + EfficientNet‐b0 + NASNetLarge + Xception | LDA | 72.6 | 83.5 | 90.5 | 81.5 | 192.42 |
| SVM | 95.9 | 98.6 | 99.9 | 99.8 | 55.071 | ||
| KNN | 92.0 | 91.4 | 99.9 | 99.7 | 223.39 | ||
| DT | 84.8 | 86.5 | 93.9 | 87.6 | 82.697 | ||
| NB | 93.2 | 93.2 | 99.0 | 93.2 | 43.496 | ||
| COVID‐19 Radiography Database | AlexNet + EfficientNet‐b0 + NASNetLarge + Xception | LDA | 99.6 | 98.7 | 99.9 | 99.8 | 418.93 |
| SVM | 99.3 | 97.8 | 99.8 | 99.5 | 227.02 | ||
| KNN | 93.2 | 76.3 | 99.1 | 96.5 | 2195.9 | ||
| DT | 88.4 | 74.3 | 95.5 | 84.0 | 201.49 | ||
| NB | 91.1 | 92.5 | 94.8 | 84.9 | 102.12 |
FIGURE 9Confusion matrices of models that achieve the best results in (A) COVID‐19 image Dataset, (B) COVID‐19 Pneumonia Normal Chest X‐ray (PA) Dataset, and (C) COVID‐19 Radiography Database
Performance comparison of the proposed method with other approaches
| Ref. no/year | Images | Method | Dataset source | Performance | Class |
|---|---|---|---|---|---|
| 2020 | CT images | Self‐Trans | COVID‐CT |
0.85 F1 0.94 AUC |
COVID‐19 Non‐COVID‐19 |
| 2020 | CT images | DenseNet121‐FPN | From six cities in China | 0.86 AUC |
COVID‐19 Bacterial pneumonia Mycoplasma pneumonia Viral pneumonia Fungal pneumonia |
| 2020 | X‐ray images | CapsNet | COVID‐Chest X‐ray dataset |
Acc. 84.22 (Multi) Acc. 97.24 (Binary) |
COVID‐19 No‐findings Pneumonia |
| 2020 | CT images | Transfer learning on inception | Three hospitals from China |
0.93 AUC Specificity 88.0 Sensitivity 87.0 | Viral pneumonia |
| 2020 | X‐ray images | DarkNet |
COVID‐Chest X‐ray dataset Chest X‐ray |
Sensitivity 85.53 Acc. 87.02 |
SARS‐COV‐2 No finding Pneumonia |
| 2020 | X‐ray images | FBD + ResNet50 + Softmax | 750 Images |
Acc. 98.6% Specificity 94% Sensitivity 96% |
Pneumonia COVID‐19 Healthy |
| 2020 | CT images | ARENET (ResNet50 and FPN) | Hospitals of two provinces in China |
Precision 93% Recall 93% Acc. 93% |
COVID‐19 Bacterial pneumonia Healthy |
| 2021 | X‐ray images | ResNet34 | Collected from two sources | Acc. 98.33% |
Normal COVID‐19 |
| 2021 | X‐ray images | 22 layer CNN | COVID‐R |
Acc. 99.1% (2 class) Acc. 94.2% (3 class) Acc. 91.2% (4 class) |
COVID‐19 Normal Pneumonia (viral + bacteria) |
| 2021 | X‐ray images | Transfer learning on CNN | COVIDGR 1.0 | Acc. 90.45% |
COVID Non‐COVID |
| 2021 | X‐ray images | InstacovNet‐19 | 3150 images |
Acc. 99.53% (2 class) Acc. 99.08% (3 class) |
COVID‐19 Pneumonia Normal |
| 2020 | X‐ray images | COVID‐Net | COVIDx Dataset | Acc. 93.3% |
COVID Non‐COVID Normal |
| 2020 | X‐RAY images | Transfer learning on VGG19 | 1424 images |
Acc.98.75 (2 class) Acc. 93.48 (3 class) |
COVID‐19 Pneumonia Normal |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|