Pyramid U-Net for Retinal Vessel Segmentation
- URL: http://arxiv.org/abs/2104.02333v1
- Date: Tue, 6 Apr 2021 07:33:52 GMT
- Title: Pyramid U-Net for Retinal Vessel Segmentation
- Authors: Jiawei Zhang, Yanchun Zhang, Xiaowei Xu
- Abstract summary: We propose pyramid U-Net for accurate retinal vessel segmentation.
The proposed pyramid-scale aggregation blocks (PSABs) are employed in both the encoder and decoder.
Our pyramid U-Net outperforms the current state-of-the-art methods on the public DRIVE and CHASE-DB1 datasets.
- Score: 22.91204798022379
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Retinal blood vessel can assist doctors in diagnosis of eye-related diseases
such as diabetes and hypertension, and its segmentation is particularly
important for automatic retinal image analysis. However, it is challenging to
segment these vessels structures, especially the thin capillaries from the
color retinal image due to low contrast and ambiguousness. In this paper, we
propose pyramid U-Net for accurate retinal vessel segmentation. In pyramid
U-Net, the proposed pyramid-scale aggregation blocks (PSABs) are employed in
both the encoder and decoder to aggregate features at higher, current and lower
levels. In this way, coarse-to-fine context information is shared and
aggregated in each block thus to improve the location of capillaries. To
further improve performance, two optimizations including pyramid inputs
enhancement and deep pyramid supervision are applied to PSABs in the encoder
and decoder, respectively. For PSABs in the encoder, scaled input images are
added as extra inputs. While for PSABs in the decoder, scaled intermediate
outputs are supervised by the scaled segmentation labels. Extensive evaluations
show that our pyramid U-Net outperforms the current state-of-the-art methods on
the public DRIVE and CHASE-DB1 datasets.
Related papers
- MDNet: Multi-Decoder Network for Abdominal CT Organs Segmentation [6.4987174473651725]
We propose a textbftextitacMDNet to handle challenges of heterogeneity in organ shapes, sizes, and complex anatomical relationships.
textitacMDNet is an encoder-decoder network that uses the pre-trained textitMiT-B2 as the encoder and multiple different decoder networks.
textitacMDNet is more interpretable and robust compared to the other baseline models.
arXiv Detail & Related papers (2024-05-10T01:03:03Z) - Light-weight Retinal Layer Segmentation with Global Reasoning [14.558920359236572]
We propose LightReSeg for retinal layer segmentation which can be applied to OCT images.
Our approach achieves a better segmentation performance compared to the current state-of-the-art method TransUnet.
arXiv Detail & Related papers (2024-04-25T05:42:41Z) - G-CASCADE: Efficient Cascaded Graph Convolutional Decoding for 2D
Medical Image Segmentation [8.550528610846456]
We propose a new graph convolution-based decoder namely, Cascaded Graph Convolutional Attention Decoder (G-CASCADE)
G-CASCADE progressively refines multi-stage feature maps generated by hierarchical transformer encoders with an efficient graph convolution block.
Our decoder can easily be used with other hierarchical encoders for general-purpose semantic and medical image segmentation tasks.
arXiv Detail & Related papers (2023-10-24T20:41:04Z) - Feature Enhancer Segmentation Network (FES-Net) for Vessel Segmentation [19.455350961592742]
We propose a novel feature enhancement segmentation network (FES-Net) that achieves accurate pixel-wise segmentation without requiring additional image enhancement steps.
FES-Net directly processes the input image and utilizes four prompt convolutional blocks (PCBs) during downsampling.
We evaluate the performance of FES-Net on four publicly available state-of-the-art datasets: DRIVE, STARE, CHASE, and HRF.
arXiv Detail & Related papers (2023-09-07T07:46:46Z) - M$^{2}$SNet: Multi-scale in Multi-scale Subtraction Network for Medical
Image Segmentation [73.10707675345253]
We propose a general multi-scale in multi-scale subtraction network (M$2$SNet) to finish diverse segmentation from medical image.
Our method performs favorably against most state-of-the-art methods under different evaluation metrics on eleven datasets of four different medical image segmentation tasks.
arXiv Detail & Related papers (2023-03-20T06:26:49Z) - Focused Decoding Enables 3D Anatomical Detection by Transformers [64.36530874341666]
We propose a novel Detection Transformer for 3D anatomical structure detection, dubbed Focused Decoder.
Focused Decoder leverages information from an anatomical region atlas to simultaneously deploy query anchors and restrict the cross-attention's field of view.
We evaluate our proposed approach on two publicly available CT datasets and demonstrate that Focused Decoder not only provides strong detection results and thus alleviates the need for a vast amount of annotated data but also exhibits exceptional and highly intuitive explainability of results via attention weights.
arXiv Detail & Related papers (2022-07-21T22:17:21Z) - SPNet: A novel deep neural network for retinal vessel segmentation based
on shared decoder and pyramid-like loss [13.021014899410684]
convolutional neural networks have shown significant ability to extract the blood vessel structure.
We propose a novel deep neural network for retinal vessel segmentation based on shared decoder and pyramid-like loss.
arXiv Detail & Related papers (2022-02-19T03:44:34Z) - Small Lesion Segmentation in Brain MRIs with Subpixel Embedding [105.1223735549524]
We present a method to segment MRI scans of the human brain into ischemic stroke lesion and normal tissues.
We propose a neural network architecture in the form of a standard encoder-decoder where predictions are guided by a spatial expansion embedding network.
arXiv Detail & Related papers (2021-09-18T00:21:17Z) - Crosslink-Net: Double-branch Encoder Segmentation Network via Fusing
Vertical and Horizontal Convolutions [58.71117402626524]
We present a novel double-branch encoder architecture for medical image segmentation.
Our architecture is inspired by two observations: 1) Since the discrimination of features learned via square convolutional kernels needs to be further improved, we propose to utilize non-square vertical and horizontal convolutional kernels.
The experiments validate the effectiveness of our model on four datasets.
arXiv Detail & Related papers (2021-07-24T02:58:32Z) - Hierarchical Deep Network with Uncertainty-aware Semi-supervised
Learning for Vessel Segmentation [58.45470500617549]
We propose a hierarchical deep network where an attention mechanism localizes the low-contrast capillary regions guided by the whole vessels.
The proposed method achieves the state-of-the-art performance in the benchmarks of both retinal artery/vein segmentation in fundus images and liver portal/hepatic vessel segmentation in CT images.
arXiv Detail & Related papers (2021-05-31T06:55:43Z) - 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)
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.