U-net与Dense-net相结合的视网膜血管提取
Retinal blood vessel extraction by combining U-net and Dense-net
- 2019年24卷第9期 页码:1569-1580
收稿:2018-09-17,
修回:2019-3-28,
纸质出版:2019-09-16
DOI: 10.11834/jig.180517
移动端阅览

浏览全部资源
扫码关注微信
收稿:2018-09-17,
修回:2019-3-28,
纸质出版:2019-09-16
移动端阅览
目的
2
视网膜血管健康状况的自动分析对糖尿病、心脑血管疾病以及多种眼科疾病的快速无创诊断具有重要参考价值。视网膜图像中血管网络结构复杂且图像背景亮度不均使得血管区域的准确自动提取具有较大难度。本文通过使用具有对称全卷积结构的U-net深度神经网络实现视网膜血管的高精度分割。
方法
2
基于U-net网络中的层次化对称结构和Dense-net网络中的稠密连接方式,提出一种改进的适用于视网膜血管精准提取的深度神经网络模型。首先使用白化预处理技术弱化原始彩色眼底图像中的亮度不均,增强图像中血管区域的对比度;接着对数据集进行随机旋转、Gamma变换操作实现数据增广;然后将每一幅图像随机分割成若干较小的图块,用于减小模型参数规模,降低训练难度。
结果
2
使用多种性能指标对训练后的模型进行综合评定,模型在DRIVE数据集上的灵敏度、特异性、准确率和AUC(area under the curve)分别达到0.740 9、0.992 9、0.970 7和0.917 1。所提算法与目前主流方法进行了全面比较,结果显示本文算法各项性能指标均表现良好。
结论
2
本文针对视网膜图像中血管区域高精度自动提取难度大的问题,提出了一种具有稠密连接方式的对称全卷积神经网络改进模型。结果表明该模型在视网膜血管分割中能够达到良好效果,具有较好的研究及应用价值。
Objective
2
The automatic analysis of retinal vascular health status is a fundamental research topic in the area of fundus image processing. Analysis results can supply significant reference information for ophthalmologists to diagnose rapidly and noninvasively a variety of retinal pathologies
such as diabetes
glaucoma
hypertension
and diseases related to the brain and heart stocks. Although great progress has been achieved in the past decades
accurate automatic retinal vessel extraction remains a challenging problem due to the complex vascular network structure of retina vessels
uneven image background illumination
and random noises introduced by optical apparatuses. The traditional unsupervised retinal vessel segmentation methods generally identify retinal vessels with matched filters
vessel tractors
or templates designed artificially according to the vessel shape or prior information of a retinal image. Conversional supervised learning-based retinal vessel extraction algorithms generally consider artifact features as input and train shallow models
such as support vector machine
K-nearest neighbor classifiers
and traditional artificial neural networks. These models perform effectively in the case of normal retinal images with high-quality illumination and contrast. However
because of the representation limit of artificially designed features
these traditional vessel extraction methods fail when fundus vessels have low contrast with respect to the retinal background or are near nonvascular structures
such as the optic disk and fovea region. Recently
deep learning technology with multifarious convolutional neural networks has been widely applied to medical image processing and has achieved the most state-of-the-art performance due to its efficient and robust self-learned features. A series of new advances in retinal image processing has been achieved with deep learning networks. To help advance the research in this field
we adopt a deep neural network called U-net
which has a symmetrical full convolutional structure
and a dense connection to achieve an accurate end-to-end extraction of retinal vessels.
Method
2
A specially modified deep neural network for accurate retinal vessel extraction is proposed based on hierarchically symmetrical structure of the U-net model and the dense connection used in the Dense-net model. The introduction of the hierarchical symmetrical structure empowers the proposed model to perceive the coarse-grained and fine-grained image features through symmetrical down-sampling and up-sampling operations. At the same time
the adoption of a dense connection facilitates multiscale feature combination across different layers
including short connections of consecutive layers and skip connections over non-adjacent layers. This feature combination strategy can utilize comprehensive retina image information and enable the entire network to learn efficient and robust features rapidly. To accelerate the training convergence and enhance the generalization of the proposed neural network
we implement image preprocessing and data augmentation prior to model training. The problem of uneven background illumination is alleviated by the whitening operation
which calculates the average value and standard deviation of each input image channel and subtracts them from each pixel of the corresponding input image channel. Then
data augmentation is achieved by random rotation and gamma correction to generate more images than the raw input dataset scale. Subsequently
each image is divided into a mass of random patches with a certain degree of overlap. This operation can reduce the parameter scale dramatically and alleviate the training of the modified neural network greatly. Finally
these image patches are entered our neural network as a feeding group to be trained iteratively.
Result
2
The modified U-net
like the deep neural network model
adopts dense connections to effectively identify and enhance actual retinal vessels at different scales and suppress background simultaneously. To evaluate the proposed model's performance quantitatively
we employ the public dataset called DRIVE
which is one of the most rarely used retina vessel segmentation evaluation datasets. DRIVE comprises 40 images with manual segmentation benchmarks and is divided into a training and a test set
each containing 20 images. In our evaluation
four performance indices are used to assess the proposed method thoroughly:accuracy (ACC)
sensitivity (SE)
specificity (SP)
and area under a curve (AUC)
all of which are widely accepted evaluation indices for retina vessel segmentation. The comprehensive experiments show that ACC
SE
SP
and AUC of the proposed algorithm for the DRIVE dataset reach 0.970 7
0.740 9
0.992 9
and 0.917 1
respectively. Compared with other state-of-the-art methods
our model presents competitive performance. The accuracy of the proposed model structure can shorten the training time dramatically; it only requires five epochs to converge and approximately one-tenth of time
the same as the initial U-net model. This contribution of the dense connection and batch normalization is used in our modified model.
Conclusion
2
A specially designed deep neural network for retinal vessel extraction is proposed to address the problems caused by the low contrast of the retinal vascular structure resulting from their background and uneven illumination. The main contributions of this modified model lie in its symmetrical structure and its dense connection over non-adjacent layers. In addition
the data augmentation with random rotation
which is only available for retina images given that the retina area is a circular-like disk
and the addition of batch normalization in the model contribute to the rapid training convergence and high accuracy of vessel segmentation. Experimental results on a widely used open dataset demonstrate that the proposed modified neural network can deal with these problems and achieve accurate retinal vessel segmentation. Compared with other mainstream deep learning algorithms
the proposed method shows enhanced retinal vessel segmentation accuracy and robustness and presents promising potential in retinal image processing.
Gu X D, Guo S D, Yu D H. New approach for noise reducing of image based on PCNN[J]. Journal of Electronics and Information Technology, 2002, 24(10):1304-1309.
顾晓东, 郭仕德, 余道衡.一种基于PCNN的图像去噪新方法[J].电子与信息学报, 2002, 24(10):1304-1309.
Fraz M M, Remagnino P, Hoppe A, et al. Blood vessel segmentation methodologies in retinal images:a survey[J]. Computer Methods and Programs in Biomedicine, 2012, 108(1):407-433.[DOI:10.1016/j.cmpb.2012.03.009]
Liu I, Sun Y. Recursive tracking of vascular networks in angiograms based on the detection-deletion scheme[J]. IEEE Transactions on Medical Imaging, 1993, 12(2):334-341.[DOI:10.1109/42.232264]
Lam B S Y, Yan H. A novel vessel segmentation algorithm for pathological retina images based on the divergence of vector fields[J]. IEEE Transactions on Medical Imaging, 2008, 27(2):237-246.[DOI:10.1109/TMI.2007.909827]
Espona L, Carreira M J, Penedo M G, et al. Retinal vessel tree segmentation using a Deformable contour model[C]//Proceedings of the 19th International Conference on Pattern Recognition. Tampa, FL, USA: IEEE, 2008: 1-4.[ DOI: 10.1109/ICPR.2008.4761762 http://dx.doi.org/10.1109/ICPR.2008.4761762 ]
Gang L, Chutatape O, Krishnan S M. Detection and measurement of retinal vessels in fundus images using amplitude modified second-order Gaussian filter[J]. IEEE Transactions on Biomedical Engineering, 2002, 49(2):168-172.[DOI:10.1109/10.979356]
Yu Z. Research on medical image segmentation based on geometric deformable model[D]. Tianjin: Tianjin University, 2007.
于正.基于几何形变模型的医学图像分割方法研究[D].天津: 天津大学, 2007.
Osareh A, Shadgar B. Retinal vessel extraction using Gabor filters and support vector machines[C]//Proceedings of the 13th International CSI Computer Conference on Advances in Computer Science and Engineering. Kish Island, Iran: Springer, 2008: 356-363.[ DOI: 10.1007/978-3-540-89985-3_44 http://dx.doi.org/10.1007/978-3-540-89985-3_44 ]
Marin D, Gegundez-Arias M E, Ponte B, et al. An exudate detection method for diagnosis risk of diabetic macular edema in retinal images using feature-based and supervised classification[J]. Medical and Biological Engineering and Computing, 2018, 56(8):1379-1390.[DOI:10.1007/s11517-017-1771-2]
Niemeijer M, Staal J, van Ginneken B, et al. Comparative study of retinal vessel segmentation methods on a new publicly available database[C]//Proceedings of SPIE 5370, Medical Imaging 2004: Image Processing. San Diego, California, USA: SPIE, 2004, 5370: 648-656.[ DOI: 10.1117/12.535349 http://dx.doi.org/10.1117/12.535349 ]
Soares J V B, Leandro J J G, Cesar R M, et al. Retinal vessel segmentation using the 2-DGabor wavelet and supervised classification[J]. IEEE Transactions on Medical Imaging, 2006, 25(9):1214-1222.[DOI:10.1109/TMI.2006.879967]
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.[DOI:10.1109/TMI.2007.898551]
You X G, Peng Q M, Yuan Y, et al. Segmentation of retinal blood vessels using the radial projection and semi-supervised approach[J]. Pattern Recognition, 2011, 44(10-11):2314-2324.[DOI:10.1016/j.patcog.2011.01.007]
Alonso-Montes C, Vilarino D L, Penedo M G. CNN-based automatic retinal vascular tree extraction[C]//Proceedings of the 9th International Workshop on Cellular Neural Networks and Their Applications. Hsinchu, Taiwan, China: IEEE, 2005: 61-64.[ DOI: 10.1109/CNNA.2005.1543161 http://dx.doi.org/10.1109/CNNA.2005.1543161 ]
Perfetti R, Ricci E, Casali D, et al. Cellular neural networks with virtual template expansion for retinal vessel segmentation[J]. IEEE Transactions on Circuits and Systems Ⅱ:Express Briefs, 2007, 54(2):141-145.[DOI:10.1109/TCSⅡ.2006.886244]
Ronneberger O, Fischer P, Brox T. U-Net: convolutional networks for biomedical image segmentation[C]//Proceedings of the 18th International Conference on Medical Image Computing and Computer-assisted Intervention. Munich, Germany: Springer, 2015.[ DOI: 10.1007/978-3-319-24574-4_28 http://dx.doi.org/10.1007/978-3-319-24574-4_28 ]
Alom M Z, Hasan M, Yakopcic C, et al. Recurrent residual convolutional neural network based on U-Net (R2U-Net) for medical image segmentation[DB/OL].[2018-09-02] . https://arxiv.org/pdf/1802.06955.pdf https://arxiv.org/pdf/1802.06955.pdf .
Han Y, Ye J C. Framing U-Net via deep convolutional framelets:application to sparse-view CT[J]. IEEE Transactions on Medical Imaging, 2018, 37(6):1418-1429.[DOI:10.1109/TMI.2018.2823768]
Huang G, Liu Z, van der Maaten L, et al. Densely connected convolutional networks[C]//Proceedings of 2017 IEEE Conference on Computer Vision and Pattern Recognition.2017: 2261-2269.[ DOI: 10.1109/CVPR.2017.243 http://dx.doi.org/10.1109/CVPR.2017.243 ]
Li Y D, Hao Z B, Lei H. Survey of convolutional neural network[J]. Journal of Computer Applications, 2016, 36(9):2508-2515, 2565.
李彦冬, 郝宗波, 雷航.卷积神经网络研究综述[J].计算机应用, 2016, 36(9):2508-2515, 2565. [DOI:10.11772/j.issn.1001-9081.2016.09.2508]
Shelhamer E, Long J, Darrell T. Fully convolutional networks for semantic segmentation[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(4):640-651.[DOI:10.1109/TPAMI.2016.2572683]
Staal J, Abramoff 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.[DOI:10.1109/TMI.2004.825627]
Nahid A A, Mehrabi M A, Kong Y N. Histopathological breast cancer image classification by deep neural network techniques guided by local clustering[J]. BioMed Research International, 2018, 2018:#2362108.
Cortes C, Gonzalvo X, Kuznetsov V, et al. AdaNet: adaptive structural learning of artificial neural networks[EB/OL].[2018-09-02] . https://arxiv.org/pdf/1607.01097.pdf https://arxiv.org/pdf/1607.01097.pdf .
Kim J, Kim S, Lee M. Convolutional neural network with biologically inspired ON/OFF ReLU[C]//Proceedings of the 22nd International Conference Neural Information Processing. Istanbul, Turkey: Springer, 2015.[ DOI: 10.1007/978-3-319-26561-2_38 http://dx.doi.org/10.1007/978-3-319-26561-2_38 ]
Zana F, Klein J C. Segmentation of vessel-like patterns using mathematical morphology and curvature evaluation[J]. IEEE Transactions on Image Processing, 2001, 10(7):1010-1019.[DOI:10.1109/83.931095]
Al-Diri B, Hunter A, Steel D. An active contour model for segmenting and measuring retinal vessels[J]. IEEE Transactions on Medical Imaging, 2009, 28(9):1488-1497.
Miri M S, Mahloojifar A. Retinal image analysis using curvelet transform and multistructure elements morphology by reconstruction[J]. IEEE Transactions on Biomedical Engineering, 2011, 58(5):1183-1192.[DOI:10.1109/TBME.2010.2097599]
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.[DOI:10.1016/j.cmpb.2011.08.009]
Sohini R, Koozekanani D D, Parhi K K. Blood vessel segmentation of fundus images by major vessel extraction and subimage classification[J]. IEEE Journal of Biomedical and Health Informatics, 2015, 19(3):1118-1128.[DOI:10.1109/JBHI.2014.2335617]
Oliveira A, Pereira S, Silva C A. Retinal vessel segmentation based on fully convolutional neural networks[J]. Expert Systems with Applications, 2018, 112:229-242.[DOI:10.1016/j.eswa.2018.06.034]
相关作者
相关机构
京公网安备11010802024621