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

面向胸部解剖分割的3D U型深度学习模型架构分析与基准测试

Architecture Analysis and Benchmarking of 3D U-shaped Deep Learning Models for Thoracic Anatomical Segmentation

Concordia University(康考迪亚大学)

Arash Harirpoush, Amirhossein Rasoulian, Marta Kersten-Oertel, Yiming Xiao

AI总结:

该研究首次针对多种3D U型深度学习模型开展胸部CT解剖分割系统基准测试,分析注意力机制、网络配置等对性能的影响,验证了CNN类U型模型及残差块的价值,STUNet表现最优。

摘要:

当前针对患者个性化胸部手术规划与模拟的研究兴趣日益增长,这要求通过自动化医学图像分割算法高效、鲁棒地构建数字解剖模型。深度学习(DL)目前是各类放射学任务中的最优技术,自2D UNet问世以来,U型深度学习模型在医学图像分割领域表现尤为突出。迄今为止,研究人员通过整合不同的注意力机制与网络配置,提出了多种U型模型变体。利用多标签数据库的最新发展对这些模型的架构进行分析的系统性基准研究,能够为临床部署和未来模型设计提供宝贵见解,但此类研究目前仍较为稀缺。\n我们开展了首项针对3D U型模型变体的系统性基准研究,涉及的模型包括3DUNet、STUNet、AttentionUNet、SwinUNETR、FocalSegNet,以及一种带有四个变体的新型3D SwinUnet,研究聚焦于基于CT的胸外科解剖分割任务。本研究系统探究了不同注意力机制、分辨率阶段数量以及网络配置对分割精度和计算复杂度的影响。为了能够与其他近期基准研究进行交叉参照,我们还纳入了BTCV腹部结构分割任务的性能评估。研究结果显示STUNet排名第一,证明了基于CNN的U型模型在上述研究任务中的价值,以及网络配置设计中残差块对提升分割性能的作用。

英文摘要:

Recent rising interests in patient-specific thoracic surgical planning and simulation require efficient and robust creation of digital anatomical models from automatic medical image segmentation algorithms. Deep learning (DL) is now state-of-the-art in various radiological tasks, and U-shaped DL models have particularly excelled in medical image segmentation since the inception of the 2D UNet. To date, many variants of U-shaped models have been proposed by the integration of different attention mechanisms and network configurations. Systematic benchmark studies which analyze the architecture of these models by leveraging the recent development of the multi-label databases, can provide valuable insights for clinical deployment and future model designs, but such studies are still rare. We conduct the first systematic benchmark study for variants of 3D U-shaped models (3DUNet, STUNet, AttentionUNet, SwinUNETR, FocalSegNet, and a novel 3D SwinUnet with four variants) with a focus on CT-based anatomical segmentation for thoracic surgery. Our study systematically examines the impact of different attention mechanisms, the number of resolution stages, and network configurations on segmentation accuracy and computational complexity. To allow cross-reference with other recent benchmarking studies, we also included a performance assessment of the BTCV abdominal structural segmentation. With the STUNet ranking at the top, our study demonstrated the value of CNN-based U-shaped models for the investigated tasks and the benefit of residual blocks in network configuration designs to boost segmentation performance.

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

↑