A Comparative Analysis on Architectures of UNet Family for Image Segmentation

Authors

  • Yueru Gong

DOI:

https://doi.org/10.54097/0chpy661

Keywords:

Image segmentation, deep learning, UNet.

Abstract

Image segmentation is an important visual task that involves categorizing each pixel within an image, assigning it a specific label that defines its content. The advancement of deep learning has revolutionized this field, with Convolutional Neural Networks (CNNs) marking a significant improvement in accuracy and efficiency. These models learn from the raw image data to the segmentation labels in an end-to-end manner, enabling them to produce precise segmentation outcomes with minimal human intervention. Among the spectrum of segmentation models, the UNet architecture has emerged as a standout performer, renowned for its robust performance and adaptability. It has not only set a high benchmark but also inspired a plethora of enhancements and adaptations to cater to various applications. This paper presents a systematic analysis of UNet, hoping to share the UNet model with its extensions. So that beginners can have a preliminary understanding of this famous family of deep-learning network models. In this paper, the author first introduces the basic UNet model. Then introduce eight classic UNet variants. This paper will guide the reader to understand the compilation and improvement principles of the UNet family and provide ideas and a foundation for the reader's subsequent research on UNet.

Downloads

Download data is not yet available.

References

[1] Ronneberger Olaf, Philipp Fischer, Thomas Brox. U-net: Convolutional networks for biomedical image segmentation. Medical image computing and computer-assisted intervention, 2015: 234-241.

[2] He Kaiming, Zhang Xiangyu, Ren Shaoqing, Sun Jian. Deep residual learning for image recognition. Proceedings of the IEEE conference on computer vision and pattern recognition. 2016: 770-778.

[3] Xiao Xiao, Lian Shen, Luo Zhiming, Li Shaozi. Weighted res-unet for high-quality retina vessel segmentation. International conference on information technology in medicine and education, 2018: 327-331.

[4] Alom Md Zahangir, Hasan Mahmudul, Yakopcic Chris, et al. Recurrent residual convolutional neural network based on u-net (r2u-net) for medical image segmentation. arXiv preprint, 2018: 1802.06955.

[5] Ibtehaz Nabil, M. Sohel Rahman. MultiResUNet: Rethinking the U-Net architecture for multimodal biomedical image segmentation. Neural networks, 2020, 121: 74-87.

[6] Zhou Zongwei, Siddiquee Md Mahfuzur Rahman, Tajbakhsh Nima, Liang Jianming. Unet++: Redesigning skip connections to exploit multiscale features in image segmentation. IEEE transactions on medical imaging, 2019 39 (6): 1856-1867.

[7] Huang Huimin, Lin Lanfen, Tong Ruofeng, et al. Unet 3+: A full-scale connected unet for medical image segmentation. IEEE international conference on acoustics, speech and signal processing, 2020: 1055-1059.

[8] Oktay Ozan, Schlemper Jo, Folgoc Loic Le, et al. Attention u-net: Learning where to look for the pancreas. arXiv preprint, 2018: 1804.03999.

[9] Chen Jieneng, Lu Yongyi, Yu Qihang, et al. Transunet: Transformers make strong encoders for medical image segmentation. arXiv preprint, 2021: 2102.04306.

[10] Çiçek Özgün, Abdulkadir Ahmed, Lienkamp Soeren, et al. 3D U-Net: learning dense volumetric segmentation from sparse annotation. Medical Image Computing and Computer-Assisted Intervention, 2016: 424-432.

Downloads

Published

25-12-2024