iUNets: Fully invertible U-Nets with Learnable Up- and Downsampling
Machine Learning
2020-07-01 v3 Computer Vision and Pattern Recognition
Image and Video Processing
Abstract
U-Nets have been established as a standard architecture for image-to-image learning problems such as segmentation and inverse problems in imaging. For large-scale data, as it for example appears in 3D medical imaging, the U-Net however has prohibitive memory requirements. Here, we present a new fully-invertible U-Net-based architecture called the iUNet, which employs novel learnable and invertible up- and downsampling operations, thereby making the use of memory-efficient backpropagation possible. This allows us to train deeper and larger networks in practice, under the same GPU memory restrictions. Due to its invertibility, the iUNet can furthermore be used for constructing normalizing flows.
Cite
@article{arxiv.2005.05220,
title = {iUNets: Fully invertible U-Nets with Learnable Up- and Downsampling},
author = {Christian Etmann and Rihuan Ke and Carola-Bibiane Schönlieb},
journal= {arXiv preprint arXiv:2005.05220},
year = {2020}
}