Convolutional Networks Versus Transformers: A Comparison in Prostate Segmentation

Fernando Vásconez, Maria Baldeon Calisto, Daniel Riofrío, Zhouping Wei, Yoga Balagurunathan

2023

Abstract

Prostate cancer is one of the most common types of cancer that affects men. One way to diagnose and treat it is by manually segmenting the prostate region and analyzing its size or consistency in MRI scans. However, this process requires an experienced radiologist, is time-consuming, and prone to human error. Convolutional Neural Networks (CNNs) have been successful at automating the segmentation of the prostate. In particular, the U-Net architecture has become the de-facto standard given its performance and efficacy. However, CNNs are unable to model long-range dependencies. Transformer networks have emerged as an alternative, obtaining better results than CNNs in image analysis when a large dataset is available for training. In this work, the residual U-Net and the transformer UNETR are compared in the task of prostate segmentation on the ProstateX dataset in terms of segmentation accuracy and computational complexity. Furthermore, to analyze the impact of the size of the dataset, four training datasets are formed with 30, 60, 90, and 120 images. The experiments show that the CNN architecture has a statistical higher performance when the dataset has 90 or 120 images. When the dataset has 60 images, both architectures have a statistical similar performance, while when the dataset has 30 images UNETR performs marginally better. Considering the complexity, the UNETR has 5x more parameters and takes 5.8x more FLOPS than the residual U-Net. Therefore, showing that in the case of prostate segmentation CNNs have an overall better performance than Transformer networks.

Paper Citation

in Harvard Style

Vásconez F., Baldeon Calisto M., Riofrío D., Wei Z. and Balagurunathan Y. (2023). Convolutional Networks Versus Transformers: A Comparison in Prostate Segmentation. In Proceedings of the 15th International Conference on Agents and Artificial Intelligence - Volume 3: ICAART, ISBN 978-989-758-623-1, pages 600-607. DOI: 10.5220/0011717600003393

in Bibtex Style

@conference{icaart23,
author={Fernando Vásconez and Maria Baldeon Calisto and Daniel Riofrío and Zhouping Wei and Yoga Balagurunathan},
title={Convolutional Networks Versus Transformers: A Comparison in Prostate Segmentation},
booktitle={Proceedings of the 15th International Conference on Agents and Artificial Intelligence - Volume 3: ICAART,},
year={2023},
pages={600-607},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0011717600003393},
isbn={978-989-758-623-1},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 15th International Conference on Agents and Artificial Intelligence - Volume 3: ICAART,
TI - Convolutional Networks Versus Transformers: A Comparison in Prostate Segmentation
SN - 978-989-758-623-1
AU - Vásconez F.
AU - Baldeon Calisto M.
AU - Riofrío D.
AU - Wei Z.
AU - Balagurunathan Y.
PY - 2023
SP - 600
EP - 607
DO - 10.5220/0011717600003393