arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.14395eess.IVcs.CVcs.LG
收藏

COVID-ResNet:一个用于从射线照片中筛选COVID19的深度学习框架

COVID-ResNet: A Deep Learning Framework for Screening of COVID19 from Radiographs

University of Alabama(阿拉巴马大学);University of Engineering & Technology, Peshawar(白沙瓦工程与技术大学)

Muhammad Farooq, Abdul Hafeez

摘要:

在过去的几个月里,新型COVID 19大流行已经蔓延到世界各地。由于COVID 19易于传播,因此开发技术来准确、轻松地识别COVID 19的存在并将其与其他形式的流感和肺炎区分开来至关重要。最近的研究表明,患有COVID 19的患者的胸部X光片描绘了放射摄影的某些异常。然而,这些方法都是封闭源代码的,不能提供给研究界进行复制和获得更深入的见解。这项工作的目标是构建开源和开放获取的数据集,并提出一个准确的卷积神经网络框架,用于区分COVID 19病例和其他肺炎病例。我们的工作利用了最先进的训练技术,包括渐进式学习、循环学习率发现和判别学习率,来训练快速准确的残差神经网络。使用这些技术,我们在开放获取的COVID-19数据集上展示了最先进的结果。这项工作提出了一种3步技术来微调预训练的ResNet-50架构,以提高模型性能并减少训练时间。我们称之为COVIDResNet。这是通过逐步将输入图像大小调整为128 x128 x3,224 x224 x3和229 x229 x3像素并在每个阶段微调网络来实现的。这种方法以及自动学习率选择使我们能够在COVIDx数据集上实现96.23%的最新准确度(所有类别),只有41个epoch。这项工作提出了一个计算效率高,精度高的模型,用于对三种不同的感染类型和正常个体进行多类分类。这种模式可以帮助早期筛查COVID 19病例,并有助于减轻医疗系统的负担。

英文摘要:

In the last few months, the novel COVID19 pandemic has spread all over the world. Due to its easy transmission, developing techniques to accurately and easily identify the presence of COVID19 and distinguish it from other forms of flu and pneumonia is crucial. Recent research has shown that the chest Xrays of patients suffering from COVID19 depicts certain abnormalities in the radiography. However, those approaches are closed source and not made available to the research community for re-producibility and gaining deeper insight. The goal of this work is to build open source and open access datasets and present an accurate Convolutional Neural Network framework for differentiating COVID19 cases from other pneumonia cases. Our work utilizes state of the art training techniques including progressive resizing, cyclical learning rate finding and discriminative learning rates to training fast and accurate residual neural networks. Using these techniques, we showed the state of the art results on the open-access COVID-19 dataset. This work presents a 3-step technique to fine-tune a pre-trained ResNet-50 architecture to improve model performance and reduce training time. We call it COVIDResNet. This is achieved through progressively re-sizing of input images to 128x128x3, 224x224x3, and 229x229x3 pixels and fine-tuning the network at each stage. This approach along with the automatic learning rate selection enabled us to achieve the state of the art accuracy of 96.23% (on all the classes) on the COVIDx dataset with only 41 epochs. This work presented a computationally efficient and highly accurate model for multi-class classification of three different infection types from along with Normal individuals. This model can help in the early screening of COVID19 cases and help reduce the burden on healthcare systems.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑