Abstract:Morphological and functional changes in retinal vessels are indicators of a variety of chronic diseases, such as diabetes, stroke, and hypertension. However, without a large number of high-quality annotations, existing deep learning-based medical image segmentation approaches may degrade their performance dramatically on the retinal vessel segmentation task. To reduce the demand of high-quality annotations and make full use of massive unlabeled data, we propose a self-supervised multi-task strategy to extract curvilinear vessel features for the retinal vessel segmentation task. Specifically, we use a dense network to extract more vessel features across different layers/slices, which is elaborately designed for hardware to train and test efficiently. Then, we combine three general pre-training tasks (i.e., intensity transformation, random pixel filling, in-painting and out-painting) in an aggregated way to learn rich hierarchical representations of curvilinear retinal vessel structures. Furthermore, a vector classification task module is introduced as another pre-training task to obtain more spatial features. Finally, to make the segmentation network pay more attention to curvilinear structures, a novel dynamic loss is proposed to learn robust vessel details from unlabeled fundus images. These four pre-training tasks greatly reduce the reliance on labeled data. Moreover, our network can learn the retinal vessel features effectively in the pre-training process, which leads to better performance in the target multi-modal segmentation task. Experimental results show that our method provides a promising direction for the retinal vessel segmentation task. Compared with other state-of-the-art supervised deep learning-based methods applied, our method requires less labeled data and achieves comparable segmentation accuracy. For instance, we match the accuracy of the traditional supervised learning methods on DRIVE and Vampire datasets without needing any labeled ground truth image. With elaborately training, we gain the 0.96 accuracy on DRIVE dataset.

Semi-Supervised Learning Method Of U-Net Deep Learning Network For Blood Vessel Segmentation In Retinal Images

A Multi-Scale Residual Attention Network for Retinal Vessel Segmentation

Retinal Vessel Segmentation Method Based on Improved U-NET Network

An improved method for retinal vessel segmentation in U-Net

An Improved U-net Based Retinal Vessel Image Segmentation Method.

LEA U-Net: a U-Net-based deep learning framework with local feature enhancement and attention for retinal vessel segmentation

Blood Vessel and Background Separation for Retinal Image Quality Assessment

DilUnet: A U-net based architecture for blood vessels segmentation

A Multi-Task Dense Network with Self-Supervised Learning for Retinal Vessel Segmentation

Retinal Blood Vessels and Optic Disc Segmentation Using U-Net

Attention residual convolution neural network based on U-net (AttentionResU-Net) for retina vessel segmentation

Retinal blood vessel segmentation by using the MS-LSDNet network and geometric skeleton reconnection method

Improved Fully Convolutional Neuron Networks on Small Retinal Vessel Segmentation Using Local Phase As Attention.

A Supervised Method Using Convolutional Neural Networks for Retinal Vessel Delineation

Retinal blood vessel segmentation using a deep learning method based on modified U-NET model

Retinal Blood Vessels Segmentation With Improved SE‐UNet Model

An Improved U-Net Architecture for Simultaneous Arteriole and Venule Segmentation in Fundus Image.

Lightweight Attention Convolutional Neural Network for Retinal Vessel Image Segmentation

RLeU-Net: Segmentation of blood vessels in retinal fundus images for Diabetic Retinopathy Screening

Dense Residual Network for Retinal Vessel Segmentation

Research on Retinal Vessel Segmentation Algorithm Based on a Modified U-Shaped Network