Please use this identifier to cite or link to this item:
http://bura.brunel.ac.uk/handle/2438/27263
Title: | Advanced Architectural Variations of nnUNet |
Authors: | McConnell, N Ndipenoch, N Cao, Y Miron, A Li, Y |
Keywords: | biomedical image segmentation;nnUnet;residual;dense;inception;attention |
Issue Date: | 30-Sep-2023 |
Publisher: | Elsevier |
Citation: | McConnell, N. et al. (2023) 'Advanced Architectural Variations of nnUNet', Neurocomputing, 560, 126837, pp. 1 - 15. doi: 10.1016/j.neucom.2023.126837. |
Abstract: | The nnUNet is a state-of-the-art deep learning based segmentation framework which automatically and systematically configures the entire network training pipeline. We extend the network architecture component of the nnUNet framework by newly integrating mechanisms from advanced U-Net variations including residual, dense, and inception blocks as well as three forms of the attention mechanism. We propose the following extensions to nnUNet, namely Residual-nnUNet, Dense-nnUNet, Inception-nnUNet, Spatial-Single-Attention-nnUNet, Spatial- Multi-Attention-nnUNet, and Channel-Spatial-Attention-nnUNet. Furthermore, within Channel-Spatial- Attention-nnUNet we integrate our newly proposed variation of the channel-attention mechanism. We demonstrate that use of the nnUNet allows for consistent and transparent comparison of U-Net architectural modifications, while maintaining network architecture as the sole independent variable across experiments with respect to a dataset. The proposed variants are evaluated on eight medical imaging datasets consisting of 20 anatomical regions which is the largest collection of datasets on which attention U-Net variants have been compared in a single work. Our results suggest that attention variants are effective at improving performance when applied to tumour segmentation tasks consisting of two or more target anatomical regions, and that segmentation performance is influenced by use of the deep supervision architectural feature. |
Description: | Data availability: Data utilised is publicly available and link has been provided. Source code available via: https://github.com/niccolo246/Advanced_nnUNet.git |
URI: | https://bura.brunel.ac.uk/handle/2438/27263 |
DOI: | https://doi.org/10.1016/j.neucom.2023.126837 |
ISSN: | 0925-2312 |
Other Identifiers: | ORCID iD: Alina Miron https://orcid.org/0000-0002-0068-4495 ORCID iD: Yongmin Li https://orcid.org/0000-0003-1668-2440. |
Appears in Collections: | Dept of Computer Science Embargoed Research Papers |
Files in This Item:
File | Description | Size | Format | |
---|---|---|---|---|
FullText.pdf | Embargoed until 30 September 2024 | 10.84 MB | Adobe PDF | View/Open |
This item is licensed under a Creative Commons License