RRWNet: Recursive Refinement Network for Effective Retinal Artery/Vein Segmentation and Classification
- URL: http://arxiv.org/abs/2402.03166v4
- Date: Thu, 8 Aug 2024 13:32:21 GMT
- Title: RRWNet: Recursive Refinement Network for Effective Retinal Artery/Vein Segmentation and Classification
- Authors: José Morano, Guilherme Aresta, Hrvoje Bogunović,
- Abstract summary: A thorough analysis of the retinal vasculature requires the segmentation of the blood vessels and their classification into arteries and veins.
We introduce RRWNet, a novel end-to-end deep learning framework that addresses this limitation.
In particular, RRWNet is composed of two specializedworks: a Base subnetwork that generates base segmentation maps from the input images, and a Recursive Refinement subnetwork that iteratively improves these maps.
- Score: 0.8386558353546658
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: The caliber and configuration of retinal blood vessels serve as important biomarkers for various diseases and medical conditions. A thorough analysis of the retinal vasculature requires the segmentation of the blood vessels and their classification into arteries and veins, typically performed on color fundus images obtained by retinography. However, manually performing these tasks is labor-intensive and prone to human error. While several automated methods have been proposed to address this task, the current state of art faces challenges due to manifest classification errors affecting the topological consistency of segmentation maps. In this work, we introduce RRWNet, a novel end-to-end deep learning framework that addresses this limitation. The framework consists of a fully convolutional neural network that recursively refines semantic segmentation maps, correcting manifest classification errors and thus improving topological consistency. In particular, RRWNet is composed of two specialized subnetworks: a Base subnetwork that generates base segmentation maps from the input images, and a Recursive Refinement subnetwork that iteratively and recursively improves these maps. Evaluation on three different public datasets demonstrates the state-of-the-art performance of the proposed method, yielding more topologically consistent segmentation maps with fewer manifest classification errors than existing approaches. In addition, the Recursive Refinement module within RRWNet proves effective in post-processing segmentation maps from other methods, further demonstrating its potential. The model code, weights, and predictions will be publicly available at https://github.com/j-morano/rrwnet.
Related papers
- The Topology-Overlap Trade-Off in Retinal Arteriole-Venule Segmentation [0.0]
convolutional neural networks can achieve high overlap between predictions and expert annotations.
We show that our model is able to produce results on par with state-of-the-art from the point of view of overlap.
arXiv Detail & Related papers (2023-03-31T13:01:05Z) - Self-Supervised Correction Learning for Semi-Supervised Biomedical Image
Segmentation [84.58210297703714]
We propose a self-supervised correction learning paradigm for semi-supervised biomedical image segmentation.
We design a dual-task network, including a shared encoder and two independent decoders for segmentation and lesion region inpainting.
Experiments on three medical image segmentation datasets for different tasks demonstrate the outstanding performance of our method.
arXiv Detail & Related papers (2023-01-12T08:19:46Z) - A Systematic Evaluation of Node Embedding Robustness [77.29026280120277]
We assess the empirical robustness of node embedding models to random and adversarial poisoning attacks.
We compare edge addition, deletion and rewiring strategies computed using network properties as well as node labels.
We found that node classification suffers from higher performance degradation as opposed to network reconstruction.
arXiv Detail & Related papers (2022-09-16T17:20:23Z) - Topology-Preserving Segmentation Network: A Deep Learning Segmentation
Framework for Connected Component [7.95119530218428]
In medical imaging, the topology of the structure, such as the kidney or lung, is usually known.
A it topology-preserving segmentation network (TPSN) is trained to give an accurate segmentation result.
TPSN is a deformation-based model that yields a deformation map through a UNet.
A multi-scale TPSN is developed in this paper that incorporates multi-level information of images to produce more precise segmentation results.
arXiv Detail & Related papers (2022-02-27T09:56:33Z) - TA-Net: Topology-Aware Network for Gland Segmentation [71.52681611057271]
We propose a novel topology-aware network (TA-Net) to accurately separate densely clustered and severely deformed glands.
TA-Net has a multitask learning architecture and enhances the generalization of gland segmentation.
It achieves state-of-the-art performance on the two datasets.
arXiv Detail & Related papers (2021-10-27T17:10:58Z) - Multipath CNN with alpha matte inference for knee tissue segmentation
from MRI [2.064612766965483]
This paper presents a deep learning based automatic segmentation framework for knee tissue segmentation.
A novel multipath CNN-based method is proposed, which consists of a decoder-based segmentation network in combination with a low rank tensor-reconstructed segmentation network.
To further improve the segmentation from CNNs, trimap generation, which effectively utilizes superimposed regions, is proposed.
arXiv Detail & Related papers (2021-09-29T07:48:47Z) - Topological Similarity Index and Loss Function for Blood Vessel
Segmentation [0.0]
We propose a similarity index which captures the consistency of the predicted segmentations having as reference the ground truth.
We also design a novel loss function based on the morphological closing operator and show how it allows to learn deep neural network models which produce more topologically coherent masks.
arXiv Detail & Related papers (2021-07-30T10:24:47Z) - CAMERAS: Enhanced Resolution And Sanity preserving Class Activation
Mapping for image saliency [61.40511574314069]
Backpropagation image saliency aims at explaining model predictions by estimating model-centric importance of individual pixels in the input.
We propose CAMERAS, a technique to compute high-fidelity backpropagation saliency maps without requiring any external priors.
arXiv Detail & Related papers (2021-06-20T08:20:56Z) - Few-shot Medical Image Segmentation using a Global Correlation Network
with Discriminative Embedding [60.89561661441736]
We propose a novel method for few-shot medical image segmentation.
We construct our few-shot image segmentor using a deep convolutional network trained episodically.
We enhance discriminability of deep embedding to encourage clustering of the feature domains of the same class.
arXiv Detail & Related papers (2020-12-10T04:01:07Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - Cross-stained Segmentation from Renal Biopsy Images Using Multi-level
Adversarial Learning [13.30545860115548]
We design a robust and flexible model for cross-stained segmentation.
It is able to improve segmentation performance on target type of stained images and use unlabeled data to achieve similar accuracy to labeled data.
arXiv Detail & Related papers (2020-02-20T06:49:48Z)
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.