高级检索
梁礼明, 冯骏, 彭仁杰, 曾嵩. 融合多标签损失与双注意力的U型视网膜血管分割[J]. 计算机辅助设计与图形学学报, 2023, 35(1): 75-86. DOI: 10.3724/SP.J.1089.2023.19257
引用本文: 梁礼明, 冯骏, 彭仁杰, 曾嵩. 融合多标签损失与双注意力的U型视网膜血管分割[J]. 计算机辅助设计与图形学学报, 2023, 35(1): 75-86. DOI: 10.3724/SP.J.1089.2023.19257
LIANG Li-ming, FENG Jun, PENG Ren-jie, CENG Song. U-Shaped Retinal Vessel Segmentation Combining Multi-Label Loss and Dual Attention[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(1): 75-86. DOI: 10.3724/SP.J.1089.2023.19257
Citation: LIANG Li-ming, FENG Jun, PENG Ren-jie, CENG Song. U-Shaped Retinal Vessel Segmentation Combining Multi-Label Loss and Dual Attention[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(1): 75-86. DOI: 10.3724/SP.J.1089.2023.19257

融合多标签损失与双注意力的U型视网膜血管分割

U-Shaped Retinal Vessel Segmentation Combining Multi-Label Loss and Dual Attention

  • 摘要: 眼底视网膜血管的检测与分析对许多眼科疾病的诊断具有重要意义.为了更精确、健全地提取视网膜血管的特征信息,提出一种融合多标签损失与双注意力的U型网络模型.首先在编码部分通过空间金字塔池化提供多尺度输入,在U型网络内部融入双注意残差块提升网络对特征信息的提取能力;其次,在网络底部嵌入特征相似模块以捕获特征之间的远程依赖关系,为了有效地抑制眼底图像中的噪声影响和捕获血管多尺度信息,在跳连部分分别引入双路径注意门机制与稠密的空洞空间金字塔池化模块;最后,在解码部分设置侧输出层生成与层级对应的局部预测图像,并配合多标签Dice损失函数进行训练.在DRIVE,STARE和CHASE_DB1数据集上进行实验,灵敏度分别为80.54%,83.97%和82.40%,受试者曲线下的面积(AUC)分别为98.07%,98.50%和98.36%.

     

    Abstract: The detection and analysis of fundus retinal blood vessels is of great significance for the diagnosis of many ophthalmological diseases. To extract retinal blood vessels more accurately and soundly, we propose a U-shaped network model combining multi-label loss and dual attention. Firstly, we introduce spatial pyramid pooling in the encoder part to provide multi-scale input, and incorporate dual attention residual block into the U-shaped network to improve the network's ability to extract feature information.Secondly, the feature similarity module is embedded at the bottom of the network to capture the long-range dependency between features, To effectively suppress the influence of noise in the fundus image and capture the multi-scale information of the blood vessels, the dual-pathway attention gates mechanism and densely connected atrous spatial pyramid pooling(DenseASPP) module are introduced in the skip connection part.Finally, the side output layer is set in the decoder part to generate a local prediction image corresponding to the level, and cooperate with the multi-label Dice loss function for training. Experiments on the DRIVE,STARE, and CHASE_DB1 datasets show that the sensitivity is 80.54%, 83.97%, and 82.40%, and the area under curve of the receiver operating characteristic is 98.07%, 98.50%, and 98.36%, respectively.

     

/

返回文章
返回