中文

COVID-ResNet:一种用于胸片筛查 COVID19 的深度学习框架

图像与视频处理 2020-04-01 v1 计算机视觉与模式识别 机器学习

摘要

近几个月来,新型 COVID19 大流行已蔓延至全球。由于其易传播性,开发能够准确且简便地识别 COVID19 存在并将其与其他类型流感和肺炎区分的技术至关重要。近期研究表明,COVID19 患者的胸部 X 光片在放射影像中表现出某些异常。然而,这些方法均为闭源,未向研究界公开以实现可复现性与更深入的洞察。本工作的目标是构建开源且开放获取的数据集,并提供一种准确的卷积神经网络框架,用于区分 COVID19 病例与其他肺炎病例。我们的工作利用最先进的训练技术,包括渐进式重缩放、周期性学习率寻找和判别性学习率,以快速且准确地训练残差神经网络。利用这些技术,我们在开放获取的 COVID-19 数据集上展示了最先进的结果。本工作提出一种三步技术,对预训练的 ResNet-50 架构进行微调,以提升模型性能并减少训练时间。我们称之为 COVIDResNet。这通过逐步将输入图像重缩放为 128x128x3、224x224x3 和 229x229x3 像素并在每个阶段微调网络来实现。该方法结合自动学习率选择,使我们仅用 41 个 epoch 即在 COVIDx 数据集上(所有类别)达到 96.23% 的最先进准确率。本工作提出了一种计算高效且高精度的模型,用于对三类不同感染类型及正常个体进行多分类。该模型可助力 COVID19 病例的早期筛查,并帮助减轻医疗系统负担。

关键词

引用

@article{arxiv.2003.14395,
  title  = {COVID-ResNet: A Deep Learning Framework for Screening of COVID19 from Radiographs},
  author = {Muhammad Farooq and Abdul Hafeez},
  journal= {arXiv preprint arXiv:2003.14395},
  year   = {2020}
}

备注

6 pages, 3 Figures,