Implicit Rank-Minimizing Autoencoder
Li Jing, Jure Zbontar, Yann LeCun
Abstract
An important component of autoencoders is the method by which the information capacity of the latent representation is minimized or limited. In this work, the rank of the covariance matrix of the codes is implicitly minimized by relying on the fact that gradient descent learning in multi-layer linear networks leads to minimum-rank solutions. By inserting a number of extra linear layers between the encoder and the decoder, the system spontaneously learns representations with a low effective dimension. The model, dubbed Implicit Rank-Minimizing Autoencoder (IRMAE), is simple, deterministic, and learns compact latent spaces. We demonstrate the validity of the method on several image generation and representation learning tasks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext cb5b985b-50a3-4573-ba59-d94e5a019c9aCited by top-tier papers16
- Understanding Dimensional Collapse in Contrastive Self-supervised LearningLi Jing, Pascal Vincent, Yann LeCun, Yuandong TianICLR 2022 · 467 citations
- How Mask Matters: Towards Theoretical Understandings of Masked AutoencodersQi Zhang, Yifei Wang, Yisen WangNeurIPS 2022 · 119 citations
- Disentanglement via Latent QuantizationKyle Hsu, William Dorrell, James C. R. Whittington, Jiajun Wu et al.NeurIPS 2023 · 54 citations
- When Is Unsupervised Disentanglement Possible?Daniella Horan, Eitan Richardson, Yair WeissNeurIPS 2021 · 54 citations
- Learning to Accelerate Partial Differential Equations via Latent Global EvolutionTailin Wu, Takashi Maruyama, Jure LeskovecNeurIPS 2022 · 49 citations
Builds on5
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- From Variational to Deterministic AutoencodersPartha Ghosh, Mehdi S. M. Sajjadi, Antonio Vergari, Michael J. Black et al.ICLR 2020 · 298 citations
- Implicit Regularization in Deep Learning May Not Be Explainable by NormsNoam Razin, Nadav CohenNeurIPS 2020 · 178 citations
- Self-Supervised Learning of Pretext-Invariant RepresentationsIshan Misra, Laurens van der MaatenCVPR 2020
- Momentum Contrast for Unsupervised Visual Representation LearningKaiming He, Haoqi Fan, Yuxin Wu, Saining Xie et al.CVPR 2020
Related papers
- Regularized linear autoencoders recover the principal components, eventuallyXuchan Bao, James Lucas, Sushant Sachdeva, Roger B. GrosseNeurIPS 2020 · 40 citations
- Fundamental Limits of Two-layer Autoencoders, and Achieving Them with Gradient MethodsAleksandr Shevchenko, Kevin Kögler, Hamed Hassani, Marco MondelliICML 2023 · 3 citations
- Implicit Bias of Large Depth Networks: a Notion of Rank for Nonlinear FunctionsArthur JacotICLR 2023 · 2 citations
- Eliminating the Invariance on the Loss Landscape of Linear AutoencodersReza Oftadeh, Jiayi Shen, Zhangyang Wang, Dylan A. ShellICML 2020 · 12 citations
- Bit Prioritization in Variational Autoencoders via Progressive CodingRui Shu, Stefano ErmonICML 2022 · 9 citations
