Mishra Narendra Kumar, Singh Pushpendra, Joshi Shiv Dutt
Department of Electrical Engineering, Indian Institute of Technology Delhi, India.
Department of ECE, National Institute of Technology Hamirpur, India.
Biocybern Biomed Eng. 2021 Apr-Jun;41(2):572-588. doi: 10.1016/j.bbe.2021.04.006. Epub 2021 Apr 30.
Under the prevailing circumstances of the global pandemic of COVID-19, early diagnosis and accurate detection of COVID-19 through tests/screening and, subsequently, isolation of the infected people would be a proactive measure. Artificial intelligence (AI) based solutions, using Convolutional Neural Network (CNN) and exploiting the Deep Learning model's diagnostic capabilities, have been studied in this paper. Transfer Learning approach, based on VGG16 and ResNet50 architectures, has been used to develop an algorithm to detect COVID-19 from CT scan images consisting of Healthy (Normal), COVID-19, and Pneumonia categories. This paper adopts data augmentation and fine-tuning techniques to improve and optimize the VGG16 and ResNet50 model. Further, stratified 5-fold cross-validation has been conducted to test the robustness and effectiveness of the model. The proposed model performs exceptionally well in case of binary classification (COVID-19 vs. Normal) with an average classification accuracy of more than 99% in both VGG16 and ResNet50 based models. In multiclass classification (COVID-19 vs. Normal vs. Pneumonia), the proposed model achieves an average classification accuracy of 86.74% and 88.52% using VGG16 and ResNet50 architectures as baseline, respectively. Experimental results show that the proposed model achieves superior performance and can be used for automated detection of COVID-19 from CT scans.
在2019冠状病毒病全球大流行的当前情况下,通过检测/筛查尽早诊断并准确检测出2019冠状病毒病,随后隔离感染者将是一项积极主动的措施。本文研究了基于人工智能(AI)的解决方案,该方案使用卷积神经网络(CNN)并利用深度学习模型的诊断能力。基于VGG16和ResNet50架构的迁移学习方法已被用于开发一种算法,以从包含健康(正常)、2019冠状病毒病和肺炎类别的CT扫描图像中检测2019冠状病毒病。本文采用数据增强和微调技术来改进和优化VGG16和ResNet50模型。此外,还进行了分层5折交叉验证,以测试该模型的稳健性和有效性。在二元分类(2019冠状病毒病与正常)的情况下,所提出的模型表现出色,基于VGG16和ResNet50的模型的平均分类准确率均超过99%。在多类分类(2019冠状病毒病与正常与肺炎)中,所提出的模型分别以VGG16和ResNet50架构为基线,平均分类准确率达到86.74%和88.52%。实验结果表明,所提出的模型具有卓越的性能,可用于从CT扫描中自动检测2019冠状病毒病。