iUNets: Fully invertible U-Nets with Learnable Up- and Downsampling

05/11/2020
by   Christian Etmann, et al.
17

U-Nets have been established as a standard neural network design architecture for image-to-image learning problems such as segmentation and inverse problems in imaging. For high-dimensional applications, as they for example appear in 3D medical imaging, U-Nets however have prohibitive memory requirements. Here, we present a new fully-invertible U-Net-based architecture called the iUNet, which allows for the application of highly memory-efficient backpropagation procedures. For this, we introduce learnable and invertible up- and downsampling operations. An open source library in Pytorch for 1D, 2D and 3D data is made available.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset