Log In Sign Up

Multi-Task Neural Networks with Spatial Activation for Retinal Vessel Segmentation and Artery/Vein Classification

by   Wenao Ma, et al.

Retinal artery/vein (A/V) classification plays a critical role in the clinical biomarker study of how various systemic and cardiovascular diseases affect the retinal vessels. Conventional methods of automated A/V classification are generally complicated and heavily depend on the accurate vessel segmentation. In this paper, we propose a multi-task deep neural network with spatial activation mechanism that is able to segment full retinal vessel, artery and vein simultaneously, without the pre-requirement of vessel segmentation. The input module of the network integrates the domain knowledge of widely used retinal preprocessing and vessel enhancement techniques. We specially customize the output block of the network with a spatial activation mechanism, which takes advantage of a relatively easier task of vessel segmentation and exploits it to boost the performance of A/V classification. In addition, deep supervision is introduced to the network to assist the low level layers to extract more semantic information. The proposed network achieves pixel-wise accuracy of 95.70 accuracy of 94.50 the AV-DRIVE dataset. Furthermore, we have also tested the model performance on INSPIRE-AVR dataset, which achieves a skeletal A/V classification accuracy of 91.6


page 6

page 7


MPG-Net: Multi-Prediction Guided Network for Segmentation of Retinal Layers in OCT Images

Optical coherence tomography (OCT) is a commonly-used method of extracti...

Residual Spatial Attention Network for Retinal Vessel Segmentation

Reliable segmentation of retinal vessels can be employed as a way of mon...

DUNet: A deformable network for retinal vessel segmentation

Automatic segmentation of retinal vessels in fundus images plays an impo...

Enhancement of Retinal Fundus Images via Pixel Color Amplification

We propose a pixel color amplification theory and family of enhancement ...

A Unified Mammogram Analysis Method via Hybrid Deep Supervision

Automatic mammogram classification and mass segmentation play a critical...

Domain knowledge assisted cyst segmentation in OCT retinal images

3D imaging modalities are becoming increasingly popular and relevant in ...