提升心脑血管图像分割的泛化能力,有效应对数据不足与过拟合问题。
An Uncertainty-Aware Generalization Framework for Cardiovascular Image Segmentation
- 融合不确定性感知损失与平滑优化,增强模型对复杂结构的适应性。
- 在ImageCAS和Aorta数据集上优于TransUNet等主流模型,准确率显著提升。
- 适合医学图像分割研究者,尤其关注鲁棒性与小样本场景的应用。
深度学习在心脑血管结构分割中已取得显著进展,但其泛化能力和鲁棒性仍需提升。现有方法常因依赖大规模标注数据和有限优化策略而面临过拟合与精度不足的问题。本文提出UU-Mamba模型,基于U-Mamba架构,引入尖锐度感知最小化(SAM)以寻找损失曲面中更平坦的极小值,提升泛化性能;同时设计不确定性感知损失函数,结合区域、分布与像素级成分,更好地捕捉局部与全局特征。我们在ImageCAS(冠状动脉)和Aorta(主动脉分支与区域)数据集上进行扩展评估,这些数据集比以往研究使用的ACDC(左右心室)更具挑战性,验证了模型的适应性与鲁棒性。实验结果表明,UU-Mamba在多个指标上优于TransUNet、Swin-Unet、nnUNet和nnFormer等先进模型,并通过大量实验深入分析了其分割精度与稳定性。
原文摘要 · Abstract (English)
Deep learning models have achieved significant success in segmenting cardiovascular structures, but there is a growing need to improve their generalization and robustness. Current methods often face challenges such as overfitting and limited accuracy, largely due to their reliance on large annotated datasets and limited optimization techniques. This paper introduces the UU-Mamba model, an extension of the U-Mamba architecture, designed to address these challenges in both cardiac and vascular segmentation. By incorporating Sharpness-Aware Minimization (SAM), the model enhances generalization by seeking flatter minima in the loss landscape. Additionally, we propose an uncertainty-aware loss function that integrates region-based, distribution-based, and pixel-based components, improving segmentation accuracy by capturing both local and global features. We expand our evaluations on the ImageCAS (coronary artery) and Aorta (aortic branches and zones) datasets, which present more complex segmentation challenges than the ACDC dataset (left and right ventricles) used in prior work, showcasing the model's adaptability and resilience. Our results confirm UU-Mamba's superior performance compared to leading models such as TransUNet, Swin-Unet, nnUNet, and nnFormer. We also provide a more in-depth assessment of the model's robustness and segmentation accuracy through extensive experiments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。