DoubleU-Net: A Deep Convolutional Neural Network for Medical Image Segmentation

06/08/2020
by   Debesh Jha, et al.
64

Semantic image segmentation is the process of labeling each pixel of an image with its corresponding class. An encoder-decoder based approach, like U-Net and its variants, is a popular strategy for solving medical image segmentation tasks. To improve the performance of U-Net on various segmentation tasks, we propose a novel architecture called DoubleU-Net, which is a combination of two U-Net architectures stacked on top of each other. The first U-Net uses a pre-trained VGG-19 as the encoder, which has already learned features from ImageNet and can be transferred to another task easily. To capture more semantic information efficiently, we added another U-Net at the bottom. We also adopt Atrous Spatial Pyramid Pooling (ASPP) to capture contextual information within the network. We have evaluated DoubleU-Net using four medical segmentation datasets, covering various imaging modalities such as colonoscopy, dermoscopy, and microscopy. Experiments on the MICCAI 2015 segmentation challenge, the CVC-ClinicDB, the 2018 Data Science Bowl challenge, and the Lesion boundary segmentation datasets demonstrate that the DoubleU-Net outperforms U-Net and the baseline models. Moreover, DoubleU-Net produces more accurate segmentation masks, especially in the case of the CVC-ClinicDB and MICCAI 2015 segmentation challenge datasets, which have challenging images such as smaller and flat polyps. These results show the improvement over the existing U-Net model. The encouraging results, produced on various medical image segmentation datasets, show that DoubleU-Net can be used as a strong baseline for both medical image segmentation and cross-dataset evaluation testing to measure the generalizability of Deep Learning (DL) models.

READ FULL TEXT

page 1

page 4

page 5

page 6

research
05/11/2020

Medical Image Segmentation Using a U-Net type of Architecture

Deep convolutional neural networks have been proven to be very effective...
research
09/19/2019

Deep learning architectures for automated image segmentation

Image segmentation is widely used in a variety of computer vision tasks,...
research
09/06/2022

An evaluation of U-Net in Renal Structure Segmentation

Renal structure segmentation from computed tomography angiography (CTA) ...
research
12/12/2022

Efficient Bayesian Uncertainty Estimation for nnU-Net

The self-configuring nnU-Net has achieved leading performance in a large...
research
01/17/2018

TernausNet: U-Net with VGG11 Encoder Pre-Trained on ImageNet for Image Segmentation

Pixel-wise image segmentation is demanding task in computer vision. Clas...
research
01/08/2020

A context based deep learning approach for unbalanced medical image segmentation

Automated medical image segmentation is an important step in many medica...
research
09/27/2018

nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation

The U-Net was presented in 2015. With its straight-forward and successfu...

Please sign up or login with your details

Forgot password? Click here to reset