DRU-net: An Efficient Deep Convolutional Neural Network for Medical Image Segmentation
Abstract
Residual network (ResNet) and densely connected network (DenseNet) have significantly improved the training efficiency and performance of deep convolutional neural networks (DCNNs) mainly for object classification tasks. In this paper, we propose an efficient network architecture by considering advantages of both networks. The proposed method is integrated into an encoder-decoder DCNN model for medical image segmentation. Our method adds additional skip connections compared to ResNet but uses significantly fewer model parameters than DenseNet. We evaluate the proposed method on a public dataset (ISIC 2018 grand-challenge) for skin lesion segmentation and a local brain MRI dataset. In comparison with ResNet-based, DenseNet-based and attention network (AttnNet) based methods within the same encoder-decoder network structure, our method achieves significantly higher segmentation accuracy with fewer number of model parameters than DenseNet and AttnNet. The code is available on GitHub (GitHub link: https://github.com/MinaJf/DRU-net).
Cite
@article{arxiv.2004.13453,
title = {DRU-net: An Efficient Deep Convolutional Neural Network for Medical Image Segmentation},
author = {Mina Jafari and Dorothee Auer and Susan Francis and Jonathan Garibaldi and Xin Chen},
journal= {arXiv preprint arXiv:2004.13453},
year = {2020}
}
Comments
Accepted for publication at IEEE International Symposium on Biomedical Imaging (ISBI) 2020, 5 pages, 3 figures