Objective-Dependent Uncertainty Driven Retinal Vessel Segmentation

by   Suraj Mishra, et al.

From diagnosing neovascular diseases to detecting white matter lesions, accurate tiny vessel segmentation in fundus images is critical. Promising results for accurate vessel segmentation have been known. However, their effectiveness in segmenting tiny vessels is still limited. In this paper, we study retinal vessel segmentation by incorporating tiny vessel segmentation into our framework for the overall accurate vessel segmentation. To achieve this, we propose a new deep convolutional neural network (CNN) which divides vessel segmentation into two separate objectives. Specifically, we consider the overall accurate vessel segmentation and tiny vessel segmentation as two individual objectives. Then, by exploiting the objective-dependent (homoscedastic) uncertainty, we enable the network to learn both objectives simultaneously. Further, to improve the individual objectives, we propose: (a) a vessel weight map based auxiliary loss for enhancing tiny vessel connectivity (i.e., improving tiny vessel segmentation), and (b) an enhanced encoder-decoder architecture for improved localization (i.e., for accurate vessel segmentation). Using 3 public retinal vessel segmentation datasets (CHASE_DB1, DRIVE, and STARE), we verify the superiority of our proposed framework in segmenting tiny vessels (8.3 achieving better area under the receiver operating characteristic curve (AUC) compared to state-of-the-art methods.



There are no comments yet.


page 2

page 4


Accurate Retinal Vessel Segmentation via Octave Convolution Neural Network

Retinal vessel segmentation is a crucial step in diagnosing and screenin...

RV-GAN : Retinal Vessel Segmentation from Fundus Images using Multi-scale Generative Adversarial Networks

Retinal vessel segmentation contributes significantly to the domain of r...

DPN: Detail-Preserving Network with High Resolution Representation for Efficient Segmentation of Retinal Vessels

Retinal vessels are important biomarkers for many ophthalmological and c...

Transfer Learning Through Weighted Loss Function and Group Normalization for Vessel Segmentation from Retinal Images

The vascular structure of blood vessels is important in diagnosing retin...

Direct Classification of Type 2 Diabetes From Retinal Fundus Images in a Population-based Sample From The Maastricht Study

Type 2 Diabetes (T2D) is a chronic metabolic disorder that can lead to b...

Dynamic Deep Networks for Retinal Vessel Segmentation

Segmenting the retinal vasculature entails a trade-off between how much ...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.