City Research Online

Implicit U-Net for Volumetric Medical Image Segmentation

Marimont, S. N. & Tarroni, G. ORCID: 0000-0002-0341-6138 (2022). Implicit U-Net for Volumetric Medical Image Segmentation. Paper presented at the Medical Image Understanding and Analysis 26th Annual Conference, MIUA 2022, 27-29 Jul 2022, Cambridge, UK. doi: 10.1007/978-3-031-12053-4_29

Abstract

U-Net has been the go-to architecture for medical image segmentation tasks, however computational challenges arise when extending the U-Net architecture to 3D images. We propose the Implicit U-Net architecture that adapts the efficient Implicit Representation paradigm to supervised image segmentation tasks. By combining a convolutional feature extractor with an implicit localization network, our implicit U-Net has 40% less parameters than the equivalent U-Net. Moreover, we propose training and inference procedures to capitalize sparse predictions. When comparing to an equivalent fully convolutional U-Net, Implicit U-Net reduces by approximately 30% inference and training time as well as training memory footprint while achieving comparable results in our experiments with two different abdominal CT scan datasets.

Publication Type: Conference or Workshop Item (Paper)
Additional Information: © 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG. Use must not be for Commercial Purposes.
Publisher Keywords: Efficient segmentation, Supervised learning, Volumetric segmentation, CT
Subjects: Q Science > QA Mathematics > QA75 Electronic computers. Computer science
Departments: School of Science & Technology > Computer Science
[img] Text - Accepted Version
This document is not freely accessible until 25 July 2023 due to copyright restrictions.

To request a copy, please use the button below.

Request a copy

Export

Downloads

Downloads per month over past year

View more statistics

Actions (login required)

Admin Login Admin Login