CT Image Classification Based on Stacked Ensemble Of Convolutional Neural Networks

Adai Shomanov, Dina Kuchenchirekova, Andrey Kurenkov, Min-Ho Lee

2022

Closed access Metadata (CSL-JSON)
DOI 10.1109/smc53654.2022.9945565
Data confidence 70 medium
How this score is calculated

It is built from checkable properties of the record itself: whether the article has a DOI, how many independent sources confirmed its data, whether the required fields are filled in, and how recent the latest confirmation is. Disagreement between sources and a missing identifier lower it. A value of 80 or above means the record was published without further checking; 60 to 79 means it is published at reduced confidence, with a manual check advisable. A record verified by an editor by hand is never pushed below 90 by a recalculation. This scores our data about the article, not the article itself.

Abstract

In a recent day, we could witness an explosive growth of artificial intelligence and deep learning in medical applications. With the increased availability of medical images, deep learning tools can provide a necessary diagnostic utility. However, current DNN models have shown high variations in their performance to each medical image datasets. In this study, we proposed ensemble learning to achieve synergistic improvements in model accuracy and thereby provide highly stabilized performance on diverse medical datasets. We first investigated the model performance of the latest deep learning architectures, e.g., Inception, VGGNet, MobileNet, Xception, ResNet50, and selected 7 state-of-the-art models to the diverse open CT datasets (SARS-COV-2 CT-Scan, USCD CT, and COVID-X dataset). The model parameters were transferred from the other domain and fine-tuned based on medical image sets. The last convolutional layers were stacked and a fully-connected neural network is employed to find generalized feature space. The peak accuracy of the fine-tuned single CNN models were InceptionV3 - 0.96, VGG16 - 0.94, VGG19 - 0.94, MobileNetV2 - 0.98, Xception - 0.9, ResNet - 0.96, DenseNet201 - 0.97. The proposed ensemble model achieves the peak accuracy of 0.99%, outperforming each individual model and achieving the highest performance in all three open CT datasets. Experimental results demonstrated that the proposed ensemble model is able to represent the hierarchical features and thereby it improves the stability and reproducibility of the classifier models.