Skip to content

MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation

License

Notifications You must be signed in to change notification settings

nibtehaz/MultiResUNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MultiResUNet

Rethinking the U-Net architecture for multimodal biomedical image segmentation

This repository contains the original implementation of "MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation" in Keras (Tensorflow as backend).

Paper

MultiResUNet has been published in Neural Networks

Ibtehaz, Nabil, and M. Sohel Rahman. "MultiResUNet: Rethinking the U-Net architecture for multimodal biomedical image segmentation." Neural Networks 121 (2020): 74-87.

Overview

In this project we take motivations from the phenomenal U-Net architecture for biomedical image segmentation and take an attempt to improve the already outstanding network.

In order to incorporate multiresolution analysis, taking inspiration from Inception family networks, we propose the following MultiRes block, and replace the pair of convolutional layer pairs in the original U-Net with it. This configuration basically is derived from factorizing 5x5 and 7x7 convolution operations to 3x3 ones, and reusing them to obtain results from 3x3, 5x5 and 7x7 convolution operations simultaneously. Moreover, a residual path is also added.

Consequnetly, to elleviate the likely semantic distance between Encoder and Decoder networks, we introduce Res Paths. We include additional convolutions along the shortcut path, in proportionate to the expected gap between the two corresponding layers.

Therefore, with the fusion of MultiRes blocks and Res paths, we obtain the proposed architecture MultiResUNet.

Codes

The model architecture codes can be found in

Tensorflow

Pytorch

Demo

A demo can be found in here (tensorflow)

License

License

MIT license

Citation Request

If you use MultiResUNet in your project, please cite the following paper

@article{ibtehaz2020multiresunet,
  title={MultiResUNet: Rethinking the U-Net architecture for multimodal biomedical image segmentation},
  author={Ibtehaz, Nabil and Rahman, M Sohel},
  journal={Neural Networks},
  volume={121},
  pages={74--87},
  year={2020},
  publisher={Elsevier}
}

About

MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation

Topics

Resources

License

Stars

Watchers

Forks