论文标题

质量或数量:迈向人体CT中多器官分割的统一方法

Quality or Quantity: Toward a Unified Approach for Multi-organ Segmentation in Body CT

论文作者

Tushar, Fakrul Islam, Nujaim, Husam, Fu, Wanyi, Abadi, Ehsan, Mazurowski, Maciej A., Samei, Ehsan, Segars, William P., Lo, Joseph Y.

论文摘要

医学图像的器官分割是虚拟成像试验中的关键步骤。但是,器官细分数据集在质量方面受到限制(因为标签仅覆盖了几个器官)和数量(因为案例数是有限的)。在这项研究中,我们探讨了质量和数量之间的权衡。我们的目标是为人体CT的多器官分割创建一种统一的方法,这将有助于创建大量准确的虚拟幻像。最初,我们比较了两个分割架构,即3D-UNET和密集vnet,它们是使用XCAT数据完全标记为22个器官的XCAT数据训练的,并选择了3D-UNET作为表现更好的模型。我们使用XCAT训练的模型来生成仅有7个器官分割的CT-ORG数据集的伪标记。我们执行了两个实验:首先,我们在XCAT数据集上训练了3D-UNET模型,代表质量数据,并在XCAT和CT-ORG数据集上进行了测试。其次,在将CT-ORG数据集(包括更多数量的培训设置)中培训了3D-UNET。性能改善了在器官中的分割,在该器官中,我们在数据集中都具有真正的标签并在依靠伪标签时降解。当两个数据集中标记器官时,EXP-2提高了XCAT和CT-ORG中的平均DSC。这证明了质量数据是改善模型性能的关键。

Organ segmentation of medical images is a key step in virtual imaging trials. However, organ segmentation datasets are limited in terms of quality (because labels cover only a few organs) and quantity (since case numbers are limited). In this study, we explored the tradeoffs between quality and quantity. Our goal is to create a unified approach for multi-organ segmentation of body CT, which will facilitate the creation of large numbers of accurate virtual phantoms. Initially, we compared two segmentation architectures, 3D-Unet and DenseVNet, which were trained using XCAT data that is fully labeled with 22 organs, and chose the 3D-Unet as the better performing model. We used the XCAT-trained model to generate pseudo-labels for the CT-ORG dataset that has only 7 organs segmented. We performed two experiments: First, we trained 3D-UNet model on the XCAT dataset, representing quality data, and tested it on both XCAT and CT-ORG datasets. Second, we trained 3D-UNet after including the CT-ORG dataset into the training set to have more quantity. Performance improved for segmentation in the organs where we have true labels in both datasets and degraded when relying on pseudo-labels. When organs were labeled in both datasets, Exp-2 improved Average DSC in XCAT and CT-ORG by 1. This demonstrates that quality data is the key to improving the model's performance.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源