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

Med-DANet V2:一种用于高效医学体积分割的灵活动态架构

Med-DANet V2: A Flexible Dynamic Architecture for Efficient Medical Volumetric Segmentation

University of Science and Technology Beijing(北京科技大学);University of Central Florida(中佛罗里达大学);Chinese Academy of Sciences(中国科学院)

Haoran Shen, Yifu Zhang, Wenxuan Wang, Chen Chen, Jing Liu, Shanshan Song, Jiangyun Li

AI总结:

本文提出Med-DANet V2,通过决策网络与裁剪位置网络动态选择前景区域,并在分割模型中引入逐阶段量化选择器,在BraTS上以更低复杂度实现与最优方法相当或更优的分割性能。

摘要:

近期研究表明,基于切片复杂度的动态推理可以显著提升3D医学图像(如CT和MRI)分割的计算效率。作为一项开创性工作,用于医学体积分割的动态架构网络(即Med-DANet)通过为不同切片从预定义模型库中动态选择合适的2D候选模型,实现了良好的精度与效率权衡。然而,Med-DANet中存在的数据分析不完整、训练成本高以及两阶段流程等问题仍需进一步改进。为此,本文从数据本身和模型结构两个角度进一步探索了动态推理框架的统一表述。对于输入体积的每个切片,我们提出的方法根据决策网络和裁剪位置网络生成的策略,动态选择一个重要的前景区域进行分割。此外,我们提出在所采用的分割模型(如U-Net)中插入一个逐阶段量化选择器,以实现动态架构自适应。在BraTS 2019和2020上的大量实验表明,我们的方法以更低的模型复杂度取得了与先前最先进方法相当或更好的性能。与分别采用动态和静态架构的先前方法Med-DANet和TransBTS相比,我们的框架在BraTS 2019上以相当的分割结果将模型效率提高了最多近4.1倍和17.3倍。

英文摘要:

Recent works have shown that the computational efficiency of 3D medical image (e.g. CT and MRI) segmentation can be impressively improved by dynamic inference based on slice-wise complexity. As a pioneering work, a dynamic architecture network for medical volumetric segmentation (i.e. Med-DANet) has achieved a favorable accuracy and efficiency trade-off by dynamically selecting a suitable 2D candidate model from the pre-defined model bank for different slices. However, the issues of incomplete data analysis, high training costs, and the two-stage pipeline in Med-DANet require further improvement. To this end, this paper further explores a unified formulation of the dynamic inference framework from the perspective of both the data itself and the model structure. For each slice of the input volume, our proposed method dynamically selects an important foreground region for segmentation based on the policy generated by our Decision Network and Crop Position Network. Besides, we propose to insert a stage-wise quantization selector to the employed segmentation model (e.g. U-Net) for dynamic architecture adapting. Extensive experiments on BraTS 2019 and 2020 show that our method achieves comparable or better performance than previous state-of-the-art methods with much less model complexity. Compared with previous methods Med-DANet and TransBTS with dynamic and static architecture respectively, our framework improves the model efficiency by up to nearly 4.1 and 17.3 times with comparable segmentation results on BraTS 2019.

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

↑