Fully Bayesian Autoencoders with Latent Sparse Gaussian Processes
Ba-Hien Tran, Babak Shahbaba, Stephan Mandt, Maurizio Filippone
Abstract
We present a fully Bayesian autoencoder model that treats both local latent variables and global decoder parameters in a Bayesian fashion. This approach allows for flexible priors and posterior approximations while keeping the inference costs low. To achieve this, we introduce an amortized MCMC approach by utilizing an implicit stochastic network to learn sampling from the posterior over local latent variables. Furthermore, we extend the model by incorporating a Sparse Gaussian Process prior over the latent space, allowing for a fully Bayesian treatment of inducing points and kernel hyperparameters and leading to improved scalability. Additionally, we enable Deep Gaussian Process priors on the latent space and the handling of missing data. We evaluate our model on a range of experiments focusing on dynamic representation learning and generative modeling, demonstrating the strong performance of our approach in comparison to existing methods that combine Gaussian Processes and autoencoders.
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.
Cited by top-tier papers5
- Preventing Model Collapse in Gaussian Process Latent Variable ModelsYing Li, Zhidi Lin, Feng Yin, Michael Minyi ZhangICML 2024 · 5 citations
- One-Line-of-Code Data Mollification Improves Optimization of Likelihood-based Generative ModelsBa-Hien Tran, Giulio Franzese, Pietro Michiardi, Maurizio FilipponeNeurIPS 2023 · 4 citations
- Neighbour-Driven Gaussian Process Variational Autoencoders for Scalable Structured Latent ModellingXinxing Shi, Xiaoyu Jiang, Mauricio A. ÁlvarezICML 2025
- Bayesian Basis Function Approximation for Scalable Gaussian Process Priors in Deep Generative ModelsMehmet Yigit Balik, Maksim Sinelnikov, Priscilla Ong, Harri LähdesmäkiICML 2025
- Modeling Temporal scRNA-seq Data with Latent Gaussian Process and Optimal TransportMehmet Yigit Balik, Harri LähdesmäkiICML 2026
Builds on6
- Bayesian Deep Learning and a Probabilistic Perspective of GeneralizationAndrew Gordon Wilson, Pavel IzmailovNeurIPS 2020 · 845 citations
- What Are Bayesian Neural Network Posteriors Really Like?Pavel Izmailov, Sharad Vikram, Matthew D. Hoffman, Andrew Gordon WilsonICML 2021 · 458 citations
- Cyclical Stochastic Gradient MCMC for Bayesian Deep LearningRuqi Zhang, Chunyuan Li, Jianyi Zhang, Changyou Chen et al.ICLR 2020 · 292 citations
- Markovian Gaussian Process Variational AutoencodersHarrison Zhu, Carles Balsells Rodas, Yingzhen LiICML 2023 · 24 citations
- Model Selection for Bayesian AutoencodersBa-Hien Tran, Simone Rossi, Dimitrios Milios, Pietro Michiardi et al.NeurIPS 2021 · 15 citations
Related papers
- Latent variable model for high-dimensional point process with structured missingnessMaksim Sinelnikov, Manuel Haussmann, Harri LähdesmäkiICML 2024 · 1 citation
- Input Dependent Sparse Gaussian ProcessesBahram Jafrasteh, Carlos Villacampa-Calvo, Daniel Hernández-LobatoICML 2022 · 7 citations
- Langevin Autoencoders for Learning Deep Latent Variable ModelsShohei Taniguchi, Yusuke Iwasawa, Wataru Kumagai, Yutaka MatsuoNeurIPS 2022 · 2 citations
- Function-space Inference with Sparse Implicit ProcessesSimón Rodríguez Santana, Bryan Zaldivar, Daniel Hernández-LobatoICML 2022 · 13 citations
- Global inducing point variational posteriors for Bayesian neural networks and deep Gaussian processesSebastian W. Ober, Laurence AitchisonICML 2021 · 65 citations
