LLM2D
UU-Mamba:面向心血管分割的不确定性感知U-Mamba
UU-Mamba: Uncertainty-aware U-Mamba for Cardiovascular Segmentation
发布日期: 9/24/2024
arXiv ID: oai:arXiv.org:2409.14305v1

摘要

基于深度学习模型在心血管结构分割领域的成功,越来越多的关注集中在提高泛化性和鲁棒性上,特别是在标注数据量较小的情况下。尽管近期有所进展,但当前方法仍面临过拟合和精度限制等挑战,主要原因是它们依赖于大规模数据集和狭窄的优化技术。本文介绍了UU-Mamba模型,它是U-Mamba架构的扩展,旨在解决心脏和血管分割中的这些挑战。通过整合锐度感知最小化(SAM),该模型通过瞄准损失景观中的平坦最小值来增强泛化能力。此外,我们提出了一种不确定性感知的损失函数,该函数结合了基于区域、基于分布和基于像素的组件,通过捕捉局部和全局特征来提高分割精度。尽管UU-Mamba模型已经展示了出色的性能,但仍需进一步测试以全面评估其泛化性和鲁棒性。我们通过在ImageCAS(冠状动脉)和Aorta(主动脉分支和区域)数据集上进行新试验来扩展评估,这些数据集比我们之前工作中使用的ACDC数据集(左心室和右心室)提出了更复杂的分割挑战,展示了模型的适应性和韧性。我们确认UU-Mamba在性能上优于TransUNet、Swin-Unet、nnUNet和nnFormer等领先模型。此外,我们通过广泛的实验提供了对模型鲁棒性和分割精度的更全面评估。