Implicit U-Net for volumetric medical image segmentation

06/30/2022
by   Sergio Naval Marimont, et al.
14

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 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 as training memory footprint while achieving comparable results in our experiments with two different abdominal CT scan datasets.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset