基于密集注意力网络的视网膜血管图像分割

doi:10.19678/j.issn.1000-3428.0054379

摘要/Abstract

摘要： 视网膜血管的结构信息对眼科疾病的诊断具有重要的指导意义，对视网膜血管图像进行高效正确的分割成为临床的迫切需求。传统的人工分割方法耗时较长且易受个人主观因素的影响，分割质量不高。为此，提出一种基于密集注意力网络的图像自动分割算法。将编码器-解码器全卷积神经网络的基础结构与密集连接网络相结合，以充分提取每一层的特征，在网络的解码器端引入注意力门模块，对不必要的特征进行抑制，提高视网膜血管图像的分割精度。在DRIVE和STARE眼底图像数据集上的实验结果表明，与其他基于深度学习的算法相比，该算法的敏感性、特异性、准确率和AUC值均较高，分割效果较好。

关键词: 图像分割, 视网膜血管, 全卷积神经网络, 密集连接, 注意力机制

Abstract: The structural information of retinal vesselsassists in the diagnosis of ophthalmic diseases,and thus efficient and accurate segmentation of retinal vessel images has become an urgent clinical demannd.The traditional artificial segmentation methods are time-consumingand frequently affected by personal subjective factors,leading to a decline in segmentation quality.To address the problem,thispaper proposes an automatic image segmentation algorithm based on dense attention network.The algorithm combines the basic structure of the encoder-decoder fully convolutional neural network with the densely connected network to fully extract the features of each layer.Then the attention gate module on the decoder side of the network is introduced to suppress unnecessary features and thus improve the segmentation accuracy of retinal vessel segmentation.Experimental results on DRIVE and STARE fundus image datasets show that compared with other algorithms based on deep learning,the proposedalgorithm has excellent segmentation performance with the sensitivity,specificity,accuracy and AUC value all improved.

Key words: image segmentation, retinal vessel, Full Convolutional Neural Network(FCNN), dense connection, attention mechanism

中图分类号:

TP391

梅旭璋, 江红, 孙军. 基于密集注意力网络的视网膜血管图像分割[J]. 计算机工程, 2020, 46(3): 267-272,279.

MEI Xuzhang, JIANG Hong, SUN Jun. Retinal Vessel Image Segmentation Based on Dense Attention Network[J]. Computer Engineering, 2020, 46(3): 267-272,279.

https://www.ecice06.com/CN/Y2020/V46/I3/267

图/表 8

20200321092244

20200321092246

20200321092249

20200321092252

20200321092255

20200321092259

20200321092302

20200321092305

参考文献

[1] FRAZ M M,BARMAN S A,REMAGNINO P,et al.An approach to localize the retinal blood vessels using bit planes and centerline detection[J].Computer Methods and Programs in Biomedicine,2012,108(2):600-616.
[2] AZZOPARDI G,STRISCIUGLIO N,VENTO M,et al.Trainable COSFIRE filters for vessel delineation with application to retinal images[J].Medical Image Analysis,2015,19(1):46-57.
[3] ZHANG Jia,LI Huiqi,NIE Qing,et al.A retinal vessel boundary tracking method based on Bayesian theory and multi-scale line detection[J].Computerized Medical Imaging and Graphics,2014,38(6):517-525.
[4] FRAZ M M,REMAGNINO P,HOPPE A,et al.An ensemble classification-based approach applied to retinal blood vessel segmentation[J].IEEE Transactions on Biomedical Engineering,2012,59(9):2538-2548.
[5] ORLANDO J I,PROKOFYEVA E,BLASCHKO M B.A discriminatively trained fully connected conditional random field model for blood vessel segmentation in fundus images[J].IEEE Transactions on Biomedical Engineering,2016,64(1):16-27.
[6] KRIZHEVSKY A,SUTSKEVER I,HINTON G E.Imagenet classification with deep convolutional neural networks[C]//Proceedings of Advances in Neural Information Processing Systems.New York,USA:ACM Press,2012:1097-1105.
[7] LONG J,SHELHAMER E,DARRELL T.Fully convolutional networks for semantic segmentation[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2014,39(4):640-651.
[8] RONNEBERGER O,FISCHER P,BROX T.U-net:convolutional networks for biomedical image segmentation[C]//Proceedings of International Conference on Medical Image Computing and Computer-Assisted Intervention.Berlin,Germany:Springer,2015:234-241.
[9] LISKOWSKI P,KRAWIEC K.Segmenting retinal blood vessels with deep neural networks[J].IEEE Transactions on Medical Imaging,2016,35(11):2369-2380.
[10] LI Qiaoliang,FENG Bowei,XIE Linpei,et al.A cross-modality learning approach for vessel segmentation in retinal images[J].IEEE Transactions on Medical Imaging,2016,35(1):109-118.
[11] FU H,XU Y,LIN S,et al.DeepVessel:retinal vessel segmentation via deep learning and conditional random field[C]//Proceedings of International Conference on Medical Image Computing and Computer-Assisted Intervention.Berlin,Germany:Springer,2016:132-139.
[12] ZHENG Tingyue,TANG Chen,LEI Zhenkun.Multi-scale retinal vessel segmentation based on full convolutional neural network[J].Acta Optica Sinica,2019,39(2):119-126.(in Chinese) 郑婷月,唐晨,雷振坤.基于全卷积神经网络的多尺度视网膜血管分割[J].光学学报,2019,39(2):119-126.
[13] ZHU Ting,WANG Yu,XIAO Hongbing,et al.Automatic segmentation of multimodal MRI brain tumors based on WRN-PPNet[J].Computer Engineering,2018,44(12):258-263,270.(in Chinese) 朱婷,王瑜,肖洪兵,等.基于WRN-PPNet的多模态MRI脑肿瘤全自动分割[J].计算机工程,2018,44(12):258-263,270.
[14] HUANG Wenming,WEI Wancheng,ZHANG Jian,et al.Recommendation method based on attention mechanism and review text deep model[J].Computer Engineering,2019,45(9):176-182.(in Chinese) 黄文明,卫万成,张健,等.基于注意力机制与评论文本深度模型的推荐方法[J].计算机工程,2019,45(9):176-182.
[15] WANG Peisen,SONG Yan,DAI Lirong.Fine-grained image classification based on multi-channel visual attention[J].Journal of Data Acquisition and Processing,2019,34(1):157-166.(in Chinese) 王培森,宋彦,戴礼荣.基于多通道视觉注意力的细粒度图像分类[J].数据采集与处理,2019,34(1):157-166.
[16] CHO K,COURVILLE A,BENGIO Y.Describing multimedia content using attention-based encoder-decoder networks[J].IEEE Transactions on Multimedia,2015,17(11):1875-1886.
[17] HUANG Gao,LIU Zhuang,MAATEN L V R,et al.Densely connected convolutional networks[C]//Proceedings of 2017 IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C.,USA:IEEE Press,2017:2261-2269.
[18] OKTAY O,SCHLEMPER J,FOLGOC L L,et al.Attention U-Net:learning where to look for the pancreas[C]//Proceedings of International Conference on Medical Imaging with Deep Learning.Amsterdam,Netherlands:[s.n.],2018:1-10.
[19] STAAL J,ABRÀMOFF M D,NIEMEIJER M,et al.Ridge-based vessel segmentation in color images of the retina[J].IEEE Transactions on Medical Imaging,2004,23(4):501-509.
[20] HOOVER A D,KOUZNETSOVA V,GOLDBAUM M.Locating blood vessels in retinal images by piecewise threshold probing of a matched filter response[J].IEEE Transactions on Medical Imaging,2002,19(3):203-210.
[21] WU Ying.Adaptive image enhancement algorithm based on bi-histogram equalization[J].Computer Engineering,2011,37(4):244-245.(in Chinese)武英.基于双直方图均衡的自适应图像增强算法[J].计算机工程,2011,37(4):244-245.
[22] KINGMA D P,BA J.Adam:a method for stochastic optimization[C]//Proceedings of the 3rd International Conference on Learning Representations.New York,USA:[s.n.],2015:1-15.

选择文件类型/文献管理软件名称

选择包含的内容