电气工程与系统科学 > 图像与视频处理
[提交于 2025年1月5日
]
标题: KM-UNet KAN Mamba UNet 用于医学图像分割
标题: KM-UNet KAN Mamba UNet for medical image segmentation
摘要: 医学图像分割是医学影像分析中的关键任务。传统的基于CNN的方法在建模长距离依赖关系方面存在困难,而基于Transformer的模型尽管取得了成功,但面临着二次计算复杂度的问题。为了解决这些限制,我们提出了KM-UNet,这是一种结合了Kolmogorov-Arnold网络(KANs)和状态空间模型(SSMs)优势的新型U型网络架构。KM-UNet利用Kolmogorov-Arnold表示定理进行高效的特征表示,并利用状态空间模型进行可扩展的长距离建模,实现了准确性和计算效率之间的平衡。我们在五个基准数据集上评估了KM-UNet:ISIC17、ISIC18、CVC、BUSI和GLAS。实验结果表明,与最先进的方法相比,KM-UNet在医学图像分割任务中表现出具有竞争力的性能。据我们所知,KM-UNet是第一个整合KANs和SSMs的医学图像分割框架。这项工作为开发更高效和可解释的医学图像分割系统提供了有价值的基线和新的见解。代码在https://github.com/2760613195/KM_UNet开源。 关键词:KAN,Manba,状态空间模型,UNet,医学图像分割,深度学习
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.