Composing graphical models with neural networks for structured representations and fast inference
Not on the map. Why?
Abstract (arXiv)
We propose a general modeling and inference framework that composes probabilistic graphical models with deep learning methods and combines their respective strengths. Our model family augments graphical structure in latent variables with neural network observation models. For inference, we extend variational autoencoders to use graphical model approximating distributions with recognition networks that output conjugate potentials. All components of these models are learned simultaneously with a single objective, giving a scalable algorithm that leverages stochastic variational inference, natural gradients, graphical model message passing, and the reparameterization trick. We illustrate this framework with several example models and an application to mouse behavioral phenotyping.
Key facts. Cited by 113 papers in the map (data through 7 Oct 2026). First posted on arXiv 20 Mar 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 1; 2020: 0; 2021: 29; 2022: 20; 2023: 22; 2024: 16; 2025: 15; 2026: 9.
113 papers.
Sort:
By area:
- 2022Identifiability of deep generative models without auxiliary informationcited by 70
- 2021Disentangling Identifiable Features from Noisy Data with Structured Nonlinear ICAcited by 64
- 2023Posterior Collapse and Latent Variable Non-identifiabilitycited by 57
- 2024SoftVQ-VAE: Efficient 1-Dimensional Continuous Tokenizercited by 55
- 2023Learning nonparametric latent causal graphs with unknown interventionscited by 34
- 2021Multi-Facet Clustering Variational Autoencoderscited by 28
- 2021A Deep Variational Approach to Clustering Survival Datacited by 25
- 2022Learning Options via Compressioncited by 22
- 2023Generative learning for nonlinear dynamicscited by 22
- 2021Deep Bayesian Unsupervised Lifelong Learningcited by 21
- 2022Controllable Data Generation by Deep Learning: A Reviewcited by 20
- 2021Amortized Variational Inference for Simple Hierarchical Modelscited by 19
38 more
- 2021Deep Explicit Duration Switching Models for Time Seriescited by 17
- 2022Emergent Communication through Metropolis-Hastings Naming Game with Deep Generative Modelscited by 16
- 2025LLM/Agent-as-Data-Analyst: A Surveycited by 15
- 2021Cauchy-Schwarz Regularized Autoencodercited by 13
- 2022Concept Graph Neural Networks for Surgical Video Understandingcited by 13
- 2022ProbNeRF: Uncertainty-Aware Inference of 3D Shapes from 2D Imagescited by 12
- 2023Generative Causal Representation Learning for Out-of-Distribution Motion Forecastingcited by 12
- 2021On Incorporating Inductive Biases into VAEscited by 11
- 2022Neural Graphical Modelscited by 11
- 2023Revisiting Structured Variational Autoencoderscited by 11
- 2024Variational Inference for Uncertainty Quantification: an Analysis of Trade-offscited by 11
- 2021Variational Inference for Continuous-Time Switching Dynamical Systemscited by 9
- 2023On the Identifiability of Switching Dynamical Systemscited by 9
- 2021Breiman's two cultures: You don't have to choose sidescited by 8
- 2021Unsupervised Learning of Neurosymbolic Encoderscited by 8
- 2021Toward a `Standard Model' of Machine Learningcited by 8
- 2018CompILE: Compositional Imitation Learning and Executioncited by 7
- 2021Lossless compression with state space models using bits back codingcited by 6
- 2021Learning interaction rules from multi-animal trajectories via augmented behavioral modelscited by 5
- 2021Introspective Robot Perception using Smoothed Predictions from Bayesian Neural Networkscited by 5
- 2022Time Series Clustering with an EM algorithm for Mixtures of Linear Gaussian State Space Modelscited by 5
- 2022Unsupervised representation learning with recognition-parametrised probabilistic modelscited by 5
- 2024Dimension-independent rates for structured neural density estimationcited by 5
- 2021Lossless Compression with Latent Variable Modelscited by 4
- 2021Online Unsupervised Learning of Visual Representations and Categoriescited by 4
- 2022Learning Sequential Latent Variable Models from Multimodal Time Series Datacited by 4
- 2023Neuro-Causal Factor Analysiscited by 4
- 2023Learning multi-modal generative models with permutation-invariant encoders and tighter variational objectivescited by 4
- 2023Identifiable Feature Learning for Spatial Data with Nonlinear ICAcited by 4
- 2024Provably Scalable Black-Box Variational Inference with Structured Variational Familiescited by 4
- 2024Gradient-free variational learning with conditional mixture networkscited by 4
- 2025Conformal Prediction for Time-series Forecasting with Change Pointscited by 4
- 2025A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systemscited by 4
- 2021Segmenting Hybrid Trajectories using Latent ODEscited by 3
- 2021Self-Supervised Inference in State-Space Modelscited by 3
- 2022Markov Chain Monte Carlo for Continuous-Time Switching Dynamical Systemscited by 3
- 2023A Causal Ordering Prior for Unsupervised Representation Learningcited by 3
- 2023Variational Temporal IRT: Fast, Accurate, and Explainable Inference of Dynamic Learner Proficiencycited by 3
50 most cited of 113 All 113 (JSON)
- 2026PAC-Bayesian Reconstruction Guarantees for Time Series Variational Autoencoders4 Sep 2026
- 2026RAMP: Recognition parametrisation by Amortised Message Passing21 Jul 2026
- 2026Diffusion-Driven State Space Models19 Jun 2026
- 2026Efficient Learning of Deep State Space Models via Importance Smoothing20 May 2026
- 2026StrEBM: A Structured Latent Energy-Based Model for Blind Source Separation19 Apr 2026
- 2026AR-Flow VAE: A Structured Autoregressive Flow Prior Variational Autoencoder for Unsupervised Blind Source Separation15 Mar 2026
- 2026Efficient Online Variational Estimation via Monte Carlo Sampling6 Feb 2026
- 2026Independent Component Discovery in Temporal Count Data29 Jan 2026
- 2026Meta-probabilistic Modeling8 Jan 2026
- 2025Neural Stochastic Flows: Solver-Free Modelling and Inference for SDE Solutions29 Oct 2025
- 2025LLM/Agent-as-Data-Analyst: A Survey28 Sep 2025
- 2025A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systems26 Sep 2025
38 more
- 2025Markov Missing Graph: A Graphical Approach for Missing Data Imputation3 Sep 2025
- 2025Conformal Prediction for Time-series Forecasting with Change Points2 Sep 2025
- 2025Structural Equation-VAE: Disentangled Latent Representations for Tabular Data8 Aug 2025
- 2025Cloud Model Characteristic Function Auto-Encoder: Integrating Cloud Model Theory with MMD Regularization for Enhanced Generative Modeling6 Aug 2025
- 2025Multi-Component VAE with Gaussian Markov Random Field16 Jul 2025
- 2025Optimization Guarantees for Square-Root Natural-Gradient Variational Inference10 Jul 2025
- 2025Intervening to Learn and Compose Causally Disentangled Representations7 Jul 2025
- 2025Expressive Score-Based Priors for Distribution Matching with Geometry-Preserving Regularization17 Jun 2025
- 2025Maximum Likelihood Learning of Latent Dynamics Without Reconstruction29 May 2025
- 2025Measuring In-Context Computation Complexity via Hidden State Prediction17 Mar 2025
- 2025Stability-based Generalization Bounds for Variational Inference17 Feb 2025
- 2025Diffusion Models for Inverse Problems in the Exponential Family9 Feb 2025
- 2024SoftVQ-VAE: Efficient 1-Dimensional Continuous Tokenizer14 Dec 2024
- 2024Dimension-independent rates for structured neural density estimation22 Nov 2024
- 2024Breaking the curse of dimensionality in structured density estimation10 Oct 2024
- 2024Latent mixed-effect models for high-dimensional longitudinal data17 Sep 2024
- 2024Gradient-free variational learning with conditional mixture networks29 Aug 2024
- 2024Preference-Optimized Pareto Set Learning for Blackbox Optimization19 Aug 2024
- 2024A Markov Random Field Multi-Modal Variational AutoEncoder18 Aug 2024
- 2024A study of animal action segmentation algorithms across supervised, unsupervised, and semi-supervised learning paradigms23 Jul 2024
- 2024Tree-based variational inference for Poisson log-normal models25 Jun 2024
- 2024Deep Latent Variable Modeling of Physiological Signals29 May 2024
- 2024Deep Dependency Networks and Advanced Inference Schemes for Multi-Label Classification17 Apr 2024
- 2024Variational Inference for Uncertainty Quantification: an Analysis of Trade-offs20 Mar 2024
- 2024The VampPrior Mixture Model6 Feb 2024
- 2024Importance sampling for online variational learning5 Feb 2024
- 2024Provably Scalable Black-Box Variational Inference with Structured Variational Families19 Jan 2024
- 2024Denoising Diffusion Variational Inference: Diffusion Models as Expressive Variational Posteriors5 Jan 2024
- 2023Identifiable Feature Learning for Spatial Data with Nonlinear ICA28 Nov 2023
- 2023Towards Improving Robustness Against Common Corruptions using Mixture of Class Specific Experts16 Nov 2023
- 2023Variational Temporal IRT: Fast, Accurate, and Explainable Inference of Dynamic Learner Proficiency14 Nov 2023
- 2023Generative learning for nonlinear dynamics7 Nov 2023
- 2023Classification-Aided Robust Multiple Target Tracking Using Neural Enhanced Message Passing19 Oct 2023
- 2023Entropic Matching for Expectation Propagation of Markov Jump Processes27 Sep 2023
- 2023Amortised Inference in Bayesian Neural Networks6 Sep 2023
- 2023Learning multi-modal generative models with permutation-invariant encoders and tighter variational objectives1 Sep 2023
- 2023A Causal Ordering Prior for Unsupervised Representation Learning11 Jul 2023
- 2023Exploiting Inferential Structure in Neural Processes27 Jun 2023
50 newest of 113 All 113 (JSON)
- 2019Imitation Learning of Factored Multi-agent Reactive Models12 Mar 2019
- 2018CompILE: Compositional Imitation Learning and Execution4 Dec 2018
- 2025LLM/Agent-as-Data-Analyst: A Surveycited by 15
- 2025LLM/Agent-as-Data-Analyst: A Survey28 Sep 2025
- 2022Identifiability of deep generative models without auxiliary informationcited by 70
- 2021Disentangling Identifiable Features from Noisy Data with Structured Nonlinear ICAcited by 64
- 2023Posterior Collapse and Latent Variable Non-identifiabilitycited by 57
- 2024SoftVQ-VAE: Efficient 1-Dimensional Continuous Tokenizercited by 55
- 2023Learning nonparametric latent causal graphs with unknown interventionscited by 34
- 2021Multi-Facet Clustering Variational Autoencoderscited by 28
- 2021A Deep Variational Approach to Clustering Survival Datacited by 25
- 2022Learning Options via Compressioncited by 22
- 2023Generative learning for nonlinear dynamicscited by 22
- 2021Deep Bayesian Unsupervised Lifelong Learningcited by 21
- 2022Controllable Data Generation by Deep Learning: A Reviewcited by 20
- 2021Amortized Variational Inference for Simple Hierarchical Modelscited by 19
12 most cited of 104 All 104 (JSON)
- 2026PAC-Bayesian Reconstruction Guarantees for Time Series Variational Autoencoders4 Sep 2026
- 2026RAMP: Recognition parametrisation by Amortised Message Passing21 Jul 2026
- 2026Efficient Learning of Deep State Space Models via Importance Smoothing20 May 2026
- 2026StrEBM: A Structured Latent Energy-Based Model for Blind Source Separation19 Apr 2026
- 2026AR-Flow VAE: A Structured Autoregressive Flow Prior Variational Autoencoder for Unsupervised Blind Source Separation15 Mar 2026
- 2026Efficient Online Variational Estimation via Monte Carlo Sampling6 Feb 2026
- 2026Independent Component Discovery in Temporal Count Data29 Jan 2026
- 2026Meta-probabilistic Modeling8 Jan 2026
- 2025Neural Stochastic Flows: Solver-Free Modelling and Inference for SDE Solutions29 Oct 2025
- 2025A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systems26 Sep 2025
- 2025Markov Missing Graph: A Graphical Approach for Missing Data Imputation3 Sep 2025
- 2025Conformal Prediction for Time-series Forecasting with Change Points2 Sep 2025
12 newest of 104 All 104 (JSON)