计算机科学与探索

北大核心,INSPEC,JST,CSCD,WJCI

国内刊号:11-5602/TP

国际刊号:1673-9418

计算机科学与探索杂志2023年第3期:改进的U-Net网络用于视网膜血管分割

发布日期:

作者:吕佳, 马超, 程超

单位:重庆师范大学 计算机与信息科学学院,重庆 401331

关键词:视网膜血管,提升算法,概率分布注意力,深监督非对称聚合

视网膜血管的形态结构信息可以为糖尿病、高血压等疾病提供诊断依据。现有的大部分算法通过融合血管不同阶段的特征信息来提升分割精度,却忽略了融合后浅层信息在采样过程中容易被稀释,且上采样次数越多,稀释越严重的问题。为此提出了一种融合概率分布注意力的提升算法用以解决上述问题。该算法基于U-Net模型,首先利用多个串行提升块学习血管浅层的特征信息,然后在提升块中引入概率分布注意力机制用来改变浅层信息的概率分布,提升浅层信息在最终分割任务中的贡献度,最后利用深度监督非对称聚合补充各提升块的梯度流,以避免模型在训练过程中产生梯度消失的问题。在DRIVE、STARE数据集上与近年来不同算法进行了对比,其中SP值分别为0.985 7、0.991 7,相较于第二名提升了0.2%、0.25%,均优于目前最新的算法结果。实验证明了提出的提升算法能较好地改善血管前期信息,一定程度上解决了信息稀释问题,提高了对视网膜血管的分割效果。

来源:2023年第3期

《计算机科学与探索》期刊编辑部

查看计算机科学与探索杂志2023年第3期

联系我们

  • 地址:北京市海淀区北四环中路211号
  • 电话:(010)89056056
  • E-mail:fcst@vip.163.com

咨询工作人员