计算机科学 > 计算机视觉与模式识别
[提交于 2025年9月15日
]
标题: SA-UNetv2:重新思考空间注意力U-Net用于视网膜血管分割
标题: SA-UNetv2: Rethinking Spatial Attention U-Net for Retinal Vessel Segmentation
摘要: 视网膜血管分割对于糖尿病视网膜病变、高血压和神经退行性疾病的早期诊断至关重要。 尽管SA-UNet在瓶颈层引入了空间注意力,但它在跳跃连接中未充分利用注意力,并未解决严重的前景-背景不平衡问题。 我们提出了SA-UNetv2,这是一种轻量级模型,将跨尺度空间注意力注入所有跳跃连接,以增强多尺度特征融合,并采用加权二元交叉熵(BCE)加上马修斯相关系数(MCC)损失来提高对类别不平衡的鲁棒性。 在公开的DRIVE和STARE数据集上,SA-UNetv2仅需1.2MB内存和0.26M参数(少于SA-UNet的50%),并在592 x 592 x 3图像上实现1秒的CPU推理,证明了其在资源受限、仅CPU设置下的强大效率和可部署性。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.