arXiv · 2303.11239
Training Invertible Neural Networks as Autoencoders
Abstract
Autoencoders are able to learn useful data representations in an unsupervised matter and have been widely used in various machine learning and computer vision tasks. In this work, we present methods to train Invertible Neural Networks (INNs) as (variational) autoencoders which we call INN (variational) autoencoders. Our experiments on MNIST, CIFAR and CelebA show that for low bottleneck sizes our INN autoencoder achieves results similar to the classical autoencoder. However, for large bottleneck sizes our INN autoencoder outperforms its classical counterpart. Based on the empirical results, we hypothesize that INN autoencoders might not have any intrinsic information loss and thereby are not bounded to a maximal number of layers (depth) after which only suboptimal results can be achieved.
Explore related subjects
Keep this discovery
The-Gia Leo Nguyen, Lynton Ardizzone, Ullrich Köthe. 2023-03-20. Training Invertible Neural Networks as Autoencoders. https://doi.org/10.1007/978-3-030-33676-9_31
Cite the original work for its findings. Save a collection to share your selection of sources.