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.

Multi-Level Attention Network for Retinal Vessel Segmentation

A Multi-Scale Residual Attention Network for Retinal Vessel Segmentation

Multi-level Deep Supervised Networks for Retinal Vessel Segmentation

Lightweight Attention Convolutional Neural Network for Retinal Vessel Image Segmentation

Accurate Retinal Vessel Segmentation in Color Fundus Images Via Fully Attention-Based Networks

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

A High-Resolution Network with Strip Attention for Retinal Vessel Segmentation

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

Self-Paced Dual-Axis Attention Fusion Network for Retinal Vessel Segmentation

AMF-NET: Attention-aware Multi-scale Fusion Network for Retinal Vessel Segmentation

U-Net With Attention Mechanism For Retinal Vessel Segmentation

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

Fundus vascular image segmentation based on multiple attention mechanisms and deep learning

Automatic Retinal Blood Vessel Segmentation Based On Multi-Level Convolutional Neural Network

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

A Multi-Scale Attention Fusion Network for Retinal Vessel Segmentation

Vessel-Net: Retinal Vessel Segmentation Under Multi-path Supervision

Skeleton-guided multi-scale dual-coordinate attention aggregation network for retinal blood vessel segmentation

Automatic Retinal Vessel Segmentation Based on an Improved U-Net Approach

Multi-Scale Residual U-Net Fundus Blood Vessel Segmentation Based on Attention Mechanism

A combination of multi‐scale and attention based on the U‐shaped network for retinal vessel segmentation