电气工程与系统科学 > 图像与视频处理
[提交于 2025年7月15日
]
标题: U-RWKV:具有方向自适应RWKV的轻量级医学图像分割
标题: U-RWKV: Lightweight medical image segmentation with direction-adaptive RWKV
摘要: 实现医疗保健可及性的公平性需要轻量级且高性能的解决方案用于医学图像分割,特别是在资源有限的环境中。 现有的方法如U-Net及其变体通常受到有限的全局有效感受野(ERFs)的困扰,这阻碍了它们捕捉长距离依赖关系的能力。 为了解决这个问题,我们提出了U-RWKV,这是一种利用循环加权键值(RWKV)架构的新框架,能够在O(N)计算成本下实现高效的长距离建模。 该框架引入了两个关键创新:方向自适应RWKV模块(DARM)和阶段自适应压缩与激励模块(SASE)。 DARM采用双-RWKV和四扫描机制来跨图像聚合上下文线索,在减轻方向偏差的同时保留全局上下文并保持高计算效率。 SASE动态适应不同的特征提取阶段,平衡高分辨率细节的保留和语义关系的捕捉。 实验表明,U-RWKV在计算效率高的情况下实现了最先进的分割性能,为在资源受限环境中普及先进的医学成像技术提供了实用的解决方案。 代码可在https://github.com/hbyecoding/U-RWKV获取。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.