Nayak Soumya Ranjan, Nayak Janmenjoy, Sinha Utkarsh, Arora Vaibhav, Ghosh Uttam, Satapathy Suresh Chandra
Amity School of Engineering and Technology, Amity University Uttar Pradesh, Noida, India.
Department of Computer Science and Engineering, Aditya Institute of Technology and Management (AITAM), Tekkali, K Kotturu, AP 532201 India.
Arab J Sci Eng. 2021 Aug 9:1-18. doi: 10.1007/s13369-021-05956-2.
Coronavirus (COVID-19) is an epidemic that is rapidly spreading and causing a severe healthcare crisis resulting in more than 40 million confirmed cases across the globe. There are many intensive studies on AI-based technique, which is time consuming and troublesome by considering heavyweight models in terms of more training parameters and memory cost, which leads to higher time complexity. To improve diagnosis, this paper is aimed to design and establish a unique lightweight deep learning-based approach to perform multi-class classification (normal, COVID-19, and pneumonia) and binary class classification (normal and COVID-19) on X-ray radiographs of chest. This proposed CNN scheme includes the combination of three CBR blocks (convolutional batch normalization ReLu) with learnable parameters and one global average pooling (GP) layer and fully connected layer. The overall accuracy of the proposed model achieved 98.33% and finally compared with the pre-trained transfer learning model (DenseNet-121, ResNet-101, VGG-19, and XceptionNet) and recent state-of-the-art model. For validation of the proposed model, several parameters are considered such as learning rate, batch size, number of epochs, and different optimizers. Apart from this, several other performance measures like tenfold cross-validation, confusion matrix, evaluation metrics, sarea under the receiver operating characteristics, kappa score and Mathew's correlation, and Grad-CAM heat map have been used to assess the efficacy of the proposed model. The outcome of this proposed model is more robust, and it may be useful for radiologists for faster diagnostics of COVID-19.
冠状病毒(COVID-19)是一种正在迅速传播并引发严重医疗危机的流行病,全球确诊病例超过4000万例。目前有许多关于基于人工智能技术的深入研究,然而考虑到重量级模型具有更多的训练参数和内存成本,这类研究既耗时又麻烦,这导致了更高的时间复杂度。为了改善诊断效果,本文旨在设计并建立一种独特的基于轻量级深度学习的方法,用于对胸部X光片进行多类分类(正常、COVID-19和肺炎)以及二分类(正常和COVID-19)。所提出的卷积神经网络(CNN)方案包括三个带有可学习参数的卷积批归一化整流线性单元(CBR)块、一个全局平均池化(GP)层和全连接层的组合。所提出模型的总体准确率达到了98.33%,最后与预训练的迁移学习模型(DenseNet-121、ResNet-101、VGG-19和XceptionNet)以及近期的先进模型进行了比较。为了验证所提出的模型,考虑了几个参数,如学习率、批量大小、轮数以及不同的优化器。除此之外,还使用了其他几种性能指标,如十折交叉验证、混淆矩阵、评估指标、受试者工作特征曲线下面积、kappa分数和马修斯相关系数,以及梯度加权类激活映射(Grad-CAM)热图来评估所提出模型的有效性。所提出模型的结果更加稳健,可能对放射科医生快速诊断COVID-19有用。