DeepAI AI Chat
Log In Sign Up

Bi-Directional ConvLSTM U-Net with Densley Connected Convolutions

by   Reza Azad, et al.
University of Barcelona
Iran University of Science and Technology

In recent years, deep learning-based networks have achieved state-of-the-art performance in medical image segmentation. Among the existing networks, U-Net has been successfully applied on medical image segmentation. In this paper, we propose an extension of U-Net, Bi-directional ConvLSTM U-Net with Densely connected convolutions (BCDU-Net), for medical image segmentation, in which we take full advantages of U-Net, bi-directional ConvLSTM (BConvLSTM) and the mechanism of dense convolutions. Instead of a simple concatenation in the skip connection of U-Net, we employ BConvLSTM to combine the feature maps extracted from the corresponding encoding path and the previous decoding up-convolutional layer in a non-linear way. To strengthen feature propagation and encourage feature reuse, we use densely connected convolutions in the last convolutional layer of the encoding path. Finally, we can accelerate the convergence speed of the proposed network by employing batch normalization (BN). The proposed model is evaluated on three datasets of: retinal blood vessel segmentation, skin lesion segmentation, and lung nodule segmentation, achieving state-of-the-art performance.


page 4

page 5

page 7

page 8


Multi-level Context Gating of Embedded Collective Knowledge for Medical Image Segmentation

Medical image segmentation has been very challenging due to the large va...

BUSU-Net: An Ensemble U-Net Framework for Medical Image Segmentation

In recent years, convolutional neural networks (CNNs) have revolutionize...

Dense Multi-path U-Net for Ischemic Stroke Lesion Segmentation in Multiple Image Modalities

Delineating infarcted tissue in ischemic stroke lesions is crucial to de...

A Dense Siamese U-Net trained with Edge Enhanced 3D IOU Loss for Image Co-segmentation

Image co-segmentation has attracted a lot of attentions in computer visi...

Contextual Hourglass Networks for Segmentation and Density Estimation

Hourglass networks such as the U-Net and V-Net are popular neural archit...

CAggNet: Crossing Aggregation Network for Medical Image Segmentation

In this paper, we present Crossing Aggregation Network (CAggNet), a nove...

Code Repositories


BCDU-Net : Medical Image Segmentation

view repo


Term Project for Jake Hellman, Shashank Manjunath, and Simon Gilbert

view repo