1Faculty of Electronic Information and Electrical Engineering, Dalian University of Technology, Dalian 116024, Liaoning, China; 2Department of Ophthalmology, Zhongshan Hospital Affiliated to Dalian University, Dalian 116001, Liaoning, China
Abstract:The blood vessel segmentation of fundus images is the basis of computer-aided diagnosis of ophthalmology and other related diseases. Early diagnosis and monitoring of diseases such as diabetic retinopathy, hypertension and arteriosclerosis can be performed by segmenting the vascular structure in the fundus image. However, existing segmentation algorithms are challenged with low accuracy and low sensitivity. This paper proposed an improved U-Net fundus image segmentation algorithm based on the basic theory of deep learning. Firstly, the problem of less fundus data set was solved by reducing the number of pooling layers and upsampling layers of the traditional U-Net. Secondly, the use efficiency of the feature was improved by changing the traditional convolutional layer serial connection method to the residual mapping. Finally, the batch normalization and PReLU activation functions are added between the convolutional layers to optimize the network, which further improved the network performance. This paper conducted experiments on two public fundus databases, DRIVE and CHASE_DB1. The 160 000 image blocks were randomly extracted from each database and are sent into the improved network for training and testing. The sensitivity, accuracy and AUC (area under the ROC curve) of the algorithm were 2.47%, 0.21% and 0.35%, higher than those of the existing contrasted algorithms. The proposed algorithm improves the low accuracy and low sensitivity of small blood vessels segmentation in fundus images, and segmented small blood vessels better with low contrast.
[1] 陆培, 王磊, 李志成, 等. 一种普适的基于多尺度滤波和统计学混合模型的血管分割方法[J]. 中国生物医学工程学报, 2016, 35(5): 519-525. [2] 李居朋, 陈后金, 张新媛. 基于先验知识随机游走模型的视网膜血管分割方法[J]. 中国生物医学工程学报, 2009, 28(4):501-507. [3] Ricci E, Perfetti R. Retinal blood vessel segmentation using line operators and support vector classification[J]. IEEE Transactions on Medical Imaging, 2007, 26(10): 1357-1365. [4] Fraz MM, 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] Fu H, Xu Y, Wong DWK, et al. Retinal vessel segmentation via deep learning network and fully-connected conditional random fields[C] //2016 IEEE 13th International Symposium on Biomedical Imaging (ISBI). Prague: IEEE, 2016: 698-701. [6] Orlando JI, Prokofyeva E, Blaschko MB. A discriminatively trained fully connected conditional random field model for blood vessel segmentation in fundus images[J]. IEEE Transactions on Biomedical Engineering. 2017, 64(1): 16-27. [7] Liskowski P, Krawiec K. Segmenting retinal blood vessels with deep neural networks[J]. IEEE Transactions on Medical Imaging, 2016, 35(11): 2369-2380. [8] 朱承璋, 邹北骥, 向遥, 等. 彩色眼底图像视网膜血管分割方法研究进展[J]. 计算机辅助设计与图形学学报, 2015,11: 2046-2057. [9] Ronneberger O, Fischer P, Brox T. U-net: Convolutional networks for biomedical image segmentation[J] //arXiv,2015,1505:04597 [10] He K, Zhang X, Ren S, et al. Deep residual learning for image recognition[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Las Vegas: IEEE, 2016: 770-778. [11] Ngo L, Han JH. Multi-level deep neural network for efficient segmentation of blood vessels in fundus images[J]. Electronics Letters, 2017, 53(16):1096-1098. [12] Gao X, Cai Y, Qiu C, et al. Retinal blood vessel segmentation based on the Gaussian matched filter and U-net[C]// 2017 10th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics (CISP-BMEI). Shanghai:IEEE, 2017: 1-5. [13] Roychowdhury S, Koozekanani DD, Parhi KK. Blood vessel segmentation of fundus images by major vessel extraction and subimage classification[J]. IEEE J Biomed Health Inform, 2017, 19(3):1118-1128. [14] Marín D, Aquino A, Gegundez-Arias ME, et al. A new supervised method for blood vessel segmentation in retinal images by using gray-level and moment invariants-based features[J]. IEEE Transactions on Medical Imaging. 2011, 30(1): 146-158. [15] Cheng E, Du L, Wu Y, et al. Discriminative vessel segmentation in retinal images by fusing context-aware hybrid features[J]. Machine Vision and Applications. 2014, 25(7): 1779-1792. [16] Li Q, Feng B, Xie L, et al. A cross-modality learning approach for vessel segmentation in retinal images[J]. IEEE Transactions on Medical Imaging, 2016, 35(1): 109-118. [17] Dasgupta A, Singh S. A fully convolutional neural network based structured prediction approach towards the retinal vessel segmentation[C] //2017 IEEE 14th International Symposium on Biomedical Imaging (ISBI 2017). Melbourne:IEEE, 2017: 248-251. [18] Yan Z, Yang X, Cheng KTT. Joint segment-level and pixel-wise losses for deep learning based retinal vessel segmentation[J]. IEEE Transactions on Biomedical Engineering, 2018, 65:1912-1923. [19] 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. [20] Roychowdhury S, Koozekanani DD, Parhi KK. Iterative vessel segmentation of fundus images[J]. IEEE Transactions on Biomedical Engineering, 2015, 62(7): 1738-1749. [21] Zhang J, Dashtbozorg B, Bekkers E, et al. Robust retinal vessel segmentation via locally adaptive derivative frames in orientation scores[J]. IEEE Transactions on Medical Imaging, 2016, 35(12): 2631-2644.