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

8 Jun 2020  ·  Debesh Jha, Michael A. Riegler, Dag Johansen, Pål Halvorsen, Håvard D. Johansen ·

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.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Medical Image Segmentation 2015 MICCAI Polyp Detection DoubleUNet Dice 0.7649 # 1
Medical Image Segmentation 2018 Data Science Bowl DoubleUNet Dice 0.9133 # 5
mIoU 0.8407 # 7
Recall 0.6407 # 4
Precision 0.9596 # 1
Medical Image Segmentation CVC-ClinicDB DoubleUNet mean Dice 0.9239 # 22
Lesion Segmentation ISIC 2018 DoubleU-Net Dice Score 0.8962 # 7
Medical Image Segmentation Kvasir-Instrument DoubleUNet DSC 0.9038 # 3
Semantic Segmentation Kvasir-Instrument DoubleUNet DSC 0.9038 # 1
mIoU 0.8430 # 2

Methods