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.

The joint learning of multi-resolution feature for multi-class retinal vessel segmentation

Retinal Vessel Segmentation Via Cross-attention Feature Fusion

A Multi-Scale Residual Attention Network for Retinal Vessel Segmentation

Multi-Scale and Multi-Branch Convolutional Neural Network for Retinal Image Segmentation

Accurate Retinal Vessel Segmentation via Octave Convolution Neural Network

A retinal vessel segmentation network with multiple-dimension attention and adaptive feature fusion

A multi-scale feature extraction and fusion-based model for retinal vessel segmentation in fundus images

Multi-scale Interactive Network with Artery/Vein Discriminator for Retinal Vessel Classification

Graph Convolution Based Cross-Network Multiscale Feature Fusion for Deep Vessel Segmentation

Learning to Address Intra-segment Misclassification in Retinal Imaging

Learning multi-scale deep fusion for retinal blood vessel extraction in fundus images

A high resolution representation network with multi-path scale for retinal vessel segmentation

Learning based multi-scale feature fusion for retinal blood vessels segmentation

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

Global and local multi-modal feature mutual learning for retinal vessel segmentation

Dual-path multi-scale context dense aggregation network for retinal vessel segmentation

Vessel Segmentation in Fundus Images with Multi-Scale Feature Extraction and Disentangled Representation

Multi-level spatial-temporal and attentional information deep fusion network for retinal vessel segmentation

Joint Extraction of Retinal Vessels and Centerlines Based on Deep Semantics and Multi-Scaled Cross-Task Aggregation

A Multi-Scale Feature Fusion Method Based on U-Net for Retinal Vessel Segmentation

CMP-UNet: A Retinal Vessel Segmentation Network Based on Multi-Scale Feature Fusion