arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.07403cs.CVeess.IV

使其更具体:一种基于 3D U-Net 及其变体的新型不确定性气道分割应用

Make it more specific: A novel uncertainty based airway segmentation application on 3D U-Net and its variants

National Heart and Lung Institute, Imperial College London(帝国理工学院国家心肺研究所);Royal Brompton Hospital(皇家布朗普顿医院)

Shiyi Wang, Yang Nan, Felder Federico N, Sheng Zhang, Walsh Simon L F, Guang Yang

AI总结:

针对肺气管树状结构分割精度不足的问题,提出 B-UNet 和 B-CE-UNet 网络,引入分支与中心线损失及不确定性估计,并结合最大连通率后处理,有效提升了气道分割的准确性和可信度。

摘要:

每个医学分割任务都应根据其场景考虑使用特定的 AI 算法,以便获得最准确的预测模型。医学分割中最流行的算法 3D U-Net 及其变体可以直接实现肺气管分割任务,但其未能考虑气管特殊的树状结构,表明其分割精度还有很大的提升空间。因此,存在一个研究空白,因为大量最先进的 DL 算法是基础 3D U-Net 结构,在肺气道分割中没有引入特殊自然图像模态所附带的各种性能增强模块。在本文中,我们提出了两种基于 U-Net 结构的不同网络结构 Branch-Level U-Net (B-UNet) 和 Branch-Level CE-UNet (B-CE-UNet),并在相同数据集上比较了预测结果。特别地,这两个网络都添加了分支损失和中心线损失,以学习气道细枝末端特征。还包含了不确定性估计算法以获得有信心的预测,从而增加我们整个模型整体的可信度。此外,在后处理过程中计算并提取了基于最大连通率的肺气管预测,用于分割细化和修剪。

英文摘要:

Each medical segmentation task should be considered with a specific AI algorithm based on its scenario so that the most accurate prediction model can be obtained. The most popular algorithms in medical segmentation, 3D U-Net and its variants, can directly implement the task of lung trachea segmentation, but its failure to consider the special tree-like structure of the trachea suggests that there is much room for improvement in its segmentation accuracy. Therefore, a research gap exists because a great amount of state-of-the-art DL algorithms are vanilla 3D U-Net structures, which do not introduce the various performance-enhancing modules that come with special natural image modality in lung airway segmentation. In this paper, we proposed two different network structures Branch-Level U-Net (B-UNet) and Branch-Level CE-UNet (B-CE-UNet) which are based on U-Net structure and compared the prediction results with the same dataset. Specially, both of the two networks add branch loss and central line loss to learn the feature of fine branch endings of the airways. Uncertainty estimation algorithms are also included to attain confident predictions and thereby, increase the overall trustworthiness of our whole model. In addition, predictions of the lung trachea based on the maximum connectivity rate were calculated and extracted during post-processing for segmentation refinement and pruning.

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

↑