Conditional Density Estimation with Bayesian Normalising Flows
Not on the map. Why?
Abstract (arXiv)
Modeling complex conditional distributions is critical in a variety of settings. Despite a long tradition of research into conditional density estimation, current methods employ either simple parametric forms or are difficult to learn in practice. This paper employs normalising flows as a flexible likelihood model and presents an efficient method for fitting them to complex densities. These estimators must trade-off between modeling distributional complexity, functional complexity and heteroscedasticity without overfitting. We recognize these trade-offs as modeling decisions and develop a Bayesian framework for placing priors over these conditional density estimators using variational Bayesian neural networks. We evaluate this method on several small benchmark regression datasets, on some of which it obtains state of the art performance. Finally, we apply the method to two spatial density modeling tasks with over 1 million datapoints using the New York City yellow taxi dataset and the Chicago crime dataset.
Key facts. Cited by 54 papers in the map (data through 7 Oct 2026). First posted on arXiv 14 Feb 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 6; 2023: 10; 2024: 12; 2025: 10; 2026: 8.
54 papers.
Sort:
By area:
- 2021Low-Light Image Enhancement with Normalizing Flowcited by 184
- 2021VCNet and Functional Targeted Regularization For Learning Causal Effects of Continuous Treatmentscited by 59
- 2023High Fidelity Image Counterfactuals with Probabilistic Causal Modelscited by 53
- 2022Probabilistic Conformal Prediction Using Conditional Random Samplescited by 47
- 2023Counterfactual Identifiability of Bijective Causal Modelscited by 42
- 2022Rhino: Deep Causal Temporal Relationship Learning With History-dependent Noisecited by 30
- 2023Conditional Optimal Transport on Function Spacescited by 26
- 2024Benchmarking Counterfactual Image Generationcited by 22
- 2022Normalizing Flows for Interventional Density Estimationcited by 21
- 2022Improvement-Focused Causal Recourse (ICR)cited by 20
- 2023Bounds on Representation-Induced Confounding Bias for Treatment Effect Estimationcited by 20
- 2024Conformal Prediction for Causal Effects of Continuous Treatmentscited by 19
38 more
- 2023Reliable Off-Policy Learning for Dosage Combinationscited by 15
- 2024Mitigating attribute amplification in counterfactual image generationcited by 12
- 2024Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal Learnercited by 10
- 2023A generative flow for conditional sampling via optimal transportcited by 7
- 2022Conditional Measurement Density Estimation in Sequential Monte Carlo via Normalizing Flowcited by 6
- 2022Short-Term Density Forecasting of Low-Voltage Load using Bernstein-Polynomial Normalizing Flowscited by 6
- 2024A Kernel-Based Conditional Two-Sample Test Using Nearest Neighbors (with Applications to Calibration, Regression Curves, and Simulation-Based Inference)cited by 5
- 2023Efficient SAGE Estimation via Causal Structure Learningcited by 4
- 2024Conditional Density Estimation with Histogram Treescited by 4
- 2026Generative models for decision-making under distributional shiftcited by 4
- 2021Flow-based Spatio-Temporal Structured Prediction of Motion Dynamicscited by 3
- 2021Deconvolutional Density Network: Modeling Free-Form Conditional Distributionscited by 3
- 2024TutteNet: Injective 3D Deformations by Composition of 2D Mesh Deformationscited by 3
- 2021Uncertainty estimation under model misspecification in neural network regressioncited by 2
- 2021Graph network for learning bi-directional physicscited by 2
- 2023Conditional Kernel Imitation Learning for Continuous State Environmentscited by 2
- 2024Generative Modeling: A Reviewcited by 2
- 2025Accelerate TarFlow Sampling with GS-Jacobi Iterationcited by 2
- 2025Adaptive Bayesian Single-Shot Quantum Sensingcited by 2
- 2025Simulation-based inference for Precision Neutrino Physics through Neural Monte Carlo tuningcited by 2
- 2021Semi-supervised Learning with Missing Values Imputationcited by 1
- 2021Viscos Flows: Variational Schur Conditional Sampling With Normalizing Flowscited by 1
- 2024Probabilistic Forecasting of Irregular Time Series via Conditional Flowscited by 1
- 2025GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomescited by 1
- 2025ActiveCQ: Active Estimation of Causal Quantitiescited by 1
- 2026Harnessing Synthetic Data from Generative AI for Statistical Inferencecited by 1
- 2026GGMPs: Generalized Gaussian Mixture Processescited by 1
- 2026UQ-SHRED: uncertainty quantification of shallow recurrent decoder networks for sparse sensing via engressioncited by 1
- 2023Weakly Supervised Label Learning Flowscited by 0
- 2023On Certified Generalization in Structured Predictioncited by 0
- 2024LoLiSRFlow: Joint Single Image Low-light Enhancement and Super-resolution via Cross-scale Transformer-based Conditional Flowcited by 0
- 2024Generative modeling of density regression through tree flowscited by 0
- 2024Learning state and proposal dynamics in state-space models using differentiable particle filters and neural networkscited by 0
- 2025CEC-MMR: Cross-Entropy Clustering Approach to Multi-Modal Regressioncited by 0
- 2025ExOSITO: Explainable Off-Policy Learning with Side Information for Intensive Care Unit Blood Test Orderscited by 0
- 2025Inherited or produced? Inferring protein production kinetics when protein counts are shaped by a cell's division historycited by 0
- 2025Smoothing-Based Conformal Prediction for Balancing Efficiency and Interpretabilitycited by 0
- 2025Addressing prior dependence in hierarchical Bayesian modeling for PTA data analysis I: Methodology and implementationcited by 0
50 most cited of 54 All 54 (JSON)
- 2026A Doeblin-Anchored Contrastive Chart for Learning Markov Transition Kernels1 Jun 2026
- 2026Decomposing Ensemble Spread in Lorenz '96 With Learned Stochastic Parameterizations21 May 2026
- 2026Super-Level-Set Regression: Conditional Quantiles via Volume Minimization7 May 2026
- 2026Complex normalizing flows can almost be information K"ahler-Ricci flows20 Apr 2026
- 2026Generative models for decision-making under distributional shift6 Apr 2026
- 2026UQ-SHRED: uncertainty quantification of shallow recurrent decoder networks for sparse sensing via engression1 Apr 2026
- 2026GGMPs: Generalized Gaussian Mixture Processes11 Mar 2026
- 2026Harnessing Synthetic Data from Generative AI for Statistical Inference5 Mar 2026
- 2025Addressing prior dependence in hierarchical Bayesian modeling for PTA data analysis I: Methodology and implementation5 Nov 2025
- 2025ActiveCQ: Active Estimation of Causal Quantities29 Sep 2025
- 2025GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomes26 Sep 2025
- 2025Smoothing-Based Conformal Prediction for Balancing Efficiency and Interpretability26 Sep 2025
38 more
- 2025Simulation-based inference for Precision Neutrino Physics through Neural Monte Carlo tuning31 Jul 2025
- 2025Adaptive Bayesian Single-Shot Quantum Sensing22 Jul 2025
- 2025Inherited or produced? Inferring protein production kinetics when protein counts are shaped by a cell's division history11 Jun 2025
- 2025Accelerate TarFlow Sampling with GS-Jacobi Iteration19 May 2025
- 2025ExOSITO: Explainable Off-Policy Learning with Side Information for Intensive Care Unit Blood Test Orders24 Apr 2025
- 2025CEC-MMR: Cross-Entropy Clustering Approach to Multi-Modal Regression9 Apr 2025
- 2024Generative Modeling: A Review24 Dec 2024
- 2024Learning state and proposal dynamics in state-space models using differentiable particle filters and neural networks23 Nov 2024
- 2024Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal Learner5 Nov 2024
- 2024Conditional Density Estimation with Histogram Trees15 Oct 2024
- 2024A Kernel-Based Conditional Two-Sample Test Using Nearest Neighbors (with Applications to Calibration, Regression Curves, and Simulation-Based Inference)23 Jul 2024
- 2024Conformal Prediction for Causal Effects of Continuous Treatments3 Jul 2024
- 2024TutteNet: Injective 3D Deformations by Composition of 2D Mesh Deformations17 Jun 2024
- 2024Generative modeling of density regression through tree flows7 Jun 2024
- 2024Benchmarking Counterfactual Image Generation29 Mar 2024
- 2024Mitigating attribute amplification in counterfactual image generation14 Mar 2024
- 2024LoLiSRFlow: Joint Single Image Low-light Enhancement and Super-resolution via Cross-scale Transformer-based Conditional Flow29 Feb 2024
- 2024Probabilistic Forecasting of Irregular Time Series via Conditional Flows9 Feb 2024
- 2023Bounds on Representation-Induced Confounding Bias for Treatment Effect Estimation19 Nov 2023
- 2023Conditional Optimal Transport on Function Spaces9 Nov 2023
- 2023Conditional Kernel Imitation Learning for Continuous State Environments24 Aug 2023
- 2023A generative flow for conditional sampling via optimal transport9 Jul 2023
- 2023High Fidelity Image Counterfactuals with Probabilistic Causal Models27 Jun 2023
- 2023On Certified Generalization in Structured Prediction15 Jun 2023
- 2023Reliable Off-Policy Learning for Dosage Combinations31 May 2023
- 2023Efficient SAGE Estimation via Causal Structure Learning6 Apr 2023
- 2023Weakly Supervised Label Learning Flows19 Feb 2023
- 2023Counterfactual Identifiability of Bijective Causal Models4 Feb 2023
- 2022Improvement-Focused Causal Recourse (ICR)27 Oct 2022
- 2022Rhino: Deep Causal Temporal Relationship Learning With History-dependent Noise26 Oct 2022
- 2022Normalizing Flows for Interventional Density Estimation13 Sep 2022
- 2022Probabilistic Conformal Prediction Using Conditional Random Samples14 Jun 2022
- 2022Short-Term Density Forecasting of Low-Voltage Load using Bernstein-Polynomial Normalizing Flows29 Apr 2022
- 2022Conditional Measurement Density Estimation in Sequential Monte Carlo via Normalizing Flow16 Mar 2022
- 2021Graph network for learning bi-directional physics13 Dec 2021
- 2021Uncertainty estimation under model misspecification in neural network regression23 Nov 2021
- 2021Low-Light Image Enhancement with Normalizing Flow13 Sep 2021
- 2021Viscos Flows: Variational Schur Conditional Sampling With Normalizing Flows6 Jul 2021
50 newest of 54 All 54 (JSON)
- 2021Low-Light Image Enhancement with Normalizing Flowcited by 184
- 2021VCNet and Functional Targeted Regularization For Learning Causal Effects of Continuous Treatmentscited by 59
- 2023High Fidelity Image Counterfactuals with Probabilistic Causal Modelscited by 53
- 2022Probabilistic Conformal Prediction Using Conditional Random Samplescited by 47
- 2023Counterfactual Identifiability of Bijective Causal Modelscited by 42
- 2022Rhino: Deep Causal Temporal Relationship Learning With History-dependent Noisecited by 30
- 2023Conditional Optimal Transport on Function Spacescited by 26
- 2024Benchmarking Counterfactual Image Generationcited by 22
- 2022Normalizing Flows for Interventional Density Estimationcited by 21
- 2022Improvement-Focused Causal Recourse (ICR)cited by 20
- 2023Bounds on Representation-Induced Confounding Bias for Treatment Effect Estimationcited by 20
- 2024Conformal Prediction for Causal Effects of Continuous Treatmentscited by 19
12 most cited of 53 All 53 (JSON)
- 2026A Doeblin-Anchored Contrastive Chart for Learning Markov Transition Kernels1 Jun 2026
- 2026Decomposing Ensemble Spread in Lorenz '96 With Learned Stochastic Parameterizations21 May 2026
- 2026Super-Level-Set Regression: Conditional Quantiles via Volume Minimization7 May 2026
- 2026Complex normalizing flows can almost be information K"ahler-Ricci flows20 Apr 2026
- 2026Generative models for decision-making under distributional shift6 Apr 2026
- 2026UQ-SHRED: uncertainty quantification of shallow recurrent decoder networks for sparse sensing via engression1 Apr 2026
- 2026GGMPs: Generalized Gaussian Mixture Processes11 Mar 2026
- 2026Harnessing Synthetic Data from Generative AI for Statistical Inference5 Mar 2026
- 2025Addressing prior dependence in hierarchical Bayesian modeling for PTA data analysis I: Methodology and implementation5 Nov 2025
- 2025ActiveCQ: Active Estimation of Causal Quantities29 Sep 2025
- 2025GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomes26 Sep 2025
- 2025Smoothing-Based Conformal Prediction for Balancing Efficiency and Interpretability26 Sep 2025
12 newest of 53 All 53 (JSON)