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 | Size | Format | |
---|---|---|---|---|
FullText.pdf | Copyright © 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 MB | Adobe PDF | View/Open |
Items in BURA are protected by copyright, with all rights reserved, unless otherwise indicated.