Amortized Bayesian Multilevel Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Habermann, Daniel, Schmitt, Marvin, Kühmichel, Lars, Bulling, Andreas, Radev, Stefan T., Bürkner, Paul-Christian
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915344465002496
author Habermann, Daniel
Schmitt, Marvin
Kühmichel, Lars
Bulling, Andreas
Radev, Stefan T.
Bürkner, Paul-Christian
author_facet Habermann, Daniel
Schmitt, Marvin
Kühmichel, Lars
Bulling, Andreas
Radev, Stefan T.
Bürkner, Paul-Christian
contents Multilevel models (MLMs) are a central building block of the Bayesian workflow. They enable joint, interpretable modeling of data across hierarchical levels and provide a fully probabilistic quantification of uncertainty. Despite their well-recognized advantages, MLMs pose significant computational challenges, often rendering their estimation and evaluation intractable within reasonable time constraints. Recent advances in simulation-based inference offer promising solutions for addressing complex probabilistic models using deep generative networks. However, the utility and reliability of deep learning methods for estimating Bayesian MLMs remains largely unexplored, especially when compared with gold-standard samplers. To this end, we explore a family of neural network architectures that leverage the probabilistic factorization of multilevel models to facilitate efficient neural network training and subsequent near-instant posterior inference on unseen datasets. We test our method on several real-world case studies and provide comprehensive comparisons to Stan's gold standard sampler, where possible. Finally, we provide an open-source implementation of our methods to stimulate further research in the nascent field of amortized Bayesian inference.
format Preprint
id arxiv_https___arxiv_org_abs_2408_13230
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Amortized Bayesian Multilevel Models
Habermann, Daniel
Schmitt, Marvin
Kühmichel, Lars
Bulling, Andreas
Radev, Stefan T.
Bürkner, Paul-Christian
Machine Learning
Computation
Multilevel models (MLMs) are a central building block of the Bayesian workflow. They enable joint, interpretable modeling of data across hierarchical levels and provide a fully probabilistic quantification of uncertainty. Despite their well-recognized advantages, MLMs pose significant computational challenges, often rendering their estimation and evaluation intractable within reasonable time constraints. Recent advances in simulation-based inference offer promising solutions for addressing complex probabilistic models using deep generative networks. However, the utility and reliability of deep learning methods for estimating Bayesian MLMs remains largely unexplored, especially when compared with gold-standard samplers. To this end, we explore a family of neural network architectures that leverage the probabilistic factorization of multilevel models to facilitate efficient neural network training and subsequent near-instant posterior inference on unseen datasets. We test our method on several real-world case studies and provide comprehensive comparisons to Stan's gold standard sampler, where possible. Finally, we provide an open-source implementation of our methods to stimulate further research in the nascent field of amortized Bayesian inference.
title Amortized Bayesian Multilevel Models
topic Machine Learning
Computation
url https://arxiv.org/abs/2408.13230