专栏名称: 学姐带你玩AI
这里有人工智能前沿信息、算法技术交流、机器学习/深度学习经验分享、AI大赛解析、大厂大咖算法面试分享、人工智能论文技巧、AI环境工具库教程等……学姐带你玩转AI!
目录
相关文章推荐
哎咆科技  ·  DeepSeek表明:未来10年可能是70- ... ·  11 小时前  
赛博禅心  ·  超级硬广:输入 ikun,解锁 ... ·  昨天  
赛博禅心  ·  超级硬广:输入 ikun,解锁 ... ·  昨天  
EETOP  ·  芯片巨头,人均年薪65万元! ·  昨天  
EETOP  ·  台媒:台积电是王昭君 ·  3 天前  
51好读  ›  专栏  ›  学姐带你玩AI

Mamba对UNet王炸升级!全新混合架构参数小了116倍,精度依旧完成超车

学姐带你玩AI  · 公众号  ·  · 2024-09-29 18:06

正文

UNet又升级了!北大最新提出LightM-UNet,用Mamba设计1.8M参数UNet,比nnU-Net小了116倍,计算量减少21倍,精度依然SOTA!

这种将 Mamba与UNet结合 的策略利用了两者在图像处理和网络架构上的优势,既能保持较低的计算成本,又能轻松实现优异的分割性能, 在高效性、准确性、创新性等方面都遥遥领先, 是医学图像分割非常有潜力的研究方向。

更赞的是,对于想发论文的同学来说,Mamba是新技术,还没那么卷,相关的 Mamba+UNet创新空间自然也大 ,还有很多优秀的成果给我们作参考,无疑是非常好的选择。

为了帮同学们抓紧机会,我已经挑选好了 12篇 Mamba结合UNet的高质量论文 供大家参考,全部都有代码,想速发论文拿下顶会的别错过啦。

扫码添加小享, 回复“ 曼巴分割

免费获取 全部论文+开源代码

Lightm-unet: Mamba assists in lightweight unet for medical image segmentation

方法: 论文介绍了LightM-UNet,这是一种基于Mamba的轻量级网络,它在2D和3D分割任务中实现了最先进的性能,仅包含100万个参数,与最新的基于Transformer的架构相比,参数数量降低了99%以上,并且GFLOPS也显著降低。

创新点:

  • LightM-UNet:基于Mamba的轻量级网络,仅包含1M参数,比最新的基于Transformer的架构参数减少了99%以上,并且计算负载显著较低。在2D和3D分割任务中,LightM-UNet实现了最先进的性能。
  • Mamba作为轻量级策略:该研究首次尝试将Mamba作为轻量级策略应用于UNet,以增强UNet对全局信息的理解能力。

Semi-mamba-unet: Pixel-level contrastive cross-supervised visual mamba-based unet for semi-supervised medical image segmentation

方法: 论文提出了一种结合Mamba架构和U-Net的半监督学习框架,名为Semi-Mamba-UNet。这个模型旨在提高医学图像分割的效率和准确性,尤其是在标注数据有限的情况下。

创新点:

  • 对医学图像分割中的Mamba架构进行了探索,这是首次使用有限标注探索Mamba架构。该架构通过使用U型编码器解码器网络中的Visual Mamba作为网络块,提高了医学图像分割的性能。
  • 引入了像素级对比学习策略,结合半监督学习,使用有标签和无标签数据最大限度地提高了特征学习能力。通过引入交叉监督学习,通过伪标签的方式增强了网络训练中无标签数据的可用性。

扫码添加小享, 回复“ 曼巴分割

免费获取 全部论文+开源代码

MSVM-UNet: Multi-Scale Vision Mamba UNet for Medical Image Segmentation

方法: 论文提出了一个用于医学图像分割的多尺度视觉状态空间模型(MSVM-UNet)。多尺度视觉Mamba UNet通过多尺度深度卷积的设计,不仅可以捕捉各种尺度的信息并模拟所有方向上的长程依赖性,还可以保持计算效率和可接受的参数数量。

创新点:

  • 通过在VSS块中引入多尺度卷积,可以更有效地捕捉和聚合来自VMamba编码器的分层特征表示,并更好地处理2D视觉数据。
  • 通过同时集成空间和通道信息,LKPE层实现了更高效的特征图上采样。使用我们提出的模块替换解码器中的原始模块后,显著改善了性能,只增加了最小的计算开销和参数数量。







请到「今天看啥」查看全文