4 个月前

全尺度表示引导网络用于视网膜血管分割

全尺度表示引导网络用于视网膜血管分割

摘要

U-Net架构及其变体在过去十年中一直是视网膜血管分割领域的前沿技术(state-of-the-art, SOTA)。在本研究中,我们提出了一种全尺度引导网络(Full Scale Guided Network, FSG-Net),其中特征表示网络通过现代化的卷积块提取全尺度信息,而引导卷积块则对这些信息进行细化。引入了注意力引导滤波器,其工作原理类似于非锐化遮罩滤波器(unsharp mask filter)。将全尺度信息传递给注意力块,可以生成改进的注意力图,这些注意力图再传递给注意力引导滤波器,从而提升分割网络的性能。在引导卷积块之前的结构可以用任何U-Net变体替代,这增强了所提方法的可扩展性。为了进行公平比较,我们重新实现了公共存储库中的最新研究,以评估它们的可扩展性和可复现性。实验结果表明,所提出的网络在多个公开数据集上与当前SOTA模型相比具有竞争力。消融研究表明,所提模型在参数量显著减少的情况下仍具有竞争力。最后,我们将所提模型应用于面部皱纹分割任务,验证了其在其他类似任务中的潜在可扩展性。我们的代码已发布在https://github.com/ZombaSY/FSG-Net-pytorch。

代码仓库

zombasy/fsg-net-pytorch
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
retinal-vessel-segmentation-on-chase_db1FSG-Net
AUC: 0.9937
Acc: 0.9751
F1 score: 0.8101
MCC: 0.7989
Sensitivity: 0.8599
mIOU: 0.8268
retinal-vessel-segmentation-on-driveFSG-Net
AUC: 0.9823
Accuracy: 0.9704
F1 score: 0.8322
MCC: 0.8173
mIoU: 0.8406
sensitivity: 0.8420
retinal-vessel-segmentation-on-hrfFSG-Net
AUC: 0.9874
Acc: 0.9710
F1 score: 0.8156
MCC: 0.8012
Sensitivity: 0.8361
mIoU: 0.8308
retinal-vessel-segmentation-on-stareFSG-Net
AUC: 0.9896
Acc: 0.9774
F1 score: 0.8510
MCC: 0.8395
Sensitivity: 0.8660
mIOU: 0.8611

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
全尺度表示引导网络用于视网膜血管分割 | 论文 | HyperAI超神经