Transfer Learning Through Weighted Loss Function and Group Normalization
for Vessel Segmentation from Retinal Images
- URL: http://arxiv.org/abs/2012.09250v1
- Date: Wed, 16 Dec 2020 20:34:48 GMT
- Title: Transfer Learning Through Weighted Loss Function and Group Normalization
for Vessel Segmentation from Retinal Images
- Authors: Abdullah Sarhan, Jon Rokne, Reda Alhajj, and Andrew Crichton
- Abstract summary: The vascular structure of blood vessels is important in diagnosing retinal conditions such as glaucoma and diabetic retinopathy.
We propose an approach for segmenting retinal vessels that uses deep learning along with transfer learning.
Our approach results in greater segmentation accuracy than other approaches.
- Score: 0.0
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: The vascular structure of blood vessels is important in diagnosing retinal
conditions such as glaucoma and diabetic retinopathy. Accurate segmentation of
these vessels can help in detecting retinal objects such as the optic disc and
optic cup and hence determine if there are damages to these areas. Moreover,
the structure of the vessels can help in diagnosing glaucoma. The rapid
development of digital imaging and computer-vision techniques has increased the
potential for developing approaches for segmenting retinal vessels. In this
paper, we propose an approach for segmenting retinal vessels that uses deep
learning along with transfer learning. We adapted the U-Net structure to use a
customized InceptionV3 as the encoder and used multiple skip connections to
form the decoder. Moreover, we used a weighted loss function to handle the
issue of class imbalance in retinal images. Furthermore, we contributed a new
dataset to this field. We tested our approach on six publicly available
datasets and a newly created dataset. We achieved an average accuracy of 95.60%
and a Dice coefficient of 80.98%. The results obtained from comprehensive
experiments demonstrate the robustness of our approach to the segmentation of
blood vessels in retinal images obtained from different sources. Our approach
results in greater segmentation accuracy than other approaches.
Related papers
- Region Guided Attention Network for Retinal Vessel Segmentation [19.587662416331682]
We present a lightweight retinal vessel segmentation network based on the encoder-decoder mechanism with region-guided attention.
Dice loss penalises false positives and false negatives equally, encouraging the model to generate more accurate segmentation.
Experiments on a benchmark dataset show better performance (0.8285, 0.8098, 0.9677, and 0.8166 recall, precision, accuracy and F1 score respectively) compared to state-of-the-art methods.
arXiv Detail & Related papers (2024-07-22T00:08:18Z) - LMBiS-Net: A Lightweight Multipath Bidirectional Skip Connection based
CNN for Retinal Blood Vessel Segmentation [0.0]
Blinding eye diseases are often correlated with altered retinal morphology, which can be clinically identified by segmenting retinal structures in fundus images.
Deep learning has shown promise in medical image segmentation, but its reliance on repeated convolution and pooling operations can hinder the representation of edge information.
We propose a lightweight pixel-level CNN named LMBiS-Net for the segmentation of retinal vessels with an exceptionally low number of learnable parameters.
arXiv Detail & Related papers (2023-09-10T09:03:53Z) - Affinity Feature Strengthening for Accurate, Complete and Robust Vessel
Segmentation [48.638327652506284]
Vessel segmentation is crucial in many medical image applications, such as detecting coronary stenoses, retinal vessel diseases and brain aneurysms.
We present a novel approach, the affinity feature strengthening network (AFN), which jointly models geometry and refines pixel-wise segmentation features using a contrast-insensitive, multiscale affinity approach.
arXiv Detail & Related papers (2022-11-12T05:39:17Z) - Parametric Scaling of Preprocessing assisted U-net Architecture for
Improvised Retinal Vessel Segmentation [1.3869502085838448]
We present an image enhancement technique based on the morphological preprocessing coupled with a scaled U-net architecture.
A significant improvement as compared to the other algorithms in the domain, in terms of the area under ROC curve (>0.9762) and classification accuracy (>95.47%) are evident from the results.
arXiv Detail & Related papers (2022-03-18T15:26:05Z) - Assessing glaucoma in retinal fundus photographs using Deep Feature
Consistent Variational Autoencoders [63.391402501241195]
glaucoma is challenging to detect since it remains asymptomatic until the symptoms are severe.
Early identification of glaucoma is generally made based on functional, structural, and clinical assessments.
Deep learning methods have partially solved this dilemma by bypassing the marker identification stage and analyzing high-level information directly to classify the data.
arXiv Detail & Related papers (2021-10-04T16:06:49Z) - 3D Convolutional Neural Networks for Stalled Brain Capillary Detection [72.21315180830733]
Brain vasculature dysfunctions such as stalled blood flow in cerebral capillaries are associated with cognitive decline and pathogenesis in Alzheimer's disease.
Here, we describe a deep learning-based approach for automatic detection of stalled capillaries in brain images based on 3D convolutional neural networks.
In this setting, our approach outperformed other methods and demonstrated state-of-the-art results, achieving 0.85 Matthews correlation coefficient, 85% sensitivity, and 99.3% specificity.
arXiv Detail & Related papers (2021-04-04T20:30:14Z) - Rethinking the Extraction and Interaction of Multi-Scale Features for
Vessel Segmentation [53.187152856583396]
We propose a novel deep learning model called PC-Net to segment retinal vessels and major arteries in 2D fundus image and 3D computed tomography angiography (CTA) scans.
In PC-Net, the pyramid squeeze-and-excitation (PSE) module introduces spatial information to each convolutional block, boosting its ability to extract more effective multi-scale features.
arXiv Detail & Related papers (2020-10-09T08:22:54Z) - Multi-Task Neural Networks with Spatial Activation for Retinal Vessel
Segmentation and Artery/Vein Classification [49.64863177155927]
We propose a multi-task deep neural network with spatial activation mechanism to segment full retinal vessel, artery and vein simultaneously.
The proposed network achieves pixel-wise accuracy of 95.70% for vessel segmentation, and A/V classification accuracy of 94.50%, which is the state-of-the-art performance for both tasks.
arXiv Detail & Related papers (2020-07-18T05:46:47Z) - Joint Learning of Vessel Segmentation and Artery/Vein Classification
with Post-processing [27.825969553813092]
Vessel segmentation and artery/vein classification provide various information on potential disorders.
We adopt a UNet-based model, SeqNet, to accurately segment vessels from the background and make prediction on the vessel type.
Our experiments show that our method improves AUC to 0.98 for segmentation and the accuracy to 0.92 in classification over DRIVE dataset.
arXiv Detail & Related papers (2020-05-27T13:06:16Z) - Dense Residual Network for Retinal Vessel Segmentation [8.778525346264466]
We propose an efficient method to segment blood vessels in Scanning Laser Ophthalmoscopy retinal images.
Inspired by U-Net, "feature map reuse" and residual learning, we propose a deep dense residual network structure called DRNet.
Our method achieves the state-of-the-art performance even without data augmentation.
arXiv Detail & Related papers (2020-04-07T20:42:13Z) - Retinopathy of Prematurity Stage Diagnosis Using Object Segmentation and
Convolutional Neural Networks [68.96150598294072]
Retinopathy of Prematurity (ROP) is an eye disorder primarily affecting premature infants with lower weights.
It causes proliferation of vessels in the retina and could result in vision loss and, eventually, retinal detachment, leading to blindness.
In recent years, there has been a significant effort to automate the diagnosis using deep learning.
This paper builds upon the success of previous models and develops a novel architecture, which combines object segmentation and convolutional neural networks (CNN)
Our proposed system first trains an object segmentation model to identify the demarcation line at a pixel level and adds the resulting mask as an additional "color" channel in
arXiv Detail & Related papers (2020-04-03T14:07:41Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.