UNet retinal blood vessel segmentation algorithm based on improved pyramid pooling method and attention mechanism

Abstract

The segmentation results of retinal vessels have a significant impact on the automatic diagnosis of retinal diabetes, hypertension, cardiovascular and cerebrovascular diseases and other ophthalmic diseases. In order to improve the performance of blood vessels segmentation, a pyramid scene parseing U-Net segmentation algorithm based on attention mechanism was proposed. The modified PSP-Net pyramid pooling module is introduced on the basis of U-Net network, which aggregates the context information of different regions so as to improve the ability of obtaining global information. At the same time, attention mechanism was introduced in the skip connection part of U-Net network, which makes the integration of low-level features and high-level semantic features more efficient and reduces the loss of feature information through nonlinear connection mode. The sensitivity, specificity, accuracy and AUC of DRIVE and CHASE_DB1 data sets are 0.7814, 0.9810, 0.9556, 0.9780; 0.8195, 0.9727, 0.9590, 0.9784. Experimental results show that the PSP-UNet segmentation algorithm based on the attention mechanism enhances the detection ability of blood vessel pixels, suppresses the interference of irrelevant information and improves the network segmentation performance, which is superior to U-Net algorithm and some mainstream retinal vascular segmentation algorithms at present.