基于空间金字塔池化模块的残差洗牌网络用于新冠病毒肺炎筛查

Residual-Shuffle Network with Spatial Pyramid Pooling Module for COVID-19 Screening.

作者信息

Zulkifley Mohd Asyraf, Abdani Siti Raihanah, Zulkifley Nuraisyah Hani, Shahrimin Mohamad Ibrani

机构信息

Department of Electrical, Electronic and Systems Engineering, Faculty of Engineering and Built Environment, Universiti Kebangsaan Malaysia, Bangi 43600, Selangor, Malaysia.

Faculty of Humanities, Management and Science, Universiti Putra Malaysia Bintulu Campus, Bintulu 97008, Sarawak, Malaysia.

出版信息

Diagnostics (Basel). 2021 Aug 19;11(8):1497. doi: 10.3390/diagnostics11081497.

Abstract

Since the start of the COVID-19 pandemic at the end of 2019, more than 170 million patients have been infected with the virus that has resulted in more than 3.8 million deaths all over the world. This disease is easily spreadable from one person to another even with minimal contact, even more for the latest mutations that are more deadly than its predecessor. Hence, COVID-19 needs to be diagnosed as early as possible to minimize the risk of spreading among the community. However, the laboratory results on the approved diagnosis method by the World Health Organization, the reverse transcription-polymerase chain reaction test, takes around a day to be processed, where a longer period is observed in the developing countries. Therefore, a fast screening method that is based on existing facilities should be developed to complement this diagnosis test, so that a suspected patient can be isolated in a quarantine center. In line with this motivation, deep learning techniques were explored to provide an automated COVID-19 screening system based on X-ray imaging. This imaging modality is chosen because of its low-cost procedures that are widely available even in many small clinics. A new convolutional neural network (CNN) model is proposed instead of utilizing pre-trained networks of the existing models. The proposed network, Residual-Shuffle-Net, comprises four stacks of the residual-shuffle unit followed by a spatial pyramid pooling (SPP) unit. The architecture of the residual-shuffle unit follows an hourglass design with reduced convolution filter size in the middle layer, where a shuffle operation is performed right after the split branches have been concatenated back. Shuffle operation forces the network to learn multiple sets of features relationship across various channels instead of a set of global features. The SPP unit, which is placed at the end of the network, allows the model to learn multi-scale features that are crucial to distinguish between the COVID-19 and other types of pneumonia cases. The proposed network is benchmarked with 12 other state-of-the-art CNN models that have been designed and tuned specially for COVID-19 detection. The experimental results show that the Residual-Shuffle-Net produced the best performance in terms of accuracy and specificity metrics with 0.97390 and 0.98695, respectively. The model is also considered as a lightweight model with slightly more than 2 million parameters, which makes it suitable for mobile-based applications. For future work, an attention mechanism can be integrated to target certain regions of interest in the X-ray images that are deemed to be more informative for COVID-19 diagnosis.

摘要

自2019年底新冠疫情爆发以来,全球已有超过1.7亿人感染该病毒,导致超过380万人死亡。这种疾病即使在极少接触的情况下也很容易在人与人之间传播,对于比其前身更致命的最新变种更是如此。因此,需要尽早诊断新冠病毒,以尽量减少在社区中传播的风险。然而,世界卫生组织批准的诊断方法——逆转录聚合酶链反应检测的实验室结果需要大约一天时间才能得出,在发展中国家所需时间更长。因此,应开发一种基于现有设备的快速筛查方法来补充这种诊断测试,以便能够将疑似患者隔离在检疫中心。出于这一动机,人们探索了深度学习技术,以提供一种基于X光成像的新冠病毒自动筛查系统。选择这种成像方式是因为其成本低廉,即使在许多小诊所也广泛可用。提出了一种新的卷积神经网络(CNN)模型,而不是使用现有模型的预训练网络。所提出的网络,即残差洗牌网络(Residual-Shuffle-Net),由四组残差洗牌单元组成,后面跟着一个空间金字塔池化(SPP)单元。残差洗牌单元的架构采用沙漏设计,中间层的卷积滤波器尺寸减小,在分支拼接回来后紧接着进行洗牌操作。洗牌操作迫使网络学习跨不同通道的多组特征关系,而不是一组全局特征。位于网络末尾的SPP单元允许模型学习多尺度特征,这对于区分新冠病毒感染和其他类型的肺炎病例至关重要。所提出的网络与其他12种专门为新冠病毒检测设计和调整的先进CNN模型进行了基准测试。实验结果表明,残差洗牌网络在准确率和特异性指标方面表现最佳,分别为0.97390和0.98695。该模型也被认为是一个轻量级模型,参数略多于200万,这使其适用于基于移动设备的应用。对于未来的工作,可以集成注意力机制,以针对X光图像中被认为对新冠病毒诊断更具信息性的某些感兴趣区域。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/c887/8394651/2a65697e8315/diagnostics-11-01497-g001.jpg

本文引用的文献

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索