用四类CNN模型分析胸片,实现超97%的新冠判别准确率。
Evaluating Convolutional Neural Networks for COVID-19 classification in chest X-ray images
- 采用AlexNet、VGG-11等四类CNN模型自动识别新冠胸片特征。
- 在小样本数据下仍达97%以上准确率,SqueezeNet最优达99.20%。
- 适合医疗影像自动化筛查,尤其适用于数据稀缺场景。
2019冠状病毒病(COVID-19)全球快速传播,影响数十亿人生命。及时筛查感染者是遏制疫情的关键,而有效治疗需尽早发现。由于缺乏精准自动化工具,对高效可扩展方法的需求激增。近期研究显示,胸部X光片包含与新冠病毒相关的显著信息,结合机器学习与放射影像有望实现精准诊断。本文提出一种基于四种卷积神经网络(AlexNet、VGG-11、SqueezeNet、DenseNet-121)的自动新冠检测方法。实验采用十折交叉验证,在训练集和测试集上进行验证。研究发现,浅层微调与数据增强策略能有效应对公开阳性病例图像数量少的问题。所有CNN模型准确率均超过97.00%,其中SqueezeNet表现最佳,达到99.20%。
原文摘要 · Abstract (English)
Coronavirus Disease 2019 (COVID-19) pandemic rapidly spread globally, impacting the lives of billions of people. The effective screening of infected patients is a critical step to struggle with COVID-19, and treating the patients avoiding this quickly disease spread. The need for automated and scalable methods has increased due to the unavailability of accurate automated toolkits. Recent researches using chest X-ray images suggest they include relevant information about the COVID-19 virus. Hence, applying machine learning techniques combined with radiological imaging promises to identify this disease accurately. It is straightforward to collect these images once it is spreadly shared and analyzed in the world. This paper presents a method for automatic COVID-19 detection using chest Xray images through four convolutional neural networks, namely: AlexNet, VGG-11, SqueezeNet, and DenseNet-121. This method had been providing accurate diagnostics for positive or negative COVID-19 classification. We validate our experiments using a ten-fold cross-validation procedure over the training and test sets. Our findings include the shallow fine-tuning and data augmentation strategies that can assist in dealing with the low number of positive COVID-19 images publicly available. The accuracy for all CNNs is higher than 97.00%, and the SqueezeNet model achieved the best result with 99.20%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。