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

COVID TV-UNet:使用连接性强加的U-Net分割COVID-19胸部CT图像

COVID TV-UNet: Segmenting COVID-19 Chest CT Images Using Connectivity Imposed U-Net

Medical Image and Signal Processing Research Center, Isfahan University of Medical Sciences(伊斯法罕医科大学医学图像与信号处理研究中心);Snap Inc.(Snap公司);ECE Department, Isfahan University of Technology(伊斯法罕理工大学电子与计算机工程系);Iowa Institute for Biomedical Imaging, The University of Iowa(爱荷华大学生物医学成像研究所)

Narges Saeedizadeh, Shervin Minaee, Rahele Kafieh, Shakib Yazdani, Milan Sonka

摘要:

新型冠状病毒病(COVID-19)大流行已在全球200多个国家引发大规模疫情,对全球许多人的健康和生活造成严重影响。截至2020年7月中旬,超过1200万人感染,超过57万人死亡。计算机断层扫描(CT)图像可用作耗时的RT-PCR检测的替代方法,以检测COVID-19。在这项工作中,我们提出了一个分割框架来检测CT图像中的胸部区域,这些区域被COVID-19感染。我们使用一个类似于U-Net模型的架构,并训练它来检测毛玻璃区域,在像素级。由于感染区域倾向于形成连通分量(而不是随机分布的像素),因此我们将适当的正则化项添加到损失函数中,以促进COVID-19像素的分割图的连通性。2D各向异性全变差用于此目的,因此所提出的模型被称为“TV-UNet”。通过对大约900幅图像的相对大规模CT分割数据集的实验结果,我们表明,与U-Net模型相比,添加这个新的正则化项会导致整体分割性能提高2%。我们的实验分析,从预测分割结果的视觉评估到分割性能的定量评估(精确度,召回率,Dice评分和mIoU),证明了识别COVID-19相关肺部区域的强大能力,实现了超过99%的mIoU率,Dice评分约为86%。

英文摘要:

The novel corona-virus disease (COVID-19) pandemic has caused a major outbreak in more than 200 countries around the world, leading to a severe impact on the health and life of many people globally. As of mid-July 2020, more than 12 million people were infected, and more than 570,000 death were reported. Computed Tomography (CT) images can be used as an alternative to the time-consuming RT-PCR test, to detect COVID-19. In this work we propose a segmentation framework to detect chest regions in CT images, which are infected by COVID-19. We use an architecture similar to U-Net model, and train it to detect ground glass regions, on pixel level. As the infected regions tend to form a connected component (rather than randomly distributed pixels), we add a suitable regularization term to the loss function, to promote connectivity of the segmentation map for COVID-19 pixels. 2D-anisotropic total-variation is used for this purpose, and therefore the proposed model is called "TV-UNet". Through experimental results on a relatively large-scale CT segmentation dataset of around 900 images, we show that adding this new regularization term leads to 2\% gain on overall segmentation performance compared to the U-Net model. Our experimental analysis, ranging from visual evaluation of the predicted segmentation results to quantitative assessment of segmentation performance (precision, recall, Dice score, and mIoU) demonstrated great ability to identify COVID-19 associated regions of the lungs, achieving a mIoU rate of over 99\%, and a Dice score of around 86\%.

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

↑