【魔改Mamba系列】UU-Mamba：用于图像分割的不确定性感知U-Mamba

文章预览

点击上方 “ 小白学视觉 ”，选择加" 星标 "或“ 置顶 ” 重磅干货，第一时间送达 论文信息 题目：UU-Mamba: Uncertainty-aware U-Mamba for Cardiac Image Segmentation UU-Mamba：用于心脏图像分割的不确定性感知U-Mamba 作者：Ting Yu Tsai，Li Lin，Shu Hu，Ming-Ching Chang，Hongtu Zhu，Xin Wang 源码：https://github.com/tiffany9056/UU-Mamba 论文创新点 不确定性感知损失函数 ：文章提出了一种新的 不确定性感知损失 函数，该函数整合了基于区域、基于分布和基于像素的损失，以提高分割精度和鲁棒性。这种设计允许模型优先考虑自信的预测并减少噪声数据的影响。 Sharpness-Aware Minimization (SAM) 优化器 ：作者将 SAM优化器 集成到模型中，该优化器通过寻找损失景观中的平坦最小值来增强泛化能力，减少过拟合，并提高模型对输入数据小扰动的鲁棒性。 UU-Mamba模型架构 ：文章介绍了 UU-Mamba模型 ………………………………