arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.06412eess.IVcs.LG

CT扫描中肺部和COVID-19病变分割的深度学习模型综合比较

Comprehensive Comparison of Deep Learning Models for Lung and COVID-19 Lesion Segmentation in CT scans

Visual Computing Lab of the Information Technologies Institute, Centre for Research and Technology Hellas(希腊研究与技术中心信息技术研究所视觉计算实验室)

Paschalis Bizopoulos, Nicholas Vretos, Petros Daras

AI总结:

本文综合比较了200种深度学习模型在CT扫描中肺部和COVID-19病变分割的性能,确定了最佳架构-编码器组合,并量化了预训练和掩膜预处理的上限改进,提供了基准和预训练模型资源。

摘要:

近年来,深度学习(DL)方法在医学图像分割中的应用呈爆炸式增长。然而,该领域的可靠性受到缺乏统一的准确性/性能评估参考基准以及以往研究使用不同数据集进行评估这一事实的阻碍。本文对计算机断层扫描(CT)扫描中的肺部和COVID-19病变分割的深度学习模型进行了广泛比较,该比较也可作为测试医学图像分割模型的基准。四种深度学习架构(Unet、Linknet、FPN、PSPNet)与25个随机初始化和预训练的编码器(VGG、DenseNet、ResNet、ResNext、DPN、MobileNet、Xception、Inception-v4、EfficientNet的变体)相结合,构建了200个测试模型。进行了三个实验设置:肺部分割、病变分割和使用原始肺部掩膜进行病变分割。使用包含100张CT扫描图像(80张用于训练,20张用于验证)的公共COVID-19数据集进行训练/验证,并使用另一个包含来自9个CT扫描体积的829张图像的公共数据集进行测试。提供了多项发现,包括每个实验的最佳架构-编码器模型,以及每个实验、架构和编码器独立的平均Dice结果。最后,量化了使用肺部掩膜作为预处理步骤或使用预训练模型时的上限改进。提供了三个实验的源代码和600个预训练模型,适用于无GPU能力的实验环境中的微调。

英文摘要:

Recently there has been an explosion in the use of Deep Learning (DL) methods for medical image segmentation. However the field's reliability is hindered by the lack of a common base of reference for accuracy/performance evaluation and the fact that previous research uses different datasets for evaluation. In this paper, an extensive comparison of DL models for lung and COVID-19 lesion segmentation in Computerized Tomography (CT) scans is presented, which can also be used as a benchmark for testing medical image segmentation models. Four DL architectures (Unet, Linknet, FPN, PSPNet) are combined with 25 randomly initialized and pretrained encoders (variations of VGG, DenseNet, ResNet, ResNext, DPN, MobileNet, Xception, Inception-v4, EfficientNet), to construct 200 tested models. Three experimental setups are conducted for lung segmentation, lesion segmentation and lesion segmentation using the original lung masks. A public COVID-19 dataset with 100 CT scan images (80 for train, 20 for validation) is used for training/validation and a different public dataset consisting of 829 images from 9 CT scan volumes for testing. Multiple findings are provided including the best architecture-encoder models for each experiment as well as mean Dice results for each experiment, architecture and encoder independently. Finally, the upper bounds improvements when using lung masks as a preprocessing step or when using pretrained models are quantified. The source code and 600 pretrained models for the three experiments are provided, suitable for fine-tuning in experimental setups without GPU capabilities.

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

↑