Please use this identifier to cite or link to this item: http://bura.brunel.ac.uk/handle/2438/30906
Title: Segmenting Medical Images: From UNet to Res-UNet and nnUNet
Authors: Huang, L
Miron, A
Hone, K
Li, Y
Keywords: deep learning;UNet;Res-UNet;Attention Res-UNet;nnUNet;medical imaging segmentation;clinical application
Issue Date: 26-Jun-2024
Publisher: Institute of Electrical and Electronics Engineers (IEEE)
Citation: Huang, L. et al. (2024) 'Segmenting Medical Images: From UNet to Res-UNet and nnUNet', 2024 IEEE 37th International Symposium on Computer-Based Medical Systems (CBMS), Guadalajara, Mexico, 26-28 June, pp. 483 - 489. doi: 10.1109/CBMS61543.2024.00086.
Abstract: This study provides a comparative analysis of deep learning models—UNet, Res-UNet, Attention Res-UNet, and nnUNet—evaluating their performance in brain tumour, polyp, and multi-class heart segmentation tasks. The analysis focuses on precision, accuracy, recall, Dice Similarity Coefficient (DSC), and Intersection over Union (IoU) to assess their clinical applicability. In brain tumour segmentation, Res-UNet and nnUNet significantly outperformed UNet, with Res-UNet leading in DSC and IoU scores, indicating superior accuracy in tumour delineation. Meanwhile, nnUNet excelled in recall and accuracy, which are crucial for reliable tumour detection in clinical diagnosis and planning. In polyp detection, nnUNet was the most effective, achieving the highest metrics across all categories and proving itself as a reliable diagnostic tool in endoscopy. In the complex task of heart segmentation, Res-UNet and Attention Res-UNet were outstanding in delineating the left ventricle, with Res-UNet also leading in right ventricle segmentation. nnUNet was unmatched in myocardium segmentation, achieving top scores in precision, recall, DSC, and IoU. The conclusion notes that although ResUNet occasionally outperforms nnUNet in specific metrics, the differences are quite small. Moreover, nnUNet consistently shows superior overall performance across the experiments. Particularly noted for its high recall and accuracy, which are crucial in clinical settings to minimize misdiagnosis and ensure timely treatment, nnUNet’s robust performance in crucial metrics across all tested categories establishes it as the most effective model for these varied and complex segmentation tasks.
URI: https://bura.brunel.ac.uk/handle/2438/30906
DOI: https://doi.org/10.1109/CBMS61543.2024.00086
ISBN: 979-8-3503-8472-7 (ebk)
979-8-3503-8473-4 (PoD)
ISSN: 2372-918X
Other Identifiers: ORCiD: Alina Miron https://orcid.org/0000-0002-0068-4495
ORCiD: Kate Hone https://orcid.org/0000-0001-5394-8354
ORCiD: Yongmin Li https://orcid.org/0000-0003-1668-2440
Appears in Collections:Dept of Computer Science Research Papers

Files in This Item:
File Description SizeFormat 
FullText.pdfCopyright © 2024 Institute of Electrical and Electronics Engineers (IEEE). Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works. See: https://journals.ieeeauthorcenter.ieee.org/become-an-ieee-journal-author/publishing-ethics/guidelines-and-policies/post-publication-policies/1.88 MBAdobe PDFView/Open


Items in BURA are protected by copyright, with all rights reserved, unless otherwise indicated.